| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ConvolutionMM3d.cpp | 37 const int64_t kernel_depth = kernel_size[0]; in compute_columns3d() local
115 int64_t kernel_depth, in slow_conv3d_shape_check()
269 int64_t kernel_depth, in slow_conv3d_update_output_frame()
317 int64_t kernel_depth, in slow_conv3d_backward_update_grad_input_frame()
379 const int64_t kernel_depth = kernel_size[0]; in slow_conv3d_backward_out_cpu_template() local
503 const int64_t kernel_depth = kernel_size[0]; in slow_conv3d_backward_parameters_out_cpu_template() local
564 const int64_t kernel_depth = kernel_size[0]; in slow_conv3d_forward_out_cpu() local
|
| H A D | NaiveConvolutionTranspose3d.cpp | 33 int kernel_depth, in slow_conv_transpose3d_shape_check()
212 int64_t kernel_depth = kernel_size[0]; in slow_conv_transpose3d_out_cpu_template() local
426 int64_t kernel_depth = kernel_size[0]; in slow_conv_transpose3d_backward_out_cpu_template() local
631 int64_t kernel_depth = kernel_size[0]; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | NaiveConvolutionTranspose3d.cu | 33 int kernel_depth, in slow_conv_transpose3d_shape_check()
208 int64_t kernel_depth = kernel_size[0]; in slow_conv_transpose3d_out_cuda_template() local
432 int64_t kernel_depth = kernel_size[0]; in slow_conv_transpose3d_backward_out_cuda_template() local
640 int64_t kernel_depth = kernel_size[0]; in slow_conv_transpose3d_acc_grad_parameters_cuda() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | convolution.c | 47 uint32_t kernel_depth, in pytorch_qnnp_create_convolution_ndhwc_q8()
606 uint32_t kernel_depth, in pytorch_qnnp_create_convolution3d_ndhwc_q8()
759 const size_t kernel_depth = convolution->kernel_depth; in pytorch_qnnp_setup_convolution_ndhwc_q8() local
|
| H A D | indirection.c | 34 const size_t kernel_depth = op->kernel_depth; in pytorch_qnnp_indirection_init_conv3d() local
238 const size_t kernel_depth = op->kernel_depth; in pytorch_qnnp_indirection_init_dwconv() local
460 const size_t kernel_depth = in pytorch_qnnp_indirection_set_step_dimensions() local
|
| H A D | conv-prepack.cc | 19 const uint32_t kernel_depth = in PrePackConvWeights() local
|
| H A D | operator-run.c | 902 const size_t kernel_depth = op->kernel_depth; in pytorch_qnnp_run_operator() local
1293 const size_t kernel_depth = op->kernel_depth; in pytorch_qnnp_run_operator() local
|
| H A D | conv-run.cc | 354 const size_t kernel_depth = convolution->kernel_depth; in qnnpackConv() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | DirectConvolution3DFixture.h | 161 …z, int pad_x, int pad_y, int pad_z, unsigned int kernel_width, int kernel_height, int kernel_depth, in setup()
174 …z, int pad_x, int pad_y, int pad_z, unsigned int kernel_width, int kernel_height, int kernel_depth, in setup()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuWeightsReshapeKernel.cpp | 117 const unsigned int kernel_depth = src->info()->dimension(2); in run_op() local
|
| H A D | CpuIm2ColKernel.cpp | 90 int kernel_depth, in linearize_volume_nchw()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/ |
| H A D | mp8x27-sse2.c | 44 const size_t kernel_depth = 3; in pytorch_q8dwconv_ukernel_mp8x27__sse2() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | QnnpackUtils.h | 159 const size_t kernel_depth = kSpatialDim == 3 ? kernel_[0] : 1; in PackedConvWeightsQnnp() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| H A D | operator.h | 64 uint32_t kernel_depth; member
|
| /aosp_15_r20/external/ComputeLibrary/src/core/ |
| H A D | Utils.cpp | 493 … int kernel_width, int kernel_height, int kernel_depth, in scaled_3d_dimensions_signed()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 1622 int32_t kernel_depth = GetTensorDim(kernel_sizes, data_format, 'C'); in MaxPoolShapeImpl() local
1761 int32_t kernel_depth = GetTensorDim(kernel_sizes, data_format, 'C'); in MaxPoolV2Shape() local
|