Searched defs:otime_for_shape_check (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | AveragePool3d.cpp | 130 …const int64_t otime_for_shape_check = pooling_output_shape<int64_t>(itime, kT, padT, dT, 1, ceil_m… in TORCH_META_FUNC() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | AveragePool3d.cu | 504 …const int64_t otime_for_shape_check = pooling_output_shape<int64_t>(itime, kT, padT, dT, 1, ceil_m… in TORCH_IMPL_FUNC() local
|