/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
D | lstm.cc | 102 const float* cell_gate, bool use_cifg, float clip) { in UpdateLstmCellFloat() 189 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepCalibration() local 311 const bool use_cifg = (input_to_input_weights == nullptr); in EvalCalibration() local
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | unidirectional_sequence_lstm.cc | 151 const bool use_cifg = (input_to_input_weights == nullptr); in PopulateQuantizedLstmParams8x8_16() local 517 const bool use_cifg = (input_to_input_weights == nullptr); in CheckInputTensorDimensions() local 961 const bool use_cifg = (input_to_input_weights == nullptr); in Prepare() local
|
D | lstm_eval.cc | 62 const int8_t* projection_weights_ptr, bool use_cifg, in ComputeRowSums() 220 const float* cell_gate, bool use_cifg, float clip) { in UpdateLstmCellFloat() 574 bool use_cifg, int16_t clip) { in UpdateLstmCellInteger() 838 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepFloat() local 1039 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepHybrid() local 1389 const bool use_cifg = (input_to_input_weight_ptr == nullptr); in LstmStepInteger8x8_16() local 1733 const bool use_cifg = (input_to_input_weights == nullptr); in EvalFloat() local 1931 const bool use_cifg = (input_to_input_weights == nullptr); in EvalHybrid() local
|
D | lstm.cc | 236 const bool use_cifg = (input_to_input_weights == nullptr); in PopulateQuantizedLstmParams8x8_16() local 540 const bool use_cifg = (input_to_input_weights == nullptr); in PopulateQuantizedLstmParams8x8_8() local 877 const bool use_cifg = (input_to_input_weights == nullptr); in CheckInputTensorDimensions() local 1417 const bool use_cifg = (input_to_input_weights == nullptr); in Prepare() local 1600 const bool use_cifg = (input_to_input_weights == nullptr); in Prepare() local
|
D | optional_tensor_test.cc | 31 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
|
D | unidirectional_sequence_lstm_test.cc | 33 int sequence_length, bool time_major, bool use_cifg, in UnidirectionalLSTMOpModel() 289 bool time_major, bool use_cifg, bool use_peephole, in HybridUnidirectionalLSTMOpModel() 2471 bool time_major, bool use_cifg, bool use_peephole, in LayerNormUnidirectionalLSTMOpModel() 2746 bool time_major, bool use_cifg, bool use_peephole, in UnidirectionalSequenceLSTMIntegerOpModel()
|
D | lstm_test.cc | 39 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() 1315 bool use_cifg, bool use_peephole, in LSTMIntegerOpModel() 2130 int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in HybridSparseLSTMOpModel()
|
D | bidirectional_sequence_lstm.cc | 309 const bool use_cifg = (input_to_input_weights == nullptr); in CheckLstmTensorDimensionsAndTypes() local
|
D | bidirectional_sequence_lstm_test.cc | 33 int sequence_length, bool use_cifg, in BidirectionalLSTMOpModel()
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
D | lstm.cc | 72 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepWithAuxInput() local 380 const bool use_cifg = (input_to_input_weights == nullptr); in EvalFloat() local
|
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
D | prepare_quantize_helper.h | 109 bool use_cifg = in GetLstmProperty() local
|
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
D | lstm_full_test.cc | 35 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 3092 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
|