Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A 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 …]
H A 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
H A 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 …]
H A 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 …]
H A 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
H A 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
H A 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
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A 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 …]
H A 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 …]
H A 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 …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A 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 …]
H A 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 …]
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DQuantUtils.cpp12 int n_batch, int n_input, int16_t* output) { in ApplyLayerNorm()
86 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { in ApplySigmoid()
99 void CwiseMul(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input, int shift, in CwiseMul()
113 int32_t n_batch, int32_t n_input, int32_t output_zp, int8_t* output) { in CwiseMul()
138 void CwiseAdd(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input, in CwiseAdd()
150 void CwiseClipping(int16_t* input, const int16_t clipping_value, int32_t n_batch, int32_t n_input) { in CwiseClipping()
164 void CwiseClipping(int8_t* input, const int8_t clipping_value, int32_t n_batch, int32_t n_input) { in CwiseClipping()
DQuantUtils.h36 int32_t shift, int32_t n_batch, int32_t n_input, in MatrixBatchVectorMultiplyAccumulate()
154 void ApplyTanh(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { in ApplyTanh()
167 inline void ApplyTanh(int32_t integer_bits, const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanh()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
H A 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
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTMTest.cpp78 LSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LSTMOpModel()
274 const int n_input = 2; in TEST() local
378 const int n_input = 2; in TEST() local
475 const int n_input = 5; in TEST() local
DLayerNormLSTMTest.cpp87 LayerNormLSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LayerNormLSTMOpModel()
288 const int n_input = 5; in TEST() local
DLSTM.cpp150 const RunTimeOperandInfo* output_layer_norm_weights, uint32_t n_input, uint32_t n_output, in CheckInputTensorDimensions()
349 const uint32_t n_input = SizeOfDimension(input_, 1); in Prepare() local
789 const uint32_t n_input = input_shape.dimensions[1]; in LSTMStep() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/microfrontend/
H A 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
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dquantized_lstm.mod.py21 n_input = 2 variable
108 n_input = 2 variable
Dlayer_norm_lstm.mod.py23 n_input = 5 variable
195 n_input = 5 variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A 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
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A 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

1234