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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
lstm_utils.h 41 kInputGateBiasTensor = 12, // Optional
identify_lstm_merge_inputs.cc 125 model->GetArray(src_op->inputs[kInputGateBiasTensor]), 0,
176 DeleteArrayIfUnused(src_op->inputs[kInputGateBiasTensor], model);
identify_lstm_split_inputs.cc 126 CopySubArrayToArray(model, &(lstm_cell_op->inputs[kInputGateBiasTensor]),
  /frameworks/ml/nn/common/operations/
LSTM.h 82 static constexpr int kInputGateBiasTensor = 12; // Optional
LSTM.cpp 58 input_gate_bias_ = GetInput(operation, operands, kInputGateBiasTensor);
180 GetInput(operation, operands, LSTMCell::kInputGateBiasTensor);
  /external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc 57 constexpr int kInputGateBiasTensor = 12; // Optional
173 GetOptionalInputTensor(context, node, kInputGateBiasTensor);
336 GetOptionalInputTensor(context, node, kInputGateBiasTensor);
unidirectional_sequence_lstm.cc 57 constexpr int kInputGateBiasTensor = 12; // Optional
173 GetOptionalInputTensor(context, node, kInputGateBiasTensor);
338 GetOptionalInputTensor(context, node, kInputGateBiasTensor);

Completed in 106 milliseconds