Searched refs:weights_feature_ (Results 1 – 2 of 2) sorted by relevance
32 weights_feature_ = GetInput(operation, operands, kWeightsFeatureTensor); in SVDF()111 getNumberOfElements(weights_feature_->shape())); in Eval()112 convertFloat16ToFloat32(reinterpret_cast<_Float16*>(weights_feature_->buffer), in Eval()134 reinterpret_cast<float*>(weights_feature_->buffer), in Eval()156 const int num_filters = SizeOfDimension(weights_feature_, 0); in EvalFloat32()
64 const RunTimeOperandInfo* weights_feature_; variable