| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | neon_tensor_utils.h | 29 int n_batch, float* result, in MatrixBatchVectorMultiplyAccumulate()
38 int n_batch, float* __restrict__ result, int result_stride) { in MatrixBatchVectorMultiplyAccumulate()
45 const int m_cols, const float* vector, int n_batch, float* result, in SparseMatrixBatchVectorMultiplyAccumulate()
54 const float* scaling_factors, int n_batch, float* __restrict__ result, in SparseMatrixBatchVectorMultiplyAccumulate()
74 const float* batch_vector, int n_batch, in VectorBatchVectorCwiseProduct()
82 int n_batch, float* result) { in VectorBatchVectorCwiseProductAccumulate()
94 int n_batch, float* result, in BatchVectorBatchVectorDotProduct()
100 void VectorBatchVectorAdd(const float* vector, int v_size, int n_batch, in VectorBatchVectorAdd()
105 void VectorBatchVectorAssign(const float* vector, int v_size, int n_batch, in VectorBatchVectorAssign()
165 int v_size, int n_batch, in MeanStddevNormalization()
|
| D | neon_tensor_utils.cc | 99 int n_batch, float* result, in NeonMatrixBatchVectorMultiplyAccumulate()
150 const int8_t* ShuffleVectors(const int8_t* vectors, const int n_batch, in ShuffleVectors()
198 const int8_t* vectors, const float* scaling_factors, int n_batch, in DotprodMatrixBatchFourVectorMultiplyAccumulate()
316 const float* scaling_factors, int n_batch, float* __restrict__ result, in DotprodSparseMatrixBatchVectorMultiplyAccumulate()
382 int n_batch, float* __restrict__ result, int result_stride) { in NeonMatrixBatchVectorMultiplyAccumulate()
512 const float* vector, int n_batch, float* result, int result_stride) { in NeonSparseMatrixBatchVectorMultiplyAccumulate()
556 const float* scaling_factors, int n_batch, float* __restrict__ result, in NeonSparseMatrixBatchVectorMultiplyAccumulate()
676 const float* batch_vector, int n_batch, in NeonVectorBatchVectorCwiseProduct()
707 int n_batch, float* result) { in NeonVectorBatchVectorCwiseProductAccumulate()
986 int n_batch, float* result, in NeonBatchVectorBatchVectorDotProduct()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | portable_tensor_utils.h | 163 int n_batch, float* result, in MatrixBatchVectorMultiplyAccumulate()
172 int n_batch, float* __restrict__ result, int result_stride) { in MatrixBatchVectorMultiplyAccumulate()
180 const float* vector, int n_batch, float* result, int result_stride) { in SparseMatrixBatchVectorMultiplyAccumulate()
188 const float* scaling_factors, int n_batch, float* __restrict__ result, in SparseMatrixBatchVectorMultiplyAccumulate()
207 const float* batch_vector, int n_batch, in VectorBatchVectorCwiseProduct()
215 int n_batch, float* result) { in VectorBatchVectorCwiseProductAccumulate()
227 int n_batch, float* result, in BatchVectorBatchVectorDotProduct()
233 void VectorBatchVectorAdd(const float* vector, int v_size, int n_batch, in VectorBatchVectorAdd()
238 void VectorBatchVectorAssign(const float* vector, int v_size, int n_batch, in VectorBatchVectorAssign()
286 int v_size, int n_batch, in MeanStddevNormalization()
|
| D | portable_tensor_utils.cc | 71 int n_batch, float* result, in PortableMatrixBatchVectorMultiplyAccumulate()
91 int n_batch, float* __restrict__ result, int result_stride) { in PortableMatrixBatchVectorMultiplyAccumulate()
115 const float* vector, int n_batch, float* result, int result_stride) { in PortableSparseMatrixBatchVectorMultiplyAccumulate()
146 const float* scaling_factors, int n_batch, float* __restrict__ result, in PortableSparseMatrixBatchVectorMultiplyAccumulate()
199 int n_batch, float* result, in PortableBatchVectorBatchVectorDotProduct()
223 int n_batch, float* result) { in PortableVectorBatchVectorCwiseProduct()
234 int n_batch, in PortableVectorBatchVectorCwiseProductAccumulate()
243 void PortableVectorBatchVectorAdd(const float* vector, int v_size, int n_batch, in PortableVectorBatchVectorAdd()
254 int n_batch, float* batch_vector) { in PortableVectorBatchVectorAssign()
325 int n_batch, float normalization_epsilon) { in PortableMeanStddevNormalization()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_test.cc | 37 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
271 HybridLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in HybridLSTMOpModel()
450 const int n_batch = 1; in TEST_F() local
507 const int n_batch = 1; in TEST_F() local
566 const int n_batch = 1; in TEST_F() local
671 const int n_batch = 1; in TEST_F() local
728 const int n_batch = 1; in TEST_F() local
787 const int n_batch = 1; in TEST_F() local
1444 const int n_batch = 2; in TEST_F() local
1505 const int n_batch = 2; in TEST_F() local
[all …]
|
| D | unidirectional_sequence_lstm_test.cc | 34 UnidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in UnidirectionalLSTMOpModel()
243 int n_batch, int n_input, int n_cell, int n_output, int sequence_length, in HybridUnidirectionalLSTMOpModel()
451 const int n_batch = 1; in TEST_F() local
513 const int n_batch = 1; in TEST_F() local
580 const int n_batch = 1; in TEST_F() local
643 const int n_batch = 1; in TEST_F() local
752 const int n_batch = 1; in TEST_F() local
814 const int n_batch = 1; in TEST_F() local
876 const int n_batch = 1; in TEST_F() local
1537 const int n_batch = 2; in TEST_F() local
[all …]
|
| D | bidirectional_sequence_lstm_test.cc | 37 BidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in BidirectionalLSTMOpModel()
406 const int n_batch = 1; in TEST_P() local
571 const int n_batch = 2; in TEST_P() local
736 const int n_batch = 1; in TEST() local
899 const int n_batch = 1; in TEST() local
1052 const int n_batch = 1; in TEST() local
1206 const int n_batch = 2; in TEST() local
1911 const int n_batch = 2; in TEST() local
2612 const int n_batch = 1; in TEST_P() local
|
| D | optional_tensor_test.cc | 33 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
231 const int n_batch = 1; in TEST() local
|
| D | lstm_eval.cc | 116 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepWithAuxInput()
430 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
899 int max_time, n_batch; in EvalFloat() local
1104 int max_time, n_batch; in EvalHybrid() local
|
| D | unidirectional_sequence_lstm.cc | 267 const int n_batch = time_major ? input->dims->data[1] : input->dims->data[0]; in Prepare() local
|
| D | lstm.cc | 341 const int n_batch = input->dims->data[0]; in Prepare() local
|
| D | bidirectional_sequence_lstm.cc | 403 const int n_batch = time_major ? input->dims->data[1] : input->dims->data[0]; in Prepare() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2259 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
2568 const int n_batch = 1; in TEST_F() local
2672 const int n_batch = 1; in TEST_F() local
3328 const int n_batch = 2; in TEST_F() local
|
| /external/deqp/external/openglcts/modules/gl/ |
| D | gl4cVertexAttrib64BitTest.cpp | 3554 for (unsigned int n_batch = 0; n_batch < m_n_batches; ++n_batch) in initBuffers() local
3883 for (unsigned int n_batch = 0; n_batch < m_n_batches; ++n_batch) in verifyXFBData() local
|