Home
last modified time | relevance | path

Searched refs:AddVariableInput (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Doptional_tensor_test.cc97 input_activation_state_ = AddVariableInput( in LSTMOpModel()
100 AddVariableInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}); in LSTMOpModel()
Dquant_basic_lstm_test.cc64 prev_output_ = AddVariableInput( in QuantizedLSTMOpModel()
74 prev_cell_state_ = AddVariableInput( in QuantizedLSTMOpModel()
Dsvdf_test.cc146 activation_state_ = AddVariableInput( in BaseSVDFOpModel()
484 activation_state_ = AddVariableInput( in IntegerSVDFOpModel()
Dlstm_test.cc104 AddVariableInput({TensorType_FLOAT32, {n_batch, n_output}}); in LSTMOpModel()
105 AddVariableInput({TensorType_FLOAT32, {n_batch, n_cell}}); in LSTMOpModel()
1415 AddVariableInput({TensorType_INT16, in LSTMIntegerOpModel()
1419 AddVariableInput({TensorType_INT16, in LSTMIntegerOpModel()
2217 output_state_ = AddVariableInput(::tflite::TensorData{ in HybridSparseLSTMOpModel()
2219 cell_state_ = AddVariableInput(::tflite::TensorData{ in HybridSparseLSTMOpModel()
Dbidirectional_sequence_lstm_test.cc163 fw_input_activation_state_ = AddVariableInput( in BidirectionalLSTMOpModel()
165 fw_input_cell_state_ = AddVariableInput( in BidirectionalLSTMOpModel()
169 bw_input_activation_state_ = AddVariableInput( in BidirectionalLSTMOpModel()
171 bw_input_cell_state_ = AddVariableInput( in BidirectionalLSTMOpModel()
Dbasic_rnn_test.cc182 hidden_state_ = AddVariableInput(TensorType_FLOAT32); in RNNOpModel()
Dunidirectional_sequence_lstm_test.cc104 output_state_ = AddVariableInput( in UnidirectionalLSTMOpModel()
107 AddVariableInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}); in UnidirectionalLSTMOpModel()
2847 AddVariableInput({TensorType_INT16, in UnidirectionalSequenceLSTMIntegerOpModel()
2851 AddVariableInput({TensorType_INT16, in UnidirectionalSequenceLSTMIntegerOpModel()
Dunidirectional_sequence_rnn_test.cc186 hidden_state_ = AddVariableInput(TensorType_FLOAT32); in UnidirectionalRNNOpModel()
Dtest_util.cc94 int SingleOpModel::AddVariableInput(const TensorData& t) { in AddVariableInput() function in tflite::SingleOpModel
Dbidirectional_sequence_rnn_test.cc683 fw_hidden_state_ = AddVariableInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
687 bw_hidden_state_ = AddVariableInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
Dtest_util.h193 int AddVariableInput(const TensorData& t);
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru_test.cc40 AddVariableInput(TensorData{TensorType_FLOAT32, {n_batch, n_output}}); in GRUOpModel()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc124 AddVariableInput({TensorType_FLOAT32, {n_batch, n_output}}); in LSTMOpModel()
125 AddVariableInput({TensorType_FLOAT32, {n_batch, n_cell}}); in LSTMOpModel()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2743 hidden_state_ = AddVariableInput(TensorType_FLOAT32); in RNNOpModel()
2935 activation_state_ = AddVariableInput( in BaseSVDFOpModel()
3160 input_activation_state_ = AddVariableInput(TensorType_FLOAT32); in LSTMOpModel()
3161 input_cell_state_ = AddVariableInput(TensorType_FLOAT32); in LSTMOpModel()