Home
last modified time | relevance | path

Searched defs:n_input (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc35 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/
Dunidirectional_sequence_lstm_test.cc36 int n_batch, int n_input, int n_cell, int n_output, int sequence_length, in HybridUnidirectionalLSTMOpModel()
249 const int n_input = 2; in TEST_F() local
311 const int n_input = 2; in TEST_F() local
378 const int n_input = 2; in TEST_P() local
441 const int n_input = 2; in TEST_P() local
552 const int n_input = 2; in TEST_F() local
614 const int n_input = 2; in TEST_P() local
676 const int n_input = 2; in TEST_P() local
1339 const int n_input = 5; in TEST_F() local
1406 const int n_input = 5; in TEST_P() local
[all …]
Dbidirectional_sequence_lstm_test.cc32 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
Dunidirectional_sequence_gru_test.cc34 explicit GRUOpModel(int n_batch, int n_input, int n_output, in GRUOpModel()
102 const int n_input = 2; in TEST() local
Dlstm_test.cc39 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 …]
Doptional_tensor_test.cc31 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
Dlstm_eval.cc80 int n_input, int n_aux_input, int n_output, in ComputeRowSums()
187 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateFloat()
363 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateHybrid()
548 const int n_batch, const int n_input, const int n_output, const int n_cell, in CalculateLstmGateInteger8x8_16()
713 const int n_batch, const int n_input, const int n_output, const int n_cell, in CalculateLstmGateInteger8x8_8()
874 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepFloat()
1033 inline void LstmStepHybrid( in LstmStepHybrid()
1372 inline void LstmStepInteger8x8_16( in LstmStepInteger8x8_16()
1627 inline void LstmStepInteger8x8_8( in LstmStepInteger8x8_8()
1782 const int n_input = input->dims->data[input->dims->size - 1]; in EvalFloat() local
[all …]
Dunidirectional_sequence_gru.cc41 const int n_input = input->dims->data[2]; in GruImpl() local
143 const int n_input = input->dims->data[2]; in Prepare() local
Dgru_cell.cc46 const int n_input = input_shape.Dims(1); in GruCell() local
Dbidirectional_sequence_lstm.cc176 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
Dunidirectional_sequence_lstm.cc418 TfLiteNode* node, int n_input, in CheckInputTensorDimensions()
896 const int n_input = input->dims->data[2]; in Prepare() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_tensor_utils.h108 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
118 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
129 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
160 int n_batch, int n_input, int16_t* output) { in ApplyLayerNorm()
169 const int32_t* bias, int n_batch, int n_input, in ApplyLayerNormFloat()
176 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoid()
181 void ApplySigmoidFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoidFloat()
187 int32_t n_input, int16_t* output) { in ApplyTanh()
191 void ApplyTanhFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanhFloat()
197 int n_input, int shift, int16_t* output) { in CwiseMul()
[all …]
Dsse_tensor_utils.h122 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
132 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
143 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
173 int n_batch, int n_input, int16_t* output) { in ApplyLayerNorm()
182 const int32_t* bias, int n_batch, int n_input, in ApplyLayerNormFloat()
189 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoid()
194 void ApplySigmoidFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoidFloat()
200 int32_t n_input, int16_t* output) { in ApplyTanh()
204 void ApplyTanhFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanhFloat()
210 int n_input, int shift, int16_t* output) { in CwiseMul()
[all …]
Dneon_tensor_utils.cc769 int32_t n_batch, int32_t n_input, in NeonMatrixBatchVectorMultiplyImpl()
1007 int32_t n_input, int32_t n_output, int32_t output_zp, in NeonCpuBackendGemm()
1040 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyAccumulate()
1056 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyAccumulate()
1517 int n_batch, int n_input, int16_t* output) { in NeonApplyLayerNorm()
1646 void NeonApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, in NeonApplySigmoid()
1686 void NeonApplyTanhImpl(const int16_t* input, int32_t n_batch, int32_t n_input, in NeonApplyTanhImpl()
1725 int32_t n_input, int16_t* output) { in NeonApplyTanh()
1746 int n_input, int shift, int16_t* output) { in NeonCwiseMul()
1782 int32_t multiplier, int shift, int n_batch, int n_input, in NeonCwiseMul()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h141 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
151 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
168 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
192 int n_batch, int n_input, int16_t* output) { in ApplyLayerNorm()
201 const int32_t* bias, int n_batch, int n_input, in ApplyLayerNormFloat()
208 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoid()
213 void ApplySigmoidFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplySigmoidFloat()
219 int32_t n_input, int16_t* output) { in ApplyTanh()
223 void ApplyTanhFloat(const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanhFloat()
229 int n_input, int shift, int16_t* output) { in CwiseMul()
[all …]
Dportable_tensor_utils.cc331 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulateImpl()
360 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulate()
370 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulate()
382 int32_t n_batch, int32_t n_input, in PortableMatrixBatchVectorMultiply()
449 int n_batch, int n_input, int16_t* output) { in PortableApplyLayerNorm()
499 const int32_t* bias, int n_batch, int n_input, in PortableApplyLayerNormFloat()
553 int32_t n_input, int16_t* output) { in PortableApplySigmoid()
567 int32_t n_input, int16_t* output) { in PortableApplySigmoidFloat()
587 int32_t n_input, int16_t* output) { in PortableApplyTanhImpl()
601 int32_t n_batch, int32_t n_input, int16_t* output) { in PortableApplyTanh()
[all …]
Dsvdf.h78 const int n_input = input_shape.Dims(1); in EvalIntegerSVDF() local
/external/tensorflow/tensorflow/lite/experimental/microfrontend/
Daudio_microfrontend_test.cc38 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/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc46 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateFloat()
181 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepCalibration()
302 const int n_input = input->dims->data[input->dims->size - 1]; in EvalCalibration() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.cc200 int n_input = 0; in GetNumberOfEngineInputs() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc63 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepWithAuxInput()
379 const int n_input = input->dims->data[input->dims->size - 1]; in EvalFloat() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_resource_ops_test.cc164 const int n_input = param_.n_inputs; in CreateTRTEngine() local
/external/pytorch/aten/src/ATen/native/
DNormalization.cpp206 int64_t n_input = input.size(1); in batch_norm_cpu_update_stats_template() local
287 int64_t n_input = input.size(1); in batch_norm_cpu_update_stats_template() local
345 int64_t n_input = input.size(1); in batch_norm_backward_cpu_template() local
/external/pytorch/aten/src/ATen/native/cuda/
DNormalization.cuh665 int64_t n_input = input_.size(1); in batch_norm_stats_cuda_template() local
698 int64_t n_input = input_.size(1); in batch_norm_elemt_cuda_template() local
783 int64_t n_input = input_.size(1); in batch_norm_backward_reduce_cuda_template() local
843 int64_t n_input = input_.size(1); in batch_norm_backward_elemt_cuda_template() local
894 int64_t n_input = input_.size(1); in batch_norm_backward_elemt_cuda_template() local
DNormalization.cu463 int64_t n_input = self.size(1); in batch_norm_cuda() local
802 const int64_t n_input = self.size(1); in batch_norm_update_stats_cuda() local

12