| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | vol2col.cuh | 34 const int dilation_h, in vol2col_kernel()
95 const int dilation_h, in vol2col()
146 const unsigned dilation_h, in vol2im_kernel()
219 const int64_t dilation_h, 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 | 96 const int dilation_h = dilations[1].cast<IntegerAttr>().getInt(); in PadForDynamicShapedInputSamePadding() local
172 const int dilation_h = dilations[1].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 | 49 uint32_t dilation_h, in xnnp_create_convolution2d_nhwc()
|
| H A D | qconv.cpp | 306 const int dilation_h = dilation_[kSpatialDim - 2]; 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 | 64 const int dilation_h = dilations[1].cast<IntegerAttr>().getInt(); in PrepareXlaConvParams() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_convolution_backward.cpp | 118 const int64_t dilation_h = val_at(dilation, 0); in conv2d_backward_impl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_conv_ops.h | 642 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvBackpropCommonOp() local
|
| H A D | mkl_conv_ops.cc | 597 const int64 dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_grad_filter_ops.cc | 291 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in Conv2DBackpropFilterOp() local
|
| H A D | conv_ops.cc | 530 const int64_t dilation_h = GetTensorDim(dilations, data_format, 'H'); in InitConv2DParameters() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_tf.cc | 889 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
980 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
|
| H A D | legalize_tfl.cc | 1037 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
1269 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
|
| H A D | legalize_common.cc | 3360 int64_t dilation_h = dilations_attr[1].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 1235 auto dilation_h = op.dilation_h_factorAttr().getInt(); in inferReturnTypes() local
|