Searched refs:resolved_axis (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 77 TfLiteTensor* resolved_axis) { in ResizeTempAxis() argument
80 return context->ResizeTensor(context, resolved_axis, axis_size); in ResizeTempAxis()
177 TfLiteTensor* resolved_axis = GetTemporary(context, node, /*index=*/1); in InitializeTemporaries() local
178 resolved_axis->type = kTfLiteInt32; in InitializeTemporaries()
214 TfLiteTensor* resolved_axis = GetTemporary(context, node, /*index=*/1); in PrepareSimple() local
218 SetTensorToDynamic(resolved_axis); in PrepareSimple()
221 resolved_axis->allocation_type = kTfLiteArenaRw; in PrepareSimple()
223 ResizeTempAxis(context, &op_context, resolved_axis)); in PrepareSimple()
276 TfLiteTensor* resolved_axis = GetTemporary(context, node, /*index=*/1); in EvalMean() local
281 ResizeTempAxis(context, &op_context, resolved_axis)); in EvalMean()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3524 bool keep_dims, int* temp_index, int* resolved_axis, in ReduceGeneric() argument
3535 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in ReduceGeneric()
3541 output_num_dims, resolved_axis, num_resolved_axis, in ReduceGeneric()
3553 int* temp_index, int* resolved_axis, U* temp_sum) { in Mean() argument
3572 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in Mean()
3578 output_num_dims, resolved_axis, num_resolved_axis, in Mean()
3586 size_t current = static_cast<size_t>(input_dims[resolved_axis[idx]]); in Mean()
3717 int* temp_index, int* resolved_axis, U* temp_sum, in QuantizedMeanOrSum() argument
3738 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in QuantizedMeanOrSum()
3744 output_num_dims, resolved_axis, num_resolved_axis, in QuantizedMeanOrSum()
[all …]
|