Home
last modified time | relevance | path

Searched refs:reduction_size (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h280 int output_size, int reduction_size) { in ReductionSumVector() argument
282 reduction_size); in ReductionSumVector()
286 int output_size, int reduction_size) { in ReductionSumVector() argument
288 reduction_size); in ReductionSumVector()
292 int output_size, int reduction_size) { in ReductionSumVector() argument
294 reduction_size); in ReductionSumVector()
Dportable_tensor_utils_impl.h205 int output_size, int reduction_size) { in PortableReductionSumVector() argument
208 for (int r = 0; r < reduction_size; r++) { in PortableReductionSumVector()
212 input_vector += reduction_size; in PortableReductionSumVector()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dsse_tensor_utils.h286 int output_size, int reduction_size) { in ReductionSumVector() argument
288 reduction_size); in ReductionSumVector()
292 int output_size, int reduction_size) { in ReductionSumVector() argument
294 reduction_size); in ReductionSumVector()
298 int output_size, int reduction_size) { in ReductionSumVector() argument
300 reduction_size); in ReductionSumVector()
Dneon_tensor_utils.h276 int output_size, int reduction_size) { in ReductionSumVector() argument
278 reduction_size); in ReductionSumVector()
282 int output_size, int reduction_size) { in ReductionSumVector() argument
284 reduction_size); in ReductionSumVector()
288 int output_size, int reduction_size) { in ReductionSumVector() argument
290 reduction_size); in ReductionSumVector()
Dsse_tensor_utils.cc418 const int output_size, const int reduction_size) { in SseReductionSumVector() argument
421 const int8_t* __restrict__ row_ptr = input_vector + row * reduction_size; in SseReductionSumVector()
424 for (; col < (reduction_size & ~(kBlockSize - 1)); col += kBlockSize) { in SseReductionSumVector()
432 if (col < (reduction_size & ~7)) { in SseReductionSumVector()
448 for (; col < reduction_size; col++) { in SseReductionSumVector()
Dneon_tensor_utils_impl.h166 int output_size, int reduction_size);
169 int output_size, int reduction_size);
Dsse_tensor_utils_impl.h64 const int output_size, const int reduction_size);
Dneon_tensor_utils.cc2467 int output_size, int reduction_size) { in NeonReductionSumVector() argument
2473 RoundDownVectors<kFloatValuesPerNeonVector>(reduction_size); in NeonReductionSumVector()
2482 for (; TFLITE_UNLIKELY(r < reduction_size); r++) { in NeonReductionSumVector()
2486 input_vector += reduction_size; in NeonReductionSumVector()
2491 const int output_size, const int reduction_size) { in NeonReductionSumVector() argument
2493 RoundDownVectors<kInt8ValuesPerNeonVector>(reduction_size); in NeonReductionSumVector()
2495 RoundDownVectors<(kInt8ValuesPerNeonVector / 2)>(reduction_size); in NeonReductionSumVector()
2509 for (; TFLITE_UNLIKELY(r < reduction_size); ++r) { in NeonReductionSumVector()
2513 input_vector += reduction_size; in NeonReductionSumVector()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dreduce.cc122 int reduction_size = it->second; in Reduce() local
124 reduction_size = DivideRoundUp(reduction_size, 4); in Reduce()
126 ordered_sizes.push_back(reduction_size); in Reduce()
432 const double reduction_size = total_src_elements / total_dst_elements; in BindArguments() local
436 const double size_1 = reduction_size / size_0; in BindArguments()
440 RETURN_IF_ERROR(args->SetFloat("inv_multiplier_1", 1.0 / reduction_size)); in BindArguments()
Dmean_stddev_normalization.cc33 std::string GetReduceCode(const GpuInfo& gpu_info, int reduction_size) { in GetReduceCode() argument
63 result += " int reduction_size = " + std::to_string(reduction_size) + ";\n"; in GetReduceCode()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils_common.h439 int output_size, int reduction_size);
443 int output_size, int reduction_size);
447 int output_size, int reduction_size);