Home
last modified time | relevance | path

Searched defs:padding_depth (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNaiveConvolutionTranspose3d.cpp39 int padding_depth, in slow_conv_transpose3d_shape_check()
218 int64_t padding_depth = padding[0]; in slow_conv_transpose3d_out_cpu_template() local
432 int64_t padding_depth = padding[0]; in slow_conv_transpose3d_backward_out_cpu_template() local
637 int64_t padding_depth = padding[0]; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNaiveConvolutionTranspose3d.cu39 int padding_depth, in slow_conv_transpose3d_shape_check()
214 int64_t padding_depth = padding[0]; in slow_conv_transpose3d_out_cuda_template() local
438 int64_t padding_depth = padding[0]; in slow_conv_transpose3d_backward_out_cuda_template() local
646 int64_t padding_depth = padding[0]; in slow_conv_transpose3d_acc_grad_parameters_cuda() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Doperations.cc468 int32_t padding_depth = CalculateSamePadding<Axis::DEPTH>(input, attr); in MakeSamePadding() local
495 int32_t padding_depth = CalculateSamePadding<Axis::DEPTH>(input, attr); in MakeSamePadding() local