Searched defs:isizeT (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AdaptiveMaxPooling3d.cu | 48 int isizeT, int isizeH, int isizeW, in adaptivemaxpool()
128 int isizeT, int isizeH, int isizeW, in adaptivemaxpool_loop()
162 int isizeT, int isizeH, int isizeW, in adaptivemaxgradinput()
208 int isizeT, int isizeH, int isizeW, in adaptivemaxgradinput_loop()
238 int isizeT, int isizeH, int isizeW, in atomicadaptivemaxgradinput()
284 int isizeT, int isizeH, int isizeW, in atomicadaptivemaxgradinput_loop()
324 int64_t sizeD, isizeT, isizeH, isizeW; in TORCH_IMPL_FUNC() local
406 int64_t sizeD, isizeT, isizeH, isizeW; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling3d.cu | 54 int isizeT, int isizeH, int isizeW, in adaptiveaveragepool()
122 int isizeT, int isizeH, int isizeW, in adaptiveaveragepool_loop()
155 int isizeT, int isizeH, int isizeW, in adaptiveaveragegradinput()
220 int isizeT, int isizeH, int isizeW, in adaptiveaveragegradinput_loop()
253 int isizeT, int isizeH, int isizeW, in atomicadaptiveaveragegradinput()
316 int isizeT, int isizeH, int isizeW, in atomicadaptiveaveragegradinput_loop()
366 int64_t sizeD, isizeT, isizeH, isizeW; in adaptive_avg_pool3d_out_cuda_template() local
445 int64_t sizeD, isizeT, isizeH, isizeW; in adaptive_avg_pool3d_backward_out_cuda_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AdaptiveMaxPooling3d.cpp | 89 int64_t isizeT, in adaptive_max_pool3d_single_out_frame()
167 int64_t isizeT, in adaptive_max_pool3d_out_frame()
198 int64_t isizeT, in adaptive_max_pool3d_backward_single_out_frame()
238 int64_t isizeT, in adaptive_max_pool3d_backward_out_frame()
265 int64_t isizeT = 0; in TORCH_IMPL_FUNC() local
361 int64_t isizeT = 0; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling3d.cpp | 31 int64_t isizeT, in adaptive_avg_pool3d_out_frame()
113 int64_t isizeT = input.size(-3); in adaptive_avg_pool3d_out_cpu_template() local
183 int64_t isizeT, in adaptive_avg_pool3d_backward_out_frame()
241 int64_t isizeT = input.size(-3); in adaptive_avg_pool3d_backward_out_cpu_template() local
|