HomeSort by relevance Sort by last modified time
    Searched refs:kForgetGateBiasTensor (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
lstm_utils.h 42 kForgetGateBiasTensor = 13,
identify_lstm_merge_inputs.cc 131 model->GetArray(src_op->inputs[kForgetGateBiasTensor]),
177 DeleteArrayIfUnused(src_op->inputs[kForgetGateBiasTensor], model);
identify_lstm_split_inputs.cc 130 CopySubArrayToArray(model, &(lstm_cell_op->inputs[kForgetGateBiasTensor]),
  /frameworks/ml/nn/common/operations/
LSTM.h 83 static constexpr int kForgetGateBiasTensor = 13;
LSTM.cpp 59 forget_gate_bias_ = GetInput(operation, operands, kForgetGateBiasTensor);
189 GetInput(operation, operands, LSTMCell::kForgetGateBiasTensor);
  /external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc 58 constexpr int kForgetGateBiasTensor = 13;
182 GetInput(context, node, kForgetGateBiasTensor);
338 GetInput(context, node, kForgetGateBiasTensor);
unidirectional_sequence_lstm.cc 58 constexpr int kForgetGateBiasTensor = 13;
182 GetInput(context, node, kForgetGateBiasTensor);
340 GetInput(context, node, kForgetGateBiasTensor);

Completed in 112 milliseconds