| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| H A D | Conv.cpp | 318 IntArrayRef dilation_, in get_onednn_conv_sum_attr()
395 IntArrayRef dilation_, in _convolution_out()
543 IntArrayRef dilation_, in _convolution()
569 IntArrayRef dilation_, in convolution_overrideable()
624 IntArrayRef stride_, padding_, dilation_, output_padding_; in convolution_backward_overrideable() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_grad_ops_3d.cc | 289 std::vector<int32> dilation_; member in tensorflow::Conv3DBackpropInputOp
637 std::vector<int32> dilation_; member in tensorflow::Conv3DCustomBackpropInputOp
805 std::vector<int32> dilation_; member in tensorflow::Conv3DBackpropFilterOp
1114 std::vector<int32> dilation_; member in tensorflow::Conv3DCustomBackpropFilterOp
1576 std::vector<int32> dilation_; member in tensorflow::Conv3DBackpropInputOp
1948 std::vector<int32> dilation_; member in tensorflow::Conv3DBackpropFilterOp
|
| H A D | conv_ops_3d.cc | 199 std::vector<int32> dilation_; member in tensorflow::Conv3DOp
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | MaxPooling.cpp | 22 const IntArrayRef dilation_, in use_max_pool2d()
141 const IntArrayRef dilation_, in max_pool2d()
|
| H A D | OpContext.h | 91 std::vector<int64_t> dilation_; variable
122 std::vector<int64_t> dilation_; variable
|
| H A D | Common.h | 39 std::array<int64_t, 2> dilation_; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Convolution.cpp | 1313 SymIntArrayRef stride_, SymIntArrayRef padding_, SymIntArrayRef dilation_, in select_conv_backend()
1470 IntArrayRef stride_, IntArrayRef padding_, IntArrayRef dilation_, in _convolution()
1710 IntArrayRef stride_, IntArrayRef padding_, IntArrayRef dilation_, in _convolution()
1740 IntArrayRef stride_, IntArrayRef padding_, IntArrayRef dilation_, in _convolution_double_backward()
|
| H A D | ConvUtils.h | 229 auto dilation_ = has_dilation ? dilation[d - 2] : 1; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Common.h | 19 std::vector<int64_t> dilation_; member
|
| H A D | OpContext.h | 34 std::vector<int64_t> dilation_; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/ |
| H A D | MetalPrepackOpContext.h | 115 std::vector<int64_t> dilation_; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | utils.h | 178 torch::List<int64_t> dilation_; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | fbgemm_utils.h | 200 torch::List<int64_t> dilation_; member
|
| H A D | QnnpackUtils.h | 301 torch::List<int64_t> dilation_; member
|
| H A D | OnednnUtils.h | 221 torch::List<int64_t> dilation_; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | conv_generic.h | 209 int4 dilation_; variable
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_convolution.cpp | 213 IntArrayRef dilation_ = dilation; in convolution_wrapper() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | Module.cpp | 1919 c10::SymInt groups_) { in initModule()
1954 std::optional<std::vector<c10::SymInt>> bias_sizes_opt) { in initModule()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesConvolution.cpp | 128 IntArrayRef stride_, IntArrayRef padding_, IntArrayRef dilation_, in _convolution_decomp()
|