Home
last modified time | relevance | path

Searched defs:dimd (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DReplicationPadding.cpp138 int64_t dimd = 1; in TORCH_META_FUNC() local
234 int dimd = 1; in replication_pad3d_backward_out_cpu_template() local
H A DNaiveConvolutionTranspose3d.cpp115 int dimd = 1; in slow_conv_transpose3d_shape_check() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReplicationPadding.cu322 int dimd = 1; in shapeAndGradOutputCheck3d() local
613 int dimd = 1; in TORCH_IMPL_FUNC() local
H A DNaiveConvolutionTranspose3d.cu114 int dimd = 1; in slow_conv_transpose3d_shape_check() local