Home
last modified time | relevance | path

Searched defs:input_offset_calculator (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A Djiterator.cu90 TrivialOffsetCalculatorVariant input_offset_calculator(iter.ninputs()); in launch_jitted_vectorized_kernel_dynamic() local
219 OffsetCalculatorVariant</*is_input=*/true> input_offset_calculator(iter); in jitted_gpu_kernel_dynamic_impl() local
249 TrivialOffsetCalculatorVariant input_offset_calculator(iter.ninputs()); in jitted_gpu_kernel_dynamic_impl() local
261 OffsetCalculatorVariant</*is_input=*/true> input_offset_calculator(iter); in jitted_gpu_kernel_dynamic_impl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DCUDAJitLoops.cuh216 auto input_offset_calculator = make_input_offset_calculator<arity>(iter); in jitted_gpu_kernel_generic() local
238 auto input_offset_calculator = TrivialOffsetCalculator<arity>(); in jitted_gpu_kernel_generic() local
247 auto input_offset_calculator = make_input_offset_calculator<arity>(iter); in jitted_gpu_kernel_generic() local
H A DMemoryAccess.cuh191 inp_calc_t input_offset_calculator; member
298 inp_calc_t input_offset_calculator; member
H A DSpectralOps.cu103 OffsetCalculator<1, int64_t> input_offset_calculator( in _fft_fill_with_conjugate_symmetry_cuda_() local