Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils_impl.h215 int reduction_size) { in PortableReductionSumVector()
Dportable_tensor_utils.h295 int output_size, int reduction_size) { in ReductionSumVector()
301 int output_size, int reduction_size) { in ReductionSumVector()
307 int output_size, int reduction_size) { in ReductionSumVector()
/external/pytorch/aten/src/ATen/native/cuda/
DNormalization.cuh878 auto reduction_size = input_.numel() / n_input; in batch_norm_backward_elemt_cuda_template() local
950 const int reduction_size, in batch_norm_collect_statistics_channels_last_kernel()
1096 const int reduction_size, in batch_norm_transform_input_channels_last_kernel()
1178 const int reduction_size, in batch_norm_backward_reduce_channels_last_kernel()
1334 const int reduction_size, in batch_norm_backward_elemt_channels_last_kernel_impl()
1389 const int reduction_size, in batch_norm_backward_elemt_channels_last_kernel()
1418 const int reduction_size, in batch_norm_backward_elemt_channels_last_kernel()
1431 const auto reduction_size = input.numel() / stride; in batch_norm_stats_channels_last_cuda_template() local
1476 const auto reduction_size = input.numel() / stride; in batch_norm_elemt_channels_last_cuda_template() local
1535 const auto reduction_size = input.numel() / stride; in batch_norm_backward_reduce_cuda_channels_last_template() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dreduce.cc138 int reduction_size = it->second; in Reduce() local
487 const double reduction_size = total_src_elements / total_dst_elements; in BindArguments() local
Dmean_stddev_normalization.cc68 int reduction_size = work_group_size.z; in GetReduceCode() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_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()
Dsse_tensor_utils.h306 int output_size, int reduction_size) { in ReductionSumVector()
312 int output_size, int reduction_size) { in ReductionSumVector()
318 int output_size, int reduction_size) { in ReductionSumVector()
Dsse_tensor_utils.cc621 const int output_size, const int reduction_size) { in SseReductionSumVector()
Dneon_tensor_utils.cc2566 int output_size, int reduction_size) { in NeonReductionSumVector()
2590 const int output_size, const int reduction_size) { in NeonReductionSumVector()