Searched refs:make_input_offset_calculator (Results 1 – 3 of 3) sorted by relevance
216 auto input_offset_calculator = make_input_offset_calculator<arity>(iter); in jitted_gpu_kernel_generic()247 auto input_offset_calculator = make_input_offset_calculator<arity>(iter); in jitted_gpu_kernel_generic()
18 static OffsetCalculator<N> make_input_offset_calculator(const TensorIteratorBase& iter) { in make_input_offset_calculator() function297 auto input_calc = make_input_offset_calculator<num_inputs>(iter); in gpu_kernel_multiple_outputs_impl()
270 make_input_offset_calculator<2>(iter), make_output_offset_calculator(iter)); in distribution_binary_kernel()