| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 83 TfLiteTensor* resolved_axis) { in ResizeTempAxis() argument
86 return context->ResizeTensor(context, resolved_axis, axis_size); in ResizeTempAxis()
185 TfLiteTensor* resolved_axis; in InitializeTemporaries() local
187 context, GetTemporarySafe(context, node, /*index=*/1, &resolved_axis)); in InitializeTemporaries()
188 resolved_axis->type = kTfLiteInt32; in InitializeTemporaries()
231 TfLiteTensor* resolved_axis; in PrepareSimple() local
233 context, GetTemporarySafe(context, node, /*index=*/1, &resolved_axis)); in PrepareSimple()
237 SetTensorToDynamic(resolved_axis); in PrepareSimple()
240 resolved_axis->allocation_type = kTfLiteArenaRw; in PrepareSimple()
242 ResizeTempAxis(context, &op_context, resolved_axis)); in PrepareSimple()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 132 bool keep_dims, int* temp_index, int* resolved_axis, in ReduceGeneric() argument
148 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in ReduceGeneric()
154 output_num_dims, resolved_axis, num_resolved_axis, in ReduceGeneric()
166 int* temp_index, int* resolved_axis, U* temp_sum) { in Mean() argument
185 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in Mean()
191 output_num_dims, resolved_axis, num_resolved_axis, in Mean()
199 size_t current = static_cast<size_t>(input_dims[resolved_axis[idx]]); in Mean()
330 int* temp_index, int* resolved_axis, U* temp_sum, in QuantizedMeanOrSum() argument
358 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in QuantizedMeanOrSum()
364 output_num_dims, resolved_axis, num_resolved_axis, in QuantizedMeanOrSum()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | reduce.cc | 149 int resolved_axis[kMaxNumberOfReducedAxis]; in EvalMean() local
176 params->keep_dims, temp_index, resolved_axis, in EvalMean()
200 params->keep_dims, temp_index, resolved_axis, temp_buffer)); in EvalMean()
212 num_axis, params->keep_dims, temp_index, resolved_axis, in EvalMean()
237 resolved_axis, temp_buffer)); in EvalMean()
249 num_axis, params->keep_dims, temp_index, resolved_axis, in EvalMean()
274 int* resolved_axis = static_cast<int*>( in EvalMax() local
285 params->keep_dims, temp_buffer, resolved_axis, in EvalMax()
302 params->keep_dims, temp_buffer, resolved_axis, in EvalMax()
|
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 326 std::vector<int> resolved_axis(num_axis); in Eval() local
330 &resolved_axis[0], &num_resolved_axis)) { in Eval()
334 return FlexibleLayerNorm(input, scale, offset, &resolved_axis[0], in Eval()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | ceva_tflm_lib.h | 554 bool keep_dims, int* temp_index, int* resolved_axis,
566 bool keep_dims, int* temp_index, int* resolved_axis,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1201 bool keep_dims, int* temp_index, int* resolved_axis, in MeanGeneral() argument
1206 resolved_axis, temp_sum); in MeanGeneral()
1214 int* temp_index, int* resolved_axis, float* temp_sum) {
1235 resolved_axis, temp_sum);
|