Home
last modified time | relevance | path

Searched refs:scratch_tensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dadd_n.cc66 TfLiteTensor* scratch_tensor; in Prepare() local
68 context, GetTemporarySafe(context, node, /*index=*/0, &scratch_tensor)); in Prepare()
69 scratch_tensor->type = input1->type; in Prepare()
70 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
86 context, context->ResizeTensor(context, scratch_tensor, scratch_shape)); in Prepare()
116 TfLiteTensor* scratch_tensor; in EvalAddN() local
118 GetTemporarySafe(context, node, 0, &scratch_tensor)); in EvalAddN()
121 GetTensorData<T>(scratch_tensor), cpu_backend_context); in EvalAddN()
H A Dsvdf.cc153 TfLiteTensor* scratch_tensor; in Prepare() local
155 context, GetTemporarySafe(context, node, /*index=*/0, &scratch_tensor)); in Prepare()
160 scratch_tensor->type = kTfLiteInt32; in Prepare()
162 scratch_tensor->type = kTfLiteFloat32; in Prepare()
164 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
165 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_tensor, in Prepare()
H A Dlstm.cc1762 TfLiteTensor* scratch_tensor; in Prepare() local
1765 GetTemporarySafe(context, node, scratch_index, &scratch_tensor)); in Prepare()
1766 scratch_tensor->type = kTfLiteInt16; in Prepare()
1768 scratch_tensor->type = kTfLiteInt8; in Prepare()
1770 scratch_tensor->type = kTfLiteInt32; in Prepare()
1772 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
1774 if (!TfLiteIntArrayEqualsArray(scratch_tensor->dims, 2, in Prepare()
1780 context->ResizeTensor(context, scratch_tensor, in Prepare()
1801 TfLiteTensor* scratch_tensor; in Prepare() local
1804 GetTemporarySafe(context, node, scratch_index, &scratch_tensor)); in Prepare()
[all …]
H A Dunidirectional_sequence_lstm.cc1186 TfLiteTensor* scratch_tensor; in Prepare() local
1188 &scratch_tensor)); in Prepare()
1190 scratch_tensor->type = kTfLiteInt16; in Prepare()
1192 scratch_tensor->type = kTfLiteInt8; in Prepare()
1194 scratch_tensor->type = kTfLiteInt32; in Prepare()
1197 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
1199 if (!TfLiteIntArrayEqualsArray(scratch_tensor->dims, 2, in Prepare()
1205 context->ResizeTensor(context, scratch_tensor, in Prepare()
H A Dreduce.cc184 TfLiteTensor* scratch_tensor; in InitializeTemporaries() local
186 context, GetTemporarySafe(context, node, /*index=*/0, &scratch_tensor)); in InitializeTemporaries()
187 scratch_tensor->type = kTfLiteInt32; in InitializeTemporaries()
188 scratch_tensor->allocation_type = kTfLiteArenaRw; in InitializeTemporaries()
192 context->ResizeTensor(context, scratch_tensor, index_size)); in InitializeTemporaries()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_op.cc467 typename TTypes<U, 2>::Tensor scratch_tensor( in operator ()() local
485 scratch_tensor.device(d) = y_backprop_rest_by_depth * x_scaled; in operator ()()
506 scratch_tensor.device(d) = y_backprop_rest_by_depth * x_centered; in operator ()()