/external/tensorflow/tensorflow/lite/kernels/internal/ |
D | kernel_utils.cc | 54 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in RnnBatchStep() 60 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in RnnBatchStep() 66 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in RnnBatchStep() 84 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in RnnBatchStep() 93 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in RnnBatchStep() 102 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in RnnBatchStep() 199 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in RnnBatchStep() 218 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in RnnBatchStep() 239 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in RnnBatchStep() 272 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in RnnBatchStep() [all …]
|
D | tensor_utils.h | 42 void MatrixBatchVectorMultiplyAccumulate( 50 void MatrixBatchVectorMultiplyAccumulate( 60 inline void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function 72 MatrixBatchVectorMultiplyAccumulate(matrix, m_rows, m_cols, vectors, in MatrixBatchVectorMultiplyAccumulate() 104 void MatrixBatchVectorMultiplyAccumulate( 135 void MatrixBatchVectorMultiplyAccumulate(
|
D | tensor_utils_test.cc | 355 MatrixBatchVectorMultiplyAccumulate(matrix, kRow, kCol, vector, kBatch, in TEST() 399 MatrixBatchVectorMultiplyAccumulate( in TEST() 459 MatrixBatchVectorMultiplyAccumulate( in TEST() 475 MatrixBatchVectorMultiplyAccumulate( in TEST() 501 MatrixBatchVectorMultiplyAccumulate( in TEST() 554 MatrixBatchVectorMultiplyAccumulate( in TEST() 1145 MatrixBatchVectorMultiplyAccumulate( in TestDotprodMatrixBatchVectorMultiply() 1169 MatrixBatchVectorMultiplyAccumulate( in TestPerChannelDotprodMatrixBatchVectorMultiply() 1451 MatrixBatchVectorMultiplyAccumulate(a_int8_data, a_rows, a_cols, b_int8_data, in TEST() 1467 MatrixBatchVectorMultiplyAccumulate( in TEST() [all …]
|
D | tensor_utils_common.h | 88 void MatrixBatchVectorMultiplyAccumulate(const float* matrix, int m_rows, 123 void MatrixBatchVectorMultiplyAccumulate( 132 void MatrixBatchVectorMultiplyAccumulate(
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | neon_tensor_utils.h | 27 void MatrixBatchVectorMultiplyAccumulate(const float* matrix, int m_rows, in MatrixBatchVectorMultiplyAccumulate() function 30 NEON_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate() 34 void MatrixBatchVectorMultiplyAccumulate(const int8_t* __restrict__ matrix, in MatrixBatchVectorMultiplyAccumulate() function 40 NEON_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate() 44 void MatrixBatchVectorMultiplyAccumulate(const int8_t* __restrict__ matrix, in MatrixBatchVectorMultiplyAccumulate() function 51 NEON_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate() 55 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function 61 NEON_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate() 90 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function 95 NEON_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, input, bias, in MatrixBatchVectorMultiplyAccumulate() [all …]
|
D | sse_tensor_utils.h | 37 void MatrixBatchVectorMultiplyAccumulate(const float* matrix, int m_rows, in MatrixBatchVectorMultiplyAccumulate() function 40 NEON_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate() 44 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function 49 SSE_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate() 53 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function 59 SSE_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate() 64 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function 70 SSE_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate() 99 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function 109 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function
|
D | optimized_ops.h | 1416 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in HybridConv()
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
D | lstm.cc | 100 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput() 105 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput() 108 tensor_utils::MatrixBatchVectorMultiplyAccumulate(input_to_cell_weights_ptr, in LstmStepWithAuxInput() 111 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput() 132 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput() 137 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput() 140 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput() 143 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput() 150 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput() 154 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput() [all …]
|
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | portable_tensor_utils.h | 58 void MatrixBatchVectorMultiplyAccumulate(const float* matrix, int m_rows, in MatrixBatchVectorMultiplyAccumulate() function 65 void MatrixBatchVectorMultiplyAccumulate(const int8_t* __restrict__ matrix, in MatrixBatchVectorMultiplyAccumulate() function 75 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function 87 void MatrixBatchVectorMultiplyAccumulate(const int8_t* __restrict__ matrix, in MatrixBatchVectorMultiplyAccumulate() function 123 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function 133 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function
|
D | svdf.h | 176 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in EvalFloatSVDF() 225 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in EvalHybridSVDF()
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
D | lstm.cc | 65 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateFloat() 71 tensor_utils::MatrixBatchVectorMultiplyAccumulate(aux_input_to_gate_weights, in CalculateLstmGateFloat() 76 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateFloat() 147 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmOutputCalibration()
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_eval.cc | 173 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateFloat() 179 tensor_utils::MatrixBatchVectorMultiplyAccumulate(aux_input_to_gate_weights, in CalculateLstmGateFloat() 184 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateFloat() 284 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmOutputFloat() 358 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateHybrid() 368 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateHybrid() 386 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateHybrid() 470 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmOutputHybrid() 521 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateInteger8x8_16() 528 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateInteger8x8_16() [all …]
|
D | fully_connected.cc | 422 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in EvalPie() 512 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in EvalHybridImpl()
|