Home
last modified time | relevance | path

Searched defs:pad_depth (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolutionMM3d.cpp40 const int64_t pad_depth = padding[0]; in compute_columns3d() local
121 int64_t pad_depth, in slow_conv3d_shape_check()
275 int64_t pad_depth, in slow_conv3d_update_output_frame()
323 int64_t pad_depth, in slow_conv3d_backward_update_grad_input_frame()
382 const int64_t pad_depth = padding[0]; in slow_conv3d_backward_out_cpu_template() local
506 const int64_t pad_depth = padding[0]; in slow_conv3d_backward_parameters_out_cpu_template() local
567 const int64_t pad_depth = padding[0]; in slow_conv3d_forward_out_cpu() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dim2col_utils.h291 int pad_depth, int pad_height, int pad_width, // Padding params. in ExtractPatchIntoBufferColumn3D()
382 const int pad_depth = params.padding_values.depth; in Im2col3D() local
440 const int pad_depth = params.padding_values.depth; in DilatedIm2col3D() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dconv3d.h52 const int pad_depth = params.padding_values.depth; in Conv3D() local
H A Dconv3d_transpose.h35 const int pad_depth = params.padding_values.depth; in Conv3DTranspose() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dpooling_ops_common.h81 int pad_depth; member
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_pooling_ops_common.h395 int pad_depth; member