Home
last modified time | relevance | path

Searched refs:MatrixBatchVectorMultiplyAccumulate (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.cc54 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 …]
Dtensor_utils.h42 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(
Dtensor_utils_test.cc357 MatrixBatchVectorMultiplyAccumulate(matrix, kRow, kCol, vector, kBatch, in TEST()
401 MatrixBatchVectorMultiplyAccumulate( in TEST()
461 MatrixBatchVectorMultiplyAccumulate( in TEST()
477 MatrixBatchVectorMultiplyAccumulate( in TEST()
503 MatrixBatchVectorMultiplyAccumulate( in TEST()
556 MatrixBatchVectorMultiplyAccumulate( in TEST()
1147 MatrixBatchVectorMultiplyAccumulate( in TestDotprodMatrixBatchVectorMultiply()
1171 MatrixBatchVectorMultiplyAccumulate( in TestPerChannelDotprodMatrixBatchVectorMultiply()
1447 MatrixBatchVectorMultiplyAccumulate(a_int8_data, a_rows, a_cols, b_int8_data, in TEST()
1463 MatrixBatchVectorMultiplyAccumulate( in TEST()
[all …]
Dportable_tensor_utils.h178 void MatrixBatchVectorMultiplyAccumulate(const float* matrix, int m_rows,
213 void MatrixBatchVectorMultiplyAccumulate(
222 void MatrixBatchVectorMultiplyAccumulate(
/external/armnn/src/backends/reference/workloads/
DLstm.cpp103 MatrixBatchVectorMultiplyAccumulate(*inputToInputWeightsTensor, in LstmImpl()
106 MatrixBatchVectorMultiplyAccumulate(*inputToForgetWeightsTensor, in LstmImpl()
108 MatrixBatchVectorMultiplyAccumulate(*inputToCellWeightsTensor, in LstmImpl()
110 MatrixBatchVectorMultiplyAccumulate(*inputToOutputWeightsTensor, in LstmImpl()
116 MatrixBatchVectorMultiplyAccumulate(*recurrentToInputWeightsTensor, in LstmImpl()
119 MatrixBatchVectorMultiplyAccumulate(*recurrentToForgetWeightsTensor, in LstmImpl()
121 MatrixBatchVectorMultiplyAccumulate(*recurrentToCellWeightsTensor, in LstmImpl()
123 MatrixBatchVectorMultiplyAccumulate(*recurrentToOutputWeightsTensor, in LstmImpl()
243 MatrixBatchVectorMultiplyAccumulate(*projectionWeightsTensor, in LstmImpl()
DRefQLstmWorkload.cpp297 MatrixBatchVectorMultiplyAccumulate(*inputToInputWeightsDecoder, in Execute()
301 MatrixBatchVectorMultiplyAccumulate(*inputToForgetWeightsDecoder, in Execute()
304 MatrixBatchVectorMultiplyAccumulate(*inputToCellWeightsDecoder, in Execute()
307 MatrixBatchVectorMultiplyAccumulate(*inputToOutputWeightsDecoder, in Execute()
313 MatrixBatchVectorMultiplyAccumulate(*recurrentToInputWeightsDecoder, in Execute()
317 MatrixBatchVectorMultiplyAccumulate(*recurrentToForgetWeightsDecoder, in Execute()
320 MatrixBatchVectorMultiplyAccumulate(*recurrentToCellWeightsDecoder, in Execute()
323 MatrixBatchVectorMultiplyAccumulate(*recurrentToOutputWeightsDecoder, in Execute()
527 …MatrixBatchVectorMultiplyAccumulate(*projectionWeightsDecoder, outputSize, numUnits, *hiddenStateD… in Execute()
DLstmUtils.hpp32 void MatrixBatchVectorMultiplyAccumulate(armnn::Decoder<float>& matrix,
DLstmUtils.cpp87 void MatrixBatchVectorMultiplyAccumulate(armnn::Decoder<float>& matrix, in MatrixBatchVectorMultiplyAccumulate() function
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_tensor_utils.h27 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()
105 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function
110 NEON_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, input, bias, in MatrixBatchVectorMultiplyAccumulate()
[all …]
Dsse_tensor_utils.h37 void MatrixBatchVectorMultiplyAccumulate(const float* matrix, int m_rows, in MatrixBatchVectorMultiplyAccumulate() function
44 NEON_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate()
49 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function
54 SSE_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate()
58 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function
64 SSE_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate()
69 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function
75 SSE_OR_PORTABLE(MatrixBatchVectorMultiplyAccumulate, matrix, m_rows, m_cols, in MatrixBatchVectorMultiplyAccumulate()
119 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function
129 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function
Doptimized_ops.h1085 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in HybridConv()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc100 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()
135 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput()
140 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput()
143 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput()
146 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput()
153 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput()
157 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in LstmStepWithAuxInput()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h58 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
138 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function
148 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function
Dsvdf.h176 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in EvalFloatSVDF()
225 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in EvalHybridSVDF()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc65 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateFloat()
71 tensor_utils::MatrixBatchVectorMultiplyAccumulate(aux_input_to_gate_weights, in CalculateLstmGateFloat()
76 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateFloat()
148 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmOutputCalibration()
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc45 void MatrixBatchVectorMultiplyAccumulate( in MatrixBatchVectorMultiplyAccumulate() function
206 MatrixBatchVectorMultiplyAccumulate(input_to_gate_weights, input, in CalculateLstmGateFloat()
214 MatrixBatchVectorMultiplyAccumulate(aux_input_to_gate_weights, aux_input, in CalculateLstmGateFloat()
220 MatrixBatchVectorMultiplyAccumulate(recurrent_to_gate_weights, output_state, in CalculateLstmGateFloat()
325 MatrixBatchVectorMultiplyAccumulate(projection_weights, scratch, in CalculateLstmOutputFloat()
400 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateHybrid()
410 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateHybrid()
428 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateHybrid()
512 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmOutputHybrid()
563 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in CalculateLstmGateInteger8x8_16()
[all …]
Dfully_connected.cc499 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in EvalPie()
566 tensor_utils::MatrixBatchVectorMultiplyAccumulate( in EvalHybridDense()