| /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()
269 const int n_output = 4; in TEST_P() local
343 const int n_output = 4; in TEST_P() local
410 const int n_output = 16; in TEST_P() local
973 const int n_output = 3; in TEST_F() local
1059 const int n_output = 3; in TEST_F() local
1123 const int n_output = 4; in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm_test.cc | 36 int n_batch, int n_input, int n_cell, int n_output, int sequence_length, in HybridUnidirectionalLSTMOpModel()
252 const int n_output = 4; in TEST_F() local
314 const int n_output = 4; in TEST_F() local
381 const int n_output = 4; in TEST_P() local
444 const int n_output = 4; in TEST_P() local
555 const int n_output = 4; in TEST_F() local
617 const int n_output = 4; in TEST_P() local
679 const int n_output = 4; in TEST_P() local
1341 const int n_output = 16; in TEST_F() local
1408 const int n_output = 16; in TEST_P() local
[all …]
|
| D | bidirectional_sequence_lstm_test.cc | 32 BidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in BidirectionalLSTMOpModel()
434 const int n_output = 4; in TEST_P() local
600 const int n_output = 4; in TEST_P() local
766 const int n_output = 4; in TEST() local
927 const int n_output = 4; in TEST() local
1078 const int n_output = 4; in TEST() local
1229 const int n_output = 16; in TEST() local
1932 const int n_output = 16; in TEST() local
2634 const int n_output = 4; in TEST_P() local
2808 const int n_output = 4; in TEST_P() local
|
| D | unidirectional_sequence_gru_test.cc | 34 explicit GRUOpModel(int n_batch, int n_input, int n_output, in GRUOpModel()
103 const int n_output = 3; in TEST() local
|
| D | lstm_test.cc | 39 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
417 const int n_output = 4; in TEST_P() local
494 const int n_output = 4; in TEST_P() local
564 const int n_output = 16; in TEST_P() local
1147 const int n_output = 3; in TEST_P() local
1242 const int n_output = 3; in TEST_P() local
1314 LSTMIntegerOpModel(int n_batch, int n_input, int n_cell, int n_output, in LSTMIntegerOpModel()
1601 const int n_output = 3; in TEST() local
1763 const int n_output = 3; in TEST() local
1936 const int n_output = 3; in TEST() local
[all …]
|
| D | lstm_eval.cc | 80 int n_input, int n_aux_input, int n_output, in ComputeRowSums()
188 const int n_output, const int n_cell, in CalculateLstmGateFloat()
302 void CalculateLstmOutputFloat(int n_batch, int n_cell, int n_output, in CalculateLstmOutputFloat()
340 void CalculateLstmGateHybrid( in CalculateLstmGateHybrid()
475 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputHybrid()
548 const int n_batch, const int n_input, const int n_output, const int n_cell, in CalculateLstmGateInteger8x8_16()
659 int n_batch, int n_cell, int n_output, const int16_t* cell_state, in CalculateLstmOutputInteger8x8_16()
713 const int n_batch, const int n_input, const int n_output, const int n_cell, in CalculateLstmGateInteger8x8_8()
767 int n_batch, int n_cell, int n_output, const int16_t* cell_state, in CalculateLstmOutputInteger8x8_8()
851 inline void LstmStepFloat( in LstmStepFloat()
[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()
233 const int n_output = 4; in TEST() local
|
| D | unidirectional_sequence_gru.cc | 42 const int n_output = output->dims->data[2]; in GruImpl() local
151 const int n_output = input_state->dims->data[1]; in Prepare() local
|
| D | gru_cell.cc | 47 const int n_output = state_shape.Dims(1); in GruCell() local
|
| D | unidirectional_sequence_lstm.cc | 419 int n_output, int n_cell, in CheckInputTensorDimensions()
915 const int n_output = recurrent_to_output_weights->dims->data[1]; in Prepare() local
|
| D | bidirectional_sequence_lstm.cc | 176 TfLiteContext* context, TfLiteNode* node, int n_input, int n_output, in CheckLstmTensorDimensionsAndTypes()
385 int n_output, int n_cell) { in CheckInputTensorDimensions()
|
| D | lstm.cc | 855 int n_output, int n_cell, in CheckInputTensorDimensions()
1340 const int n_output = recurrent_to_output_weights->dims->data[1]; in Prepare() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 47 const int n_output, const int n_cell, in CalculateLstmGateFloat()
124 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputCalibration()
158 inline void LstmStepCalibration( in LstmStepCalibration()
308 const int n_output = recurrent_to_output_weights->dims->data[1]; in EvalCalibration() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | sse_tensor_utils.h | 122 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()
156 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
|
| D | neon_tensor_utils.h | 108 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()
142 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
|
| D | neon_tensor_utils.cc | 770 int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyImpl()
879 int32_t multiplier, int32_t shift, int32_t n_batch, int32_t n_output, in NeonMatrixBatchVectorAccumulateImpl()
932 int32_t multiplier, int32_t shift, int32_t n_batch, int32_t n_output, in NeonMatrixBatchVectorAccumulateImpl()
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()
|
| D | sse_tensor_utils.cc | 380 int32_t n_input, int32_t n_output, int32_t output_zp, in SseCpuBackendGemm()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | portable_tensor_utils.h | 141 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()
181 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
|
| D | portable_tensor_utils.cc | 331 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()
413 int32_t n_output, int32_t output_zp, int8_t* proj_output) { in PortableMatrixBatchVectorMultiply()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
385 const int n_output = recurrent_to_output_weights->dims->data[1]; in EvalFloat() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3169 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
3536 const int n_output = 4; in TEST_F() local
3600 const int n_output = 4; in TEST_F() local
3712 const int n_output = 4; in TEST_F() local
4370 const int n_output = 16; in TEST_F() local
4503 const int n_output = 3; in TEST_F() local
4644 const int n_output = 3; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 688 const int n_output = recurrent_kernel_type.getDimSize(0); in ConvertKerasLSTMLayer() local
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadData.cpp | 2016 const uint32_t n_output = m_RecurrentToOutputWeights->GetShape()[1]; in Validate() local
3882 const uint32_t n_output = m_RecurrentToOutputWeights->GetShape()[1]; in Validate() local
|
| /external/deqp/external/openglcts/modules/glesext/geometry_shader/ |
| D | esextcGeometryShaderRendering.cpp | 425 unsigned int n_output = 0; in init() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2522 const int n_output = recurrent_to_output_weights.getDimSize(1); in verify() local
2645 int n_output = output_state_type.getDimSize(1); in inferReturnTypes() local
|