| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 86 TfLiteTensor* resolved_axis) { in ResizeTempAxis() argument
89 return context->ResizeTensor(context, resolved_axis, axis_size); in ResizeTempAxis()
196 TfLiteTensor* resolved_axis; in InitializeTemporaries() local
198 context, GetTemporarySafe(context, node, /*index=*/1, &resolved_axis)); in InitializeTemporaries()
199 resolved_axis->type = kTfLiteInt32; in InitializeTemporaries()
249 TfLiteTensor* resolved_axis; in PrepareSimple() local
251 context, GetTemporarySafe(context, node, /*index=*/1, &resolved_axis)); in PrepareSimple()
266 SetTensorToDynamic(resolved_axis); in PrepareSimple()
269 resolved_axis->allocation_type = kTfLiteArenaRw; in PrepareSimple()
271 ResizeTempAxis(context, &op_context, resolved_axis)); in PrepareSimple()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 184 bool keep_dims, int* temp_index, int* resolved_axis, in ReduceGeneric() argument
203 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in ReduceGeneric()
209 output_num_dims, resolved_axis, num_resolved_axis, in ReduceGeneric()
221 int* temp_index, int* resolved_axis, U* temp_sum) { in Mean() argument
240 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in Mean()
246 output_num_dims, resolved_axis, num_resolved_axis, in Mean()
254 size_t current = static_cast<size_t>(input_dims[resolved_axis[idx]]); in Mean()
385 int* temp_index, int* resolved_axis, U* temp_sum, in QuantizedMeanOrSum() argument
421 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in QuantizedMeanOrSum()
427 output_num_dims, resolved_axis, num_resolved_axis, in QuantizedMeanOrSum()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | reduce.h | 409 int* resolved_axis, U* temp_sum, bool compute_sum) { in QuantizedMeanOrSum() argument
435 resolved_axis, num_resolved_axis, input_dims, in QuantizedMeanOrSum()
447 input_data, normalized_dims, normalized_num_dims, resolved_axis, in QuantizedMeanOrSum()
457 size_t current = static_cast<size_t>(normalized_dims[resolved_axis[idx]]); in QuantizedMeanOrSum()
624 const int64_t num_axis_dimensions, int* resolved_axis, int* normalized_dims, in QuantizedReduceProd() argument
633 num_axis_dimensions, resolved_axis, in QuantizedReduceProd()
640 input_data, normalized_dims, normalized_num_dims, resolved_axis, in QuantizedReduceProd()
676 int* resolved_axis, U* temp_sum) { in MeanGeneral() argument
682 resolved_axis, num_resolved_axis, input_dims, in MeanGeneral()
703 input_data, normalized_dims, normalized_num_dims, resolved_axis, in MeanGeneral()
[all …]
|
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 422 std::vector<int> resolved_axis(num_axis); in Eval() local
426 &resolved_axis[0], &num_resolved_axis)) { in Eval()
430 return FlexibleLayerNorm(input, scale, offset, &resolved_axis[0], in Eval()
|