Searched defs:resolved_axis (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | reduce.cc | 86 TfLiteTensor* resolved_axis) { in ResizeTempAxis()
196 TfLiteTensor* resolved_axis; in InitializeTemporaries() local
249 TfLiteTensor* resolved_axis; in PrepareSimple() local
378 int* temp_index, int* resolved_axis, U* temp_sum, in Mean()
398 int* resolved_axis, int* temp_sum, in QuantizedMeanOrSum()
426 TfLiteTensor* resolved_axis, in EvalIntegerMean()
496 TfLiteTensor* resolved_axis; in EvalMean() local
707 TfLiteTensor* resolved_axis; in EvalType() local
867 TfLiteTensor* resolved_axis; in EvalSum() local
911 TfLiteTensor* resolved_axis; in EvalQuantizedProd() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | reduce.h | 184 bool keep_dims, int* temp_index, int* resolved_axis, in ReduceGeneric()
221 int* temp_index, int* resolved_axis, U* temp_sum) { in Mean()
385 int* temp_index, int* resolved_axis, U* temp_sum, in QuantizedMeanOrSum()
479 int* resolved_axis, int32_t* temp_prod, in QuantizedReduceProd()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | reduce.h | 409 int* resolved_axis, U* temp_sum, bool compute_sum) { in QuantizedMeanOrSum()
624 const int64_t num_axis_dimensions, int* resolved_axis, int* normalized_dims, in QuantizedReduceProd()
676 int* resolved_axis, U* temp_sum) { in MeanGeneral()
733 int* normalized_dims, int* resolved_axis, U* temp_sum) { in Mean()
783 int* resolved_axis, int* normalized_dims, in ReduceGeneric()
|
| /aosp_15_r20/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| H A D | layer_norm.cc | 422 std::vector<int> resolved_axis(num_axis); in Eval() local
|