Searched defs:osizeT (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AdaptiveMaxPooling3d.cpp | 51 int64_t osizeT = output_size[0]; in TORCH_META_FUNC() local
92 int64_t osizeT, in adaptive_max_pool3d_single_out_frame()
170 int64_t osizeT, in adaptive_max_pool3d_out_frame()
201 int64_t osizeT, in adaptive_max_pool3d_backward_single_out_frame()
241 int64_t osizeT, in adaptive_max_pool3d_backward_out_frame()
295 int64_t osizeT = output_size[0]; in TORCH_IMPL_FUNC() local
364 int64_t osizeT = 0; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling3d.cpp | 34 int64_t osizeT, in adaptive_avg_pool3d_out_frame()
122 auto osizeT = output_size[0]; in adaptive_avg_pool3d_out_cpu_template() local
186 int64_t osizeT, in adaptive_avg_pool3d_backward_out_frame()
244 int64_t osizeT = gradOutput.size(-3); in adaptive_avg_pool3d_backward_out_cpu_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AdaptiveMaxPooling3d.cu | 49 int osizeT, int osizeH, int osizeW, in adaptivemaxpool()
129 int osizeT, int osizeH, int osizeW, in adaptivemaxpool_loop()
163 int osizeT, int osizeH, int osizeW, in adaptivemaxgradinput()
209 int osizeT, int osizeH, int osizeW) in adaptivemaxgradinput_loop()
239 int osizeT, int osizeH, int osizeW, in atomicadaptivemaxgradinput()
285 int osizeT, int osizeH, int osizeW) in atomicadaptivemaxgradinput_loop()
320 int64_t osizeT = output_size[0]; in TORCH_IMPL_FUNC() local
407 int64_t osizeT, osizeH, osizeW; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling3d.cu | 55 int osizeT, int osizeH, int osizeW, in adaptiveaveragepool()
123 int osizeT, int osizeH, int osizeW, in adaptiveaveragepool_loop()
156 int osizeT, int osizeH, int osizeW, in adaptiveaveragegradinput()
221 int osizeT, int osizeH, int osizeW) { in adaptiveaveragegradinput_loop()
254 int osizeT, int osizeH, int osizeW, in atomicadaptiveaveragegradinput()
317 int osizeT, int osizeH, int osizeW) { in atomicadaptiveaveragegradinput_loop()
362 int64_t osizeT = output_size[0]; in adaptive_avg_pool3d_out_cuda_template() local
446 int64_t osizeT, osizeH, osizeW; in adaptive_avg_pool3d_backward_out_cuda_template() local
|