Searched defs:normalized_dims (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | reduce.cc | 170 TfLiteTensor* normalized_dims) { in ResizeTempDims()
229 TfLiteTensor* normalized_dims; in InitializeTemporaries() local
252 TfLiteTensor* normalized_dims; in PrepareSimple() local
428 TfLiteTensor* normalized_dims, in EvalIntegerMean()
509 TfLiteTensor* normalized_dims; in EvalMean() local
791 TfLiteTensor* normalized_dims; in EvalType() local
917 TfLiteTensor* normalized_dims; in EvalQuantizedProd() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | reduce.h | 408 bool keep_dims, int* normalized_dims, in QuantizedMeanOrSum()
624 const int64_t num_axis_dimensions, int* resolved_axis, int* normalized_dims, in QuantizedReduceProd()
675 bool keep_dims, int* normalized_dims, 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/XNNPACK/src/operators/ |
| H A D | transpose-nd.c | 188 size_t normalized_dims; in setup_transpose_nd() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | shape_util.cc | 1714 absl::Span<const int64_t> normalized_dims = normalized_shape.dimensions(); in FindTranspose021() local
|