| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | fully_connected.cc | 375 TfLiteTensor* accum_scratch; in PrepareImpl() local
516 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalHybridDense()
585 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalSparseHybridImpl()
711 TfLiteTensor* accum_scratch; member
722 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalHybrid()
963 TfLiteTensor* accum_scratch; in EvalQuantized() local
|
| H A D | basic_rnn.cc | 155 TfLiteTensor* accum_scratch; in Prepare() local
237 TfLiteTensor* zero_points, TfLiteTensor* accum_scratch, in EvalHybrid()
319 TfLiteTensor* accum_scratch; in Eval() local
|
| H A D | unidirectional_sequence_rnn.cc | 162 TfLiteTensor* accum_scratch; in Prepare() local
274 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalHybrid()
394 TfLiteTensor* accum_scratch; in Eval() local
|
| H A D | batch_matmul.cc | 249 TfLiteTensor* accum_scratch; in InitializeTemporaries() local
452 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalHybrid()
631 TfLiteTensor* accum_scratch; in EvalQuantized() local
|
| H A D | bidirectional_sequence_rnn.cc | 264 TfLiteTensor* accum_scratch; in Prepare() local
530 TfLiteTensor* accum_scratch, TfLiteTensor* fw_row_sums, in EvalHybrid()
811 TfLiteTensor* accum_scratch; in Eval() local
|
| H A D | conv.cc | 585 TfLiteTensor* accum_scratch; in Prepare() local
1070 TfLiteTensor* accum_scratch, TfLiteTensor* output) { in EvalHybrid()
1185 TfLiteTensor* accum_scratch = in EvalImpl() local
|
| H A D | bidirectional_sequence_lstm.cc | 833 TfLiteTensor* accum_scratch; in Prepare() local
1255 TfLiteTensor* accum_scratch; in Eval() local
|
| H A D | unidirectional_sequence_lstm.cc | 1108 TfLiteTensor* accum_scratch; in Prepare() local
|
| H A D | lstm.cc | 1557 TfLiteTensor* accum_scratch; in Prepare() local
|
| H A D | lstm_eval.cc | 340 void CalculateLstmGateHybrid( in CalculateLstmGateHybrid()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A 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()
|
| H A D | tensor_utils_test.cc | 1460 std::vector<int32_t> accum_scratch(a_rows * batches); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | batch_matmul.h | 120 int32_t* accum_scratch, float* output_data, in BatchMatMul()
|
| H A D | optimized_ops.h | 1009 int32_t* accum_scratch, const RuntimeShape& output_shape, in HybridConv()
|