Lines Matching refs:operation
39 SVDF::SVDF(const Operation& operation,
41 input_ = GetInput(operation, operands, kInputTensor);
42 weights_feature_ = GetInput(operation, operands, kWeightsFeatureTensor);
43 weights_time_ = GetInput(operation, operands, kWeightsTimeTensor);
44 bias_ = GetInput(operation, operands, kBiasTensor);
45 state_in_ = GetInput(operation, operands, kStateInTensor);
47 params_.rank_ = getScalarData<int>(*GetInput(operation, operands, kRankParam));
49 *GetInput(operation, operands, kActivationParam)));
51 state_out_ = GetOutput(operation, operands, kStateOutTensor);
52 output_ = GetOutput(operation, operands, kOutputTensor);
55 bool SVDF::Prepare(const Operation &operation,
60 const int num_inputs = NumInputsWithValues(operation, operands);
63 NN_CHECK_EQ(NumOutputs(operation), 2);
66 GetInput(operation, operands, SVDF::kInputTensor);
68 GetInput(operation, operands, SVDF::kWeightsFeatureTensor);
70 GetInput(operation, operands, SVDF::kWeightsTimeTensor);
74 const int rank = getScalarData<int>(*GetInput(operation, operands, kRankParam));
84 GetInput(operation, operands, kBiasTensor);