Home
last modified time | relevance | path

Searched defs:randomSamples (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DFractionalMaxPool2d.cpp134 const scalar_t* randomSamples, in fractional_max_pool2d_out_single_batch_frame()
195 const scalar_t* randomSamples, in fractional_max_pool2d_out_frame()
303 auto randomSamples = randomSamples_.contiguous(); in TORCH_IMPL_FUNC() local
H A DFractionalMaxPool3d.cpp106 const scalar_t* randomSamples, in fractional_max_pool3d_out_single_batch_frame()
178 const scalar_t* randomSamples, in fractional_max_pool3d_out_frame()
237 auto randomSamples = randomSamples_.contiguous(); in TORCH_IMPL_FUNC() local
H A DFractionalMaxPooling.h33 const Tensor& randomSamples) { in fractional_max_pool_check_shape()