Home
last modified time | relevance | path

Searched refs:kWeightsFeatureTensor (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/operations/
DSVDF.cpp36 weights_feature_ = GetInput(operation, operands, kWeightsFeatureTensor); in SVDF()
59 kInputTensor, kWeightsFeatureTensor, kWeightsTimeTensor, kStateInTensor, in Prepare()
76 GetInput(operation, operands, SVDF::kWeightsFeatureTensor); in Prepare()
DSVDF.h48 static constexpr int kWeightsFeatureTensor = 1; variable