Searched defs:accum_scratch (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn.cc | 155 TfLiteTensor* accum_scratch; in Prepare() local
236 TfLiteTensor* zero_points, TfLiteTensor* accum_scratch, in EvalHybrid()
318 TfLiteTensor* accum_scratch; in Eval() local
|
| 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
|
| D | fully_connected.cc | 298 TfLiteTensor* accum_scratch; in PrepareImpl() local
439 TfLiteTensor* scaling_factors, TfLiteTensor* accum_scratch, in EvalHybridImpl()
569 TfLiteTensor* accum_scratch; member
585 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalHybrid()
701 TfLiteTensor* accum_scratch; in EvalQuantized() local
|
| D | batch_matmul.cc | 246 TfLiteTensor* accum_scratch; in InitializeTemporaries() local
448 TfLiteTensor* accum_scratch, TfLiteTensor* row_sums, in EvalHybrid()
582 TfLiteTensor* accum_scratch; in EvalQuantized() local
|
| D | bidirectional_sequence_rnn.cc | 264 TfLiteTensor* accum_scratch; in Prepare() local
528 TfLiteTensor* accum_scratch, TfLiteTensor* fw_row_sums, in EvalHybrid()
809 TfLiteTensor* accum_scratch; in Eval() local
|
| D | conv.cc | 537 TfLiteTensor* accum_scratch; in Prepare() local
922 TfLiteTensor* accum_scratch, TfLiteTensor* output) { in EvalHybrid()
1024 TfLiteTensor* accum_scratch = in EvalImpl() local
|
| D | bidirectional_sequence_lstm.cc | 829 TfLiteTensor* accum_scratch; in Prepare() local
1250 TfLiteTensor* accum_scratch; in Eval() local
|
| D | unidirectional_sequence_lstm.cc | 1096 TfLiteTensor* accum_scratch; in Prepare() local
|
| D | lstm.cc | 1554 TfLiteTensor* accum_scratch; in Prepare() local
|
| D | lstm_eval.cc | 298 void CalculateLstmGateHybrid( in CalculateLstmGateHybrid()
|
| /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_test.cc | 1464 std::vector<int32_t> accum_scratch(a_rows * batches); in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | batch_matmul.h | 120 int32_t* accum_scratch, float* output_data, in BatchMatMul()
|
| D | optimized_ops.h | 1340 int32_t* accum_scratch, const RuntimeShape& output_shape, in HybridConv()
|