Home
last modified time | relevance | path

Searched defs:use_cifg (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc187 const bool use_cifg = (input_to_input_weights == nullptr); in CheckInputTensorDimensions() local
328 const bool use_cifg = (input_to_input_weights == nullptr); in Prepare() local
Dlstm_eval.cc123 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepWithAuxInput() local
477 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepWithAuxInput() local
917 const bool use_cifg = (input_to_input_weights == nullptr); in EvalFloat() local
1120 const bool use_cifg = (input_to_input_weights == nullptr); in EvalHybrid() local
Dlstm_test.cc37 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
272 bool use_cifg, bool use_peephole, in HybridLSTMOpModel()
1634 bool use_cifg, bool use_peephole, in LayerNormLSTMOpModel()
1648 bool use_cifg, bool use_peephole, in HybridLayerNormLSTMOpModel()
Dlstm.cc130 const bool use_cifg = (input_to_input_weights == nullptr); in CheckInputTensorDimensions() local
404 const bool use_cifg = (input_to_input_weights == nullptr); in Prepare() local
Doptional_tensor_test.cc33 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
Dunidirectional_sequence_lstm_test.cc35 int sequence_length, bool time_major, bool use_cifg, in UnidirectionalLSTMOpModel()
244 bool time_major, bool use_cifg, bool use_peephole, in HybridUnidirectionalLSTMOpModel()
Dbidirectional_sequence_lstm.cc280 const bool use_cifg = (input_to_input_weights == nullptr); in CheckLstmTensorDimensionsAndTypes() local
Dbidirectional_sequence_lstm_test.cc38 int sequence_length, bool use_cifg, in BidirectionalLSTMOpModel()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2259 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()