Lines Matching refs:operands
30 RNN::RNN(const Operation& operation, RunTimeOperandInfo* operands) { in RNN() argument
32 input_ = GetInput(operation, operands, kInputTensor); in RNN()
33 weights_ = GetInput(operation, operands, kWeightsTensor); in RNN()
34 recurrent_weights_ = GetInput(operation, operands, kRecurrentWeightsTensor); in RNN()
35 hidden_state_in_ = GetInput(operation, operands, kHiddenStateInTensor); in RNN()
36 bias_ = GetInput(operation, operands, kBiasTensor); in RNN()
39 getScalarData<int32_t>(operands[operation.inputs[kActivationParam]])); in RNN()
41 hidden_state_out_ = GetOutput(operation, operands, kHiddenStateOutTensor); in RNN()
42 output_ = GetOutput(operation, operands, kOutputTensor); in RNN()
45 bool RNN::Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* hiddenStateShape, in Prepare() argument
49 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare()
53 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
54 const RunTimeOperandInfo* input_weights = GetInput(operation, operands, kWeightsTensor); in Prepare()
56 GetInput(operation, operands, kRecurrentWeightsTensor); in Prepare()
57 const RunTimeOperandInfo* bias = GetInput(operation, operands, kBiasTensor); in Prepare()