Searched defs:resolved_axis (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 77 TfLiteTensor* resolved_axis) { in ResizeTempAxis()
177 TfLiteTensor* resolved_axis = GetTemporary(context, node, /*index=*/1); in InitializeTemporaries() local
214 TfLiteTensor* resolved_axis = GetTemporary(context, node, /*index=*/1); in PrepareSimple() local
276 TfLiteTensor* resolved_axis = GetTemporary(context, node, /*index=*/1); in EvalMean() local
406 TfLiteTensor* resolved_axis = GetTemporary(context, node, /*index=*/1); in EvalLogic() local
533 TfLiteTensor* resolved_axis = GetTemporary(context, node, /*index=*/1); in EvalSum() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3524 bool keep_dims, int* temp_index, int* resolved_axis, in ReduceGeneric()
3553 int* temp_index, int* resolved_axis, U* temp_sum) { in Mean()
3717 int* temp_index, int* resolved_axis, U* temp_sum, in QuantizedMeanOrSum()
|