Home
last modified time | relevance | path

Searched defs:reduction_size (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils_impl.h206 int reduction_size) { in PortableReductionSumVector()
Dportable_tensor_utils.h280 int output_size, int reduction_size) { in ReductionSumVector()
286 int output_size, int reduction_size) { in ReductionSumVector()
292 int output_size, int reduction_size) { in ReductionSumVector()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dmean_stddev_normalization.cc28 std::string GetReduceCode(const GpuInfo& gpu_info, int reduction_size) { in GetReduceCode()
Dreduce.cc122 int reduction_size = it->second; in Reduce() local
432 const double reduction_size = total_src_elements / total_dst_elements; in BindArguments() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dsse_tensor_utils.h291 int output_size, int reduction_size) { in ReductionSumVector()
297 int output_size, int reduction_size) { in ReductionSumVector()
303 int output_size, int reduction_size) { in ReductionSumVector()
Dneon_tensor_utils.h276 int output_size, int reduction_size) { in ReductionSumVector()
282 int output_size, int reduction_size) { in ReductionSumVector()
288 int output_size, int reduction_size) { in ReductionSumVector()
Dsse_tensor_utils.cc511 const int output_size, const int reduction_size) { in SseReductionSumVector()
Dneon_tensor_utils.cc2463 int output_size, int reduction_size) { in NeonReductionSumVector()
2487 const int output_size, const int reduction_size) { in NeonReductionSumVector()