Home
last modified time | relevance | path

Searched refs:AddVariableInput (Results 1 – 15 of 15) 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()
Dunidirectional_sequence_lstm_test_util.h102 output_state_ = AddVariableInput( in n_batch_()
105 AddVariableInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}); in n_batch_()
Dsvdf_test.cc146 activation_state_ = AddVariableInput( in BaseSVDFOpModel()
484 activation_state_ = AddVariableInput( in IntegerSVDFOpModel()
Dunidirectional_sequence_gru_test.cc40 AddVariableInput(TensorData{TensorType_FLOAT32, {n_batch, n_output}}); in GRUOpModel()
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_rnn_test.cc186 hidden_state_ = AddVariableInput(TensorType_FLOAT32); in UnidirectionalRNNOpModel()
Dbidirectional_sequence_rnn_test.cc683 fw_hidden_state_ = AddVariableInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
687 bw_hidden_state_ = AddVariableInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
Dtest_util.cc95 int SingleOpModel::AddVariableInput(const TensorData& t) { in AddVariableInput() function in tflite::SingleOpModel
Dunidirectional_sequence_lstm_test.cc2595 AddVariableInput({TensorType_INT16, in UnidirectionalSequenceLSTMIntegerOpModel()
2599 AddVariableInput({TensorType_INT16, in UnidirectionalSequenceLSTMIntegerOpModel()
Dtest_util.h202 int AddVariableInput(const TensorData& t);
/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.cc2820 hidden_state_ = AddVariableInput(TensorType_FLOAT32); in RNNOpModel()
3012 activation_state_ = AddVariableInput( in BaseSVDFOpModel()
3237 input_activation_state_ = AddVariableInput(TensorType_FLOAT32); in LSTMOpModel()
3238 input_cell_state_ = AddVariableInput(TensorType_FLOAT32); in LSTMOpModel()