Lines Matching refs:operation
25 RNN::RNN(const Operation& operation,
27 input_ = GetInput(operation, operands, kInputTensor);
28 weights_ = GetInput(operation, operands, kWeightsTensor);
29 recurrent_weights_ = GetInput(operation, operands, kRecurrentWeightsTensor);
30 hidden_state_in_ = GetInput(operation, operands, kHiddenStateInTensor);
31 bias_ = GetInput(operation, operands, kBiasTensor);
34 getScalarData<int32_t>(operands[operation.inputs[kActivationParam]]));
36 hidden_state_out_ = GetOutput(operation, operands, kHiddenStateOutTensor);
37 output_ = GetOutput(operation, operands, kOutputTensor);
40 bool RNN::Prepare(const Operation &operation,
45 const int num_inputs = NumInputsWithValues(operation, operands);
47 NN_CHECK_EQ(NumOutputs(operation), 2);
50 GetInput(operation, operands, kInputTensor);
52 GetInput(operation, operands, kWeightsTensor);
54 GetInput(operation, operands, kRecurrentWeightsTensor);
56 GetInput(operation, operands, kBiasTensor);