Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DSVDF.cpp47 weights_feature_ = GetInput(operation, operands, kWeightsFeatureTensor); in SVDF()
109 const int num_units = weights_feature_->shape().dimensions[0]; in Eval()
111 const int weights_feature_stride = weights_feature_->shape().dimensions[1]; in Eval()
115 const float* weights_feature_ptr = reinterpret_cast<float *>(weights_feature_->buffer); in Eval()
161 weights_feature_ptr = reinterpret_cast<float*>(weights_feature_->buffer); in Eval()
DSVDF.h72 const RunTimeOperandInfo *weights_feature_; variable