Home
last modified time | relevance | path

Searched refs:contiguousIfZeroInStrides (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/cudnn/
DGridSampler.cpp88 auto input_contig = contiguousIfZeroInStrides(input_t); in cudnn_grid_sampler_forward()
138 auto input_contig = contiguousIfZeroInStrides(input_t); in cudnn_grid_sampler_backward()
140 auto grad_output_contig = contiguousIfZeroInStrides(grad_output_t); in cudnn_grid_sampler_backward()
/external/pytorch/aten/src/ATen/cudnn/
DUtils.h14 inline Tensor contiguousIfZeroInStrides(const Tensor& t) { in contiguousIfZeroInStrides() function
/external/pytorch/aten/src/ATen/miopen/
DUtils.h11 inline Tensor contiguousIfZeroInStrides(const Tensor& t) { in contiguousIfZeroInStrides() function