Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/cuda/
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
/external/pytorch/aten/src/ATen/native/cuda/
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
DSpectralOps.cu103 OffsetCalculator<1, int64_t> input_offset_calculator( in _fft_fill_with_conjugate_symmetry_cuda_() local
DMemoryAccess.cuh191 inp_calc_t input_offset_calculator; member
298 inp_calc_t input_offset_calculator; member