| /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()
266 const int n_input = 2; in TEST_P() local
340 const int n_input = 2; in TEST_P() local
408 const int n_input = 5; in TEST_P() local
971 const int n_input = 5; in TEST_F() local
1057 const int n_input = 5; in TEST_F() local
1121 const int n_input = 2; in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | portable_tensor_utils.h | 126 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
136 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
153 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
177 int n_batch, int n_input, int16_t* output) { in ApplyLayerNorm()
186 const int32_t* bias, int n_batch, int n_input, in ApplyLayerNormFloat()
193 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoid()
198 void ApplySigmoidFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoidFloat()
204 int32_t n_input, int16_t* output) { in ApplyTanh()
208 void ApplyTanhFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanhFloat()
214 int n_input, int shift, int16_t* output) { in CwiseMul()
[all …]
|
| D | portable_tensor_utils.cc | 296 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulateImpl()
325 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulate()
335 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulate()
347 int32_t n_batch, int32_t n_input, in PortableMatrixBatchVectorMultiply()
414 int n_batch, int n_input, int16_t* output) { in PortableApplyLayerNorm()
464 const int32_t* bias, int n_batch, int n_input, in PortableApplyLayerNormFloat()
519 int32_t n_input, int16_t* output) { in PortableApplySigmoid()
533 int32_t n_input, int16_t* output) { in PortableApplySigmoidFloat()
552 int32_t n_input, int16_t* output) { in PortableApplyTanhImpl()
566 int32_t n_batch, int32_t n_input, int16_t* output) { in PortableApplyTanh()
[all …]
|
| D | svdf.h | 78 const int n_input = input_shape.Dims(1); in EvalIntegerSVDF() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | sse_tensor_utils.h | 102 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
112 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
123 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
153 int n_batch, int n_input, int16_t* output) { in ApplyLayerNorm()
162 const int32_t* bias, int n_batch, int n_input, in ApplyLayerNormFloat()
169 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoid()
174 void ApplySigmoidFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoidFloat()
180 int32_t n_input, int16_t* output) { in ApplyTanh()
184 void ApplyTanhFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanhFloat()
190 int n_input, int shift, int16_t* output) { in CwiseMul()
[all …]
|
| D | neon_tensor_utils.h | 93 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
103 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
114 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
145 int n_batch, int n_input, int16_t* output) { in ApplyLayerNorm()
154 const int32_t* bias, int n_batch, int n_input, in ApplyLayerNormFloat()
161 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoid()
166 void ApplySigmoidFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoidFloat()
172 int32_t n_input, int16_t* output) { in ApplyTanh()
176 void ApplyTanhFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanhFloat()
182 int n_input, int shift, int16_t* output) { in CwiseMul()
[all …]
|
| D | neon_tensor_utils.cc | 743 int32_t n_batch, int32_t n_input, in NeonMatrixBatchVectorMultiplyImpl()
981 int32_t n_input, int32_t n_output, int32_t output_zp, in NeonCpuBackendGemm()
1014 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyAccumulate()
1030 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyAccumulate()
1494 int n_batch, int n_input, int16_t* output) { in NeonApplyLayerNorm()
1622 void NeonApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, in NeonApplySigmoid()
1662 void NeonApplyTanhImpl(const int16_t* input, int32_t n_batch, int32_t n_input, in NeonApplyTanhImpl()
1701 int32_t n_input, int16_t* output) { in NeonApplyTanh()
1722 int n_input, int shift, int16_t* output) { in NeonCwiseMul()
1758 int32_t multiplier, int shift, int n_batch, int n_input, in NeonCwiseMul()
[all …]
|
| D | sse_tensor_utils.cc | 178 int32_t n_input, int32_t n_output, int32_t output_zp, in SseCpuBackendGemm()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm_test.cc | 32 UnidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in UnidirectionalLSTMOpModel()
288 int n_batch, int n_input, int n_cell, int n_output, int sequence_length, in HybridUnidirectionalLSTMOpModel()
501 const int n_input = 2; in TEST_F() local
563 const int n_input = 2; in TEST_F() local
630 const int n_input = 2; in TEST_P() local
693 const int n_input = 2; in TEST_P() local
804 const int n_input = 2; in TEST_F() local
866 const int n_input = 2; in TEST_P() local
928 const int n_input = 2; in TEST_P() local
1591 const int n_input = 5; in TEST_F() local
[all …]
|
| D | bidirectional_sequence_lstm_test.cc | 32 BidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in BidirectionalLSTMOpModel()
431 const int n_input = 2; in TEST_P() local
597 const int n_input = 2; in TEST_P() local
763 const int n_input = 2; in TEST() local
924 const int n_input = 2; in TEST() local
1075 const int n_input = 2; in TEST() local
1227 const int n_input = 5; in TEST() local
1930 const int n_input = 5; in TEST() local
2631 const int n_input = 2; in TEST_P() local
2805 const int n_input = 2; in TEST_P() local
|
| D | lstm_test.cc | 39 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
414 const int n_input = 2; in TEST_P() local
491 const int n_input = 2; in TEST_P() local
562 const int n_input = 5; in TEST_P() local
1145 const int n_input = 5; in TEST_P() local
1240 const int n_input = 5; in TEST_P() local
1314 LSTMIntegerOpModel(int n_batch, int n_input, int n_cell, int n_output, in LSTMIntegerOpModel()
1599 const int n_input = 5; in TEST() local
1761 const int n_input = 5; in TEST() local
1934 const int n_input = 5; in TEST() local
[all …]
|
| D | optional_tensor_test.cc | 31 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
230 const int n_input = 2; in TEST() local
|
| D | lstm_eval.cc | 49 int n_input, int n_aux_input, int n_output, in ComputeRowSums()
156 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateFloat()
321 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateHybrid()
506 const int n_batch, const int n_input, const int n_output, const int n_cell, in CalculateLstmGateInteger8x8_16()
671 const int n_batch, const int n_input, const int n_output, const int n_cell, in CalculateLstmGateInteger8x8_8()
831 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepFloat()
977 inline void LstmStepHybrid( in LstmStepHybrid()
1316 inline void LstmStepInteger8x8_16( in LstmStepInteger8x8_16()
1569 inline void LstmStepInteger8x8_8( in LstmStepInteger8x8_8()
1723 const int n_input = input->dims->data[input->dims->size - 1]; in EvalFloat() local
[all …]
|
| D | bidirectional_sequence_lstm.cc | 176 TfLiteContext* context, TfLiteNode* node, int n_input, int n_output, in CheckLstmTensorDimensionsAndTypes()
384 TfLiteNode* node, int n_input, in CheckInputTensorDimensions()
439 const int n_input = input->dims->data[2]; in Prepare() local
|
| D | unidirectional_sequence_lstm.cc | 410 TfLiteNode* node, int n_input, in CheckInputTensorDimensions()
886 const int n_input = input->dims->data[2]; in Prepare() local
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | unidirectional_sequence_gru_test.cc | 34 explicit GRUOpModel(int n_batch, int n_input, int n_output, in GRUOpModel()
102 const int n_input = 2; in TEST() local
|
| D | unidirectional_sequence_gru.cc | 39 const int n_input = input->dims->data[2]; in GruImpl() local
135 const int n_input = input->dims->data[2]; in Prepare() local
|
| D | gru_cell.cc | 46 const int n_input = input_shape.Dims(1); in GruCell() local
|
| /external/tensorflow/tensorflow/lite/experimental/microfrontend/ |
| D | audio_microfrontend_test.cc | 38 MicroFrontendOpModel(int n_input, int n_frame, int n_frequency_per_frame, in MicroFrontendOpModel()
179 const int n_input = 36; in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | svdf.cc | 64 const int n_input = input_tensor->dims->data[1]; in EvalIntegerSvdfHifimini() local
259 const int n_input = input_tensor->dims->data[1]; in EvalIntegerSvdfHifi4() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 46 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateFloat()
180 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepCalibration()
301 const int n_input = input->dims->data[input->dims->size - 1]; in EvalCalibration() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | svdf_common.cc | 43 const int n_input = input_tensor->dims->data[1]; in EvalIntegerSvdfReference() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 63 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepWithAuxInput()
370 const int n_input = input->dims->data[input->dims->size - 1]; in EvalFloat() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 243 int n_input = 0; in GetNumberOfEngineInputs() local
|
| /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()
3456 const int n_input = 2; in TEST_F() local
3520 const int n_input = 2; in TEST_F() local
3632 const int n_input = 2; in TEST_F() local
4291 const int n_input = 5; in TEST_F() local
4424 const int n_input = 5; in TEST_F() local
4565 const int n_input = 5; in TEST_F() local
|