| /external/tensorflow/tensorflow/lite/kernels/ |
| D | optional_tensor_test.cc | 31 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() argument
41 if (use_cifg) { in LSTMOpModel()
51 if (use_cifg) { in LSTMOpModel()
62 if (use_cifg) { in LSTMOpModel()
75 if (use_cifg) { in LSTMOpModel()
|
| D | lstm_eval.cc | 62 const int8_t* projection_weights_ptr, bool use_cifg, in ComputeRowSums() argument
65 if (!use_cifg) { in ComputeRowSums()
77 if (!use_cifg) { in ComputeRowSums()
92 if (!use_cifg) { in ComputeRowSums()
220 const float* cell_gate, bool use_cifg, float clip) { in UpdateLstmCellFloat() argument
224 if (use_cifg) { in UpdateLstmCellFloat()
574 bool use_cifg, int16_t clip) { in UpdateLstmCellInteger() argument
582 if (use_cifg) { in UpdateLstmCellInteger()
838 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepFloat() local
852 if (!use_cifg) { in LstmStepFloat()
[all …]
|
| D | lstm_test.cc | 39 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() argument
50 if (use_cifg) { in LSTMOpModel()
59 if (use_cifg) { in LSTMOpModel()
69 if (use_cifg) { in LSTMOpModel()
82 if (use_cifg) { in LSTMOpModel()
110 if (use_cifg) { in LSTMOpModel()
1315 bool use_cifg, bool use_peephole, in LSTMIntegerOpModel() argument
1326 if (use_cifg) { in LSTMIntegerOpModel()
1347 if (use_cifg) { in LSTMIntegerOpModel()
1369 if (use_cifg) { in LSTMIntegerOpModel()
[all …]
|
| D | unidirectional_sequence_lstm_test.cc | 33 int sequence_length, bool time_major, bool use_cifg, in UnidirectionalLSTMOpModel() argument
48 if (use_cifg) { in UnidirectionalLSTMOpModel()
58 if (use_cifg) { in UnidirectionalLSTMOpModel()
69 if (use_cifg) { in UnidirectionalLSTMOpModel()
82 if (use_cifg) { in UnidirectionalLSTMOpModel()
111 if (use_cifg) { in UnidirectionalLSTMOpModel()
289 bool time_major, bool use_cifg, bool use_peephole, in HybridUnidirectionalLSTMOpModel() argument
295 use_cifg, use_peephole, use_projection_weights, use_projection_bias, in HybridUnidirectionalLSTMOpModel()
2471 bool time_major, bool use_cifg, bool use_peephole, in LayerNormUnidirectionalLSTMOpModel() argument
2477 use_cifg, use_peephole, use_projection_weights, use_projection_bias, in LayerNormUnidirectionalLSTMOpModel()
[all …]
|
| D | unidirectional_sequence_lstm.cc | 151 const bool use_cifg = (input_to_input_weights == nullptr); in PopulateQuantizedLstmParams8x8_16() local
220 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
226 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
234 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
261 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
293 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
364 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
517 const bool use_cifg = (input_to_input_weights == nullptr); in CheckInputTensorDimensions() local
519 ((cell_to_input_weights != nullptr || use_cifg) && in CheckInputTensorDimensions()
530 if (use_cifg) { in CheckInputTensorDimensions()
[all …]
|
| D | lstm.cc | 236 const bool use_cifg = (input_to_input_weights == nullptr); in PopulateQuantizedLstmParams8x8_16() local
305 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
311 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
319 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
346 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
378 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
449 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_16()
540 const bool use_cifg = (input_to_input_weights == nullptr); in PopulateQuantizedLstmParams8x8_8() local
610 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_8()
619 if (!use_cifg) { in PopulateQuantizedLstmParams8x8_8()
[all …]
|
| D | bidirectional_sequence_lstm_test.cc | 33 int sequence_length, bool use_cifg, in BidirectionalLSTMOpModel() argument
52 if (use_cifg) { in BidirectionalLSTMOpModel()
62 if (use_cifg) { in BidirectionalLSTMOpModel()
73 if (use_cifg) { in BidirectionalLSTMOpModel()
86 if (use_cifg) { in BidirectionalLSTMOpModel()
107 if (use_cifg) { in BidirectionalLSTMOpModel()
117 if (use_cifg) { in BidirectionalLSTMOpModel()
128 if (use_cifg) { in BidirectionalLSTMOpModel()
141 if (use_cifg) { in BidirectionalLSTMOpModel()
|
| D | bidirectional_sequence_lstm.cc | 309 const bool use_cifg = (input_to_input_weights == nullptr); in CheckLstmTensorDimensionsAndTypes() local
311 ((cell_to_input_weights != nullptr || use_cifg) && in CheckLstmTensorDimensionsAndTypes()
322 if (use_cifg) { in CheckLstmTensorDimensionsAndTypes()
|
| /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
79 if (!use_cifg) { in LstmStepWithAuxInput()
86 if (!use_cifg) { in LstmStepWithAuxInput()
99 if (!use_cifg) { in LstmStepWithAuxInput()
117 if (!use_cifg) { in LstmStepWithAuxInput()
131 if (!use_cifg) { in LstmStepWithAuxInput()
149 if (!use_cifg) { in LstmStepWithAuxInput()
165 if (!use_cifg) { in LstmStepWithAuxInput()
197 if (!use_cifg) { in LstmStepWithAuxInput()
254 if (use_cifg) { in LstmStepWithAuxInput()
[all …]
|
| /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() argument
67 if (use_cifg) { in LSTMOpModel()
76 if (use_cifg) { in LSTMOpModel()
89 if (use_cifg) { in LSTMOpModel()
102 if (use_cifg) { in LSTMOpModel()
130 if (use_cifg) { in LSTMOpModel()
|
| /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() argument
106 if (use_cifg) { in UpdateLstmCellFloat()
189 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepCalibration() local
203 if (!use_cifg) { in LstmStepCalibration()
237 forget_gate_scratch, cell_gate_scratch, use_cifg, in LstmStepCalibration()
311 const bool use_cifg = (input_to_input_weights == nullptr); in EvalCalibration() local
319 if (use_cifg) { in EvalCalibration()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_quantize_helper.h | 109 bool use_cifg = in GetLstmProperty() local
111 if (use_cifg) { in GetLstmProperty()
|
| /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() argument
3104 if (use_cifg) { in LSTMOpModel()
3114 if (use_cifg) { in LSTMOpModel()
3125 if (use_cifg) { in LSTMOpModel()
3138 if (use_cifg) { in LSTMOpModel()
3171 if (use_cifg) { in LSTMOpModel()
|