Home
last modified time | relevance | path

Searched refs:make_output_offset_calculator (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DCUDAJitLoops.cuh217 auto output_offset_calculator = make_output_offset_calculator(iter); in jitted_gpu_kernel_generic()
248 auto output_offset_calculator = make_output_offset_calculator(iter); in jitted_gpu_kernel_generic()
DLoops.cuh32 static OffsetCalculator<num_outputs> make_output_offset_calculator(const TensorIteratorBase& iter) { in make_output_offset_calculator() function
298 auto output_calc = make_output_offset_calculator<num_outputs>(iter); in gpu_kernel_multiple_outputs_impl()
DDistributionTemplates.h270 make_input_offset_calculator<2>(iter), make_output_offset_calculator(iter)); in distribution_binary_kernel()