Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DReplicationPadding.cpp157 int64_t odepth = idepth + pfront + pback; in TORCH_META_FUNC() local
246 int64_t odepth = idepth + pfront + pback; in replication_pad3d_backward_out_cpu_template() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReplicationPadding.cu336 int odepth = idepth + pfront + pback; in shapeAndGradOutputCheck3d() local