Lines Matching refs:operands
40 std::vector<RunTimeOperandInfo>& operands) { in SVDF() argument
41 input_ = GetInput(operation, operands, kInputTensor); in SVDF()
42 weights_feature_ = GetInput(operation, operands, kWeightsFeatureTensor); in SVDF()
43 weights_time_ = GetInput(operation, operands, kWeightsTimeTensor); in SVDF()
44 bias_ = GetInput(operation, operands, kBiasTensor); in SVDF()
45 state_in_ = GetInput(operation, operands, kStateInTensor); in SVDF()
47 params_.rank_ = getScalarData<int>(*GetInput(operation, operands, kRankParam)); in SVDF()
49 *GetInput(operation, operands, kActivationParam))); in SVDF()
51 state_out_ = GetOutput(operation, operands, kStateOutTensor); in SVDF()
52 output_ = GetOutput(operation, operands, kOutputTensor); in SVDF()
56 std::vector<RunTimeOperandInfo> &operands, in Prepare() argument
60 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare()
66 GetInput(operation, operands, SVDF::kInputTensor); in Prepare()
68 GetInput(operation, operands, SVDF::kWeightsFeatureTensor); in Prepare()
70 GetInput(operation, operands, SVDF::kWeightsTimeTensor); in Prepare()
74 const int rank = getScalarData<int>(*GetInput(operation, operands, kRankParam)); in Prepare()
84 GetInput(operation, operands, kBiasTensor); in Prepare()