| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | vol2col.cuh | 35 const int dilation_w, in vol2col_kernel()
96 const int dilation_w, in vol2col()
147 const unsigned dilation_w, in vol2im_kernel()
220 const int64_t dilation_w, in col2vol()
|
| H A D | DilatedMaxPool2d.cu | 50 const int dilation_h, const int dilation_w, scalar_t* top_data, in max_pool_forward_nchw()
182 const int dilation_h, const int dilation_w, in max_pool_backward_nchw()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/ |
| H A D | tf_to_xla_attribute_utils.cc | 97 const int dilation_w = dilations[2].cast<IntegerAttr>().getInt(); in PadForDynamicShapedInputSamePadding() local
173 const int dilation_w = dilations[2].cast<IntegerAttr>().getInt(); in CalculatePaddingAndPadIfNeeded() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/ |
| H A D | gpu_compatibility.cc | 189 absl::Status CheckDilation(int dilation_h, int dilation_w) { in CheckDilation()
199 int dilation_h, int dilation_w) { in CheckStridesAndDilation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| H A D | ops_defs.py | 54 dilation_w, dilation_h, padding, act): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | XnnpackUtils.h | 50 uint32_t dilation_w, in xnnp_create_convolution2d_nhwc()
|
| H A D | qconv.cpp | 307 const int dilation_w = dilation_[kSpatialDim - 1]; in apply_impl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| H A D | replace_cast_hacks_with_tf_xla_ops.cc | 65 const int dilation_w = dilations[2].cast<IntegerAttr>().getInt(); in PrepareXlaConvParams() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_convolution_backward.cpp | 121 const int64_t dilation_w = val_at(dilation, 1); in conv2d_backward_impl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_conv_ops.h | 643 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvBackpropCommonOp() local
|
| H A D | mkl_conv_ops.cc | 598 const int64 dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_grad_filter_ops.cc | 292 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in Conv2DBackpropFilterOp() local
|
| H A D | conv_ops.cc | 531 const int64_t dilation_w = GetTensorDim(dilations, data_format, 'W'); in InitConv2DParameters() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_tf.cc | 890 int64_t dilation_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
981 int64_t dilation_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
|
| H A D | legalize_tfl.cc | 1038 int64_t dilation_w = tfl_conv2d_op.dilation_w_factor(); in matchAndRewrite() local
1270 int64_t dilation_w = tfl_conv2d_op.dilation_w_factor(); in matchAndRewrite() local
|
| H A D | legalize_common.cc | 3361 int64_t dilation_w = dilations_attr[2].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 1236 auto dilation_w = op.dilation_w_factorAttr().getInt(); in inferReturnTypes() local
|