Home
last modified time | relevance | path

Searched defs:oFrame (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAveragePool3d.cu47 int oFrame = (blockIdx.z + offsetZ) % output.size(1); // output frame/time in avg_pool3d_cuda_update_output() local
116 int oFrame = (blockIdx.z + offsetZ) % output.size(1); // output frame/time in avg_pool3d_cuda_update_output() local
190 for (int oFrame = max(0, iFrame - kT + 1); in avg_pool3d_single_backward_out_frame_stride1() local
227 int oFrame = (blockIdx.z + offsetZ) % gradOutput.size(1); // gradOutput frame/time in avg_pool3d_cuda_update_grad_input_atomic() local
285 int oFrame = (blockIdx.z + offsetZ) % gradOutput.size(1); // gradOutput frame/time in avg_pool3d_cuda_update_grad_input() local
H A DDilatedMaxPool3d.cu50 int oFrame = 0; in max_pool3d_with_indices_single_out_frame() local
199 int oFrame = 0; in max_pool3d_with_indices_backward_single_out_frame() local