| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn.cc | 183 TfLiteTensor* row_sums; in Prepare() local
238 TfLiteTensor* row_sums, bool* compute_row_sums) { in EvalHybrid()
325 TfLiteTensor* row_sums; in Eval() local
|
| D | unidirectional_sequence_rnn.cc | 190 TfLiteTensor* row_sums; in Prepare() local
274 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalHybrid()
400 TfLiteTensor* row_sums; in Eval() local
|
| D | fully_connected.cc | 403 TfLiteTensor* row_sums; in PrepareImpl() local
516 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalHybridDense()
585 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalSparseHybridImpl()
712 TfLiteTensor* row_sums; member
722 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalHybrid()
969 TfLiteTensor* row_sums; in EvalQuantized() local
|
| D | svdf.cc | 232 TfLiteTensor* row_sums; in Prepare() local
343 TfLiteTensor* row_sums; in Eval() local
|
| D | batch_matmul.cc | 277 TfLiteTensor* row_sums; in InitializeTemporaries() local
452 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalHybrid()
637 TfLiteTensor* row_sums; in EvalQuantized() local
|
| D | unidirectional_sequence_lstm.cc | 1150 TfLiteTensor* row_sums; in Prepare() local
1363 TfLiteTensor* row_sums; in Eval() local
|
| D | conv.cc | 628 TfLiteTensor* row_sums; in Prepare() local
1039 TfLiteTensor* row_sums; in EvalHybridPerChannel() local
|
| D | lstm.cc | 1611 TfLiteTensor* row_sums; in Prepare() local
1940 TfLiteTensor* row_sums; in Eval() local
|
| D | lstm_eval.cc | 79 int32_t* projection_weights_row_sums, int32_t* row_sums, int n_cell, in ComputeRowSums()
1033 inline void LstmStepHybrid( in LstmStepHybrid()
1932 TfLiteStatus EvalHybrid( in EvalHybrid()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | kernel_utils.cc | 128 int32_t* accum_scratch, int32_t* row_sums, bool* compute_row_sums) { in RnnBatchStep()
155 int32_t* accum_scratch, int32_t* row_sums, bool* compute_row_sums) { in RnnBatchStep()
|
| D | tensor_utils.h | 65 const int32_t* input_offset, int32_t* scratch, int32_t* row_sums, in MatrixBatchVectorMultiplyAccumulate()
|
| D | tensor_utils_test.cc | 459 int32_t* row_sums = scratch.data() + 8 * 4; in TEST() local
1168 std::vector<int32_t> row_sums(rows); in TestPerChannelDotprodMatrixBatchVectorMultiply() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | sse_tensor_utils.cc | 204 const int32_t* input_offset, const int32_t* row_sums) { in Avx2MatrixBatchVectorMultiplyAccumulateImpl()
296 const int32_t* input_offset, const int32_t* row_sums) { in SseMatrixBatchVectorMultiplyAccumulateImpl()
477 const int32_t* input_offset, int32_t* scratch, int32_t* row_sums, in SseMatrixBatchVectorMultiplyAccumulate()
|
| D | batch_matmul.h | 118 const int32_t* input_offset, int32_t* row_sums, in BatchMatMul()
|
| D | neon_tensor_utils.cc | 461 const int32_t* input_offset, int32_t* row_sums) { in DotprodMatrixBatchFourVectorMultiplyAccumulate()
624 const int32_t* input_offset, int32_t* row_sums) { in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()
1273 const int32_t* input_offset, int32_t* row_sums) { in NeonMatrixBatchVectorMultiplyAccumulateImpl()
1401 const int32_t* input_offset, int32_t* scratch, int32_t* row_sums, in NeonMatrixBatchVectorMultiplyAccumulate()
|
| D | neon_tensor_utils.h | 59 const int32_t* input_offset, int32_t* scratch, int32_t* row_sums, in MatrixBatchVectorMultiplyAccumulate()
|
| D | sse_tensor_utils.h | 62 const int32_t* input_offset, int32_t* scratch, int32_t* row_sums, in MatrixBatchVectorMultiplyAccumulate()
|
| D | optimized_ops.h | 1102 const RuntimeShape& scratch_shape, int32_t* scratch, int32_t* row_sums, in HybridConvPerChannel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | svdf.h | 199 int32_t* row_sums, bool* compute_row_sums) { in EvalHybridSVDF()
|
| D | batch_matmul.h | 112 const int32_t* input_offset, int32_t* row_sums, in BatchMatMul()
|
| D | portable_tensor_utils.h | 79 const int32_t* input_offset, int32_t* scratch, int32_t* row_sums, in MatrixBatchVectorMultiplyAccumulate()
|
| D | portable_tensor_utils.cc | 167 const int32_t* input_offset, int32_t* scratch, int32_t* row_sums, in PortableMatrixBatchVectorMultiplyAccumulate()
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/ |
| D | gemm_hybrid_indirect.hpp | 199 int32_t row_sums[strategy::out_height()]; in run() local
|