Searched defs:dilation_depth (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | NaiveConvolutionTranspose3d.cpp | 42 int dilation_depth, in slow_conv_transpose3d_shape_check()
215 int64_t dilation_depth = dilation[0]; in slow_conv_transpose3d_out_cpu_template() local
429 int64_t dilation_depth = dilation[0]; in slow_conv_transpose3d_backward_out_cpu_template() local
634 int64_t dilation_depth = dilation[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 | 42 int dilation_depth, in slow_conv_transpose3d_shape_check()
211 int64_t dilation_depth = dilation[0]; in slow_conv_transpose3d_out_cuda_template() local
435 int64_t dilation_depth = dilation[0]; in slow_conv_transpose3d_backward_out_cuda_template() local
643 int64_t dilation_depth = dilation[0]; in slow_conv_transpose3d_acc_grad_parameters_cuda() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | conv3d_test.cc | 37 int32_t dilation_depth = 1, int32_t dilation_width = 1, in Conv3dOpModel()
57 int32_t dilation_depth = 1, int32_t dilation_width = 1, in Conv3dOpModel()
|
| H A D | conv3d_transpose_test.cc | 44 int32_t dilation_depth = 1, int32_t dilation_width = 1, in Conv3dTransposeOpModel()
75 int32_t dilation_depth = 1, int32_t dilation_width = 1, in Conv3dTransposeOpModel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | convolution.c | 53 uint32_t dilation_depth, in pytorch_qnnp_create_convolution_ndhwc_q8()
612 uint32_t dilation_depth, in pytorch_qnnp_create_convolution3d_ndhwc_q8()
|
| H A D | indirection.c | 40 const size_t dilation_depth = op->dilation_depth; in pytorch_qnnp_indirection_init_conv3d() local
244 const size_t dilation_depth = op->dilation_depth; in pytorch_qnnp_indirection_init_dwconv() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| H A D | operator.h | 70 uint32_t dilation_depth; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | types.h | 727 int dilation_depth; member
|