Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/micro/kernels/
Dsvdf_common.cc51 int32_t* scratch_tensor = static_cast<int32_t*>( in EvalIntegerSvdfReference() local
109 int32_t* scratch_ptr_batch = scratch_tensor + b * n_filter; in EvalIntegerSvdfReference()
152 int32_t* scratch_ptr_batch = scratch_tensor + b * n_filter; in EvalIntegerSvdfReference()
/external/tensorflow/tensorflow/lite/kernels/
Dsvdf.cc152 TfLiteTensor* scratch_tensor; in Prepare() local
154 context, GetTemporarySafe(context, node, /*index=*/0, &scratch_tensor)); in Prepare()
159 scratch_tensor->type = kTfLiteInt32; in Prepare()
161 scratch_tensor->type = kTfLiteFloat32; in Prepare()
163 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
164 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_tensor, in Prepare()
Dlstm.cc1758 TfLiteTensor* scratch_tensor; in Prepare() local
1761 GetTemporarySafe(context, node, scratch_index, &scratch_tensor)); in Prepare()
1762 scratch_tensor->type = kTfLiteInt16; in Prepare()
1764 scratch_tensor->type = kTfLiteInt8; in Prepare()
1766 scratch_tensor->type = kTfLiteInt32; in Prepare()
1768 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
1770 if (!TfLiteIntArrayEqualsArray(scratch_tensor->dims, 2, in Prepare()
1776 context->ResizeTensor(context, scratch_tensor, in Prepare()
1797 TfLiteTensor* scratch_tensor; in Prepare() local
1800 GetTemporarySafe(context, node, scratch_index, &scratch_tensor)); in Prepare()
[all …]
Dreduce.cc173 TfLiteTensor* scratch_tensor; in InitializeTemporaries() local
175 context, GetTemporarySafe(context, node, /*index=*/0, &scratch_tensor)); in InitializeTemporaries()
176 scratch_tensor->type = kTfLiteInt32; in InitializeTemporaries()
177 scratch_tensor->allocation_type = kTfLiteArenaRw; in InitializeTemporaries()
181 context->ResizeTensor(context, scratch_tensor, index_size)); in InitializeTemporaries()
Dunidirectional_sequence_lstm.cc1173 TfLiteTensor* scratch_tensor; in Prepare() local
1175 &scratch_tensor)); in Prepare()
1177 scratch_tensor->type = kTfLiteInt16; in Prepare()
1179 scratch_tensor->type = kTfLiteInt8; in Prepare()
1181 scratch_tensor->type = kTfLiteInt32; in Prepare()
1184 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
1186 if (!TfLiteIntArrayEqualsArray(scratch_tensor->dims, 2, in Prepare()
1192 context->ResizeTensor(context, scratch_tensor, in Prepare()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsvdf.cc72 int32_t* scratch_tensor = static_cast<int32_t*>( in EvalIntegerSvdfHifimini() local
74 TFLITE_DCHECK(scratch_tensor != nullptr); in EvalIntegerSvdfHifimini()
163 int32_t* scratch_ptr_batch = scratch_tensor + b * n_filter; in EvalIntegerSvdfHifimini()
218 int32_t* scratch_ptr_batch = scratch_tensor + b * n_filter; in EvalIntegerSvdfHifimini()
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc452 typename TTypes<U, 2>::Tensor scratch_tensor( in operator ()() local
470 scratch_tensor.device(d) = y_backprop_rest_by_depth * x_scaled; in operator ()()
491 scratch_tensor.device(d) = y_backprop_rest_by_depth * x_centered; in operator ()()