Lines Matching refs:operation
44 SVDF::SVDF(const Operation& operation, in SVDF() argument
46 input_ = GetInput(operation, operands, kInputTensor); in SVDF()
47 weights_feature_ = GetInput(operation, operands, kWeightsFeatureTensor); in SVDF()
48 weights_time_ = GetInput(operation, operands, kWeightsTimeTensor); in SVDF()
49 bias_ = GetInput(operation, operands, kBiasTensor); in SVDF()
50 state_in_ = GetInput(operation, operands, kStateInTensor); in SVDF()
52 params_.rank_ = getInt32ScalarData(*GetInput(operation, operands, kRankParam)); in SVDF()
54 *GetInput(operation, operands, kActivationParam))); in SVDF()
56 state_out_ = GetOutput(operation, operands, kStateOutTensor); in SVDF()
57 output_ = GetOutput(operation, operands, kOutputTensor); in SVDF()
60 bool SVDF::Prepare(const Operation &operation, in Prepare() argument
65 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare()
67 NN_CHECK_EQ(NumOutputs(operation), 2); in Prepare()
70 GetInput(operation, operands, SVDF::kInputTensor); in Prepare()
72 GetInput(operation, operands, SVDF::kWeightsFeatureTensor); in Prepare()
74 GetInput(operation, operands, SVDF::kWeightsTimeTensor); in Prepare()
85 GetInput(operation, operands, kBiasTensor); in Prepare()