Searched refs:make_output_offset_calculator (Results 1 – 3 of 3) sorted by relevance
217 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()
32 static OffsetCalculator<num_outputs> make_output_offset_calculator(const TensorIteratorBase& iter) { in make_output_offset_calculator() function298 auto output_calc = make_output_offset_calculator<num_outputs>(iter); in gpu_kernel_multiple_outputs_impl()
270 make_input_offset_calculator<2>(iter), make_output_offset_calculator(iter)); in distribution_binary_kernel()