Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A Djiterator.cu92 TrivialOffsetCalculatorVariant output_offset_calculator(iter.noutputs()); in launch_jitted_vectorized_kernel_dynamic() local
221 OffsetCalculatorVariant</*is_input=*/false> output_offset_calculator(iter); in jitted_gpu_kernel_dynamic_impl() local
251 TrivialOffsetCalculatorVariant output_offset_calculator(iter.noutputs()); in jitted_gpu_kernel_dynamic_impl() local
263 OffsetCalculatorVariant</*is_input=*/false> output_offset_calculator(iter); in jitted_gpu_kernel_dynamic_impl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DCUDAJitLoops.cuh217 auto output_offset_calculator = make_output_offset_calculator(iter); in jitted_gpu_kernel_generic() local
239 auto output_offset_calculator = TrivialOffsetCalculator<1>(); in jitted_gpu_kernel_generic() local
248 auto output_offset_calculator = make_output_offset_calculator(iter); in jitted_gpu_kernel_generic() local
H A DMemoryAccess.cuh192 out_calc_t output_offset_calculator; member
299 out_calc_t output_offset_calculator; member
H A DSpectralOps.cu105 HermitianSymmetryOffsetCalculator<int64_t> output_offset_calculator( in _fft_fill_with_conjugate_symmetry_cuda_() local