Home
last modified time | relevance | path

Searched refs:dilation_h (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
Dtf_to_xla_attribute_utils.cc96 const int dilation_h = dilations[1].cast<IntegerAttr>().getInt(); in PadForDynamicShapedInputSamePadding() local
105 GetSamePaddingValues(builder, loc, input_size_h, filter_h, dilation_h, in PadForDynamicShapedInputSamePadding()
172 const int dilation_h = dilations[1].cast<IntegerAttr>().getInt(); in CalculatePaddingAndPadIfNeeded() local
175 stride_h, stride_w, dilation_h, dilation_w, in CalculatePaddingAndPadIfNeeded()
/external/tensorflow/tensorflow/lite/tools/versioning/
Dgpu_compatibility.cc189 absl::Status CheckDilation(int dilation_h, int dilation_w) { in CheckDilation() argument
190 if (dilation_h <= 0 || dilation_w <= 0) { in CheckDilation()
192 "Incorrect dilation values: dilation_height = ", dilation_h, in CheckDilation()
199 int dilation_h, int dilation_w) { in CheckStridesAndDilation() argument
201 RETURN_IF_ERROR(CheckDilation(dilation_h, dilation_w)); in CheckStridesAndDilation()
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dreplace_cast_hacks_with_tf_xla_ops.cc64 const int dilation_h = dilations[1].cast<IntegerAttr>().getInt(); in PrepareXlaConvParams() local
68 Create1DConstValue<int32_t>(builder, loc, {dilation_h, dilation_w}); in PrepareXlaConvParams()
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dops_defs.py54 dilation_w, dilation_h, padding, act): argument
59 dilations=[1, dilation_w, dilation_h, 1],
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcudnn_6_0.inc451 int dilation_h, // filter dilation in the vertical dimension
459 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
469 int *dilation_h, // filter dilation in the vertical dimension
477 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1757 int dilation_h, // filter dilation in the vertical dimension
1766 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
1775 int dilation_h, // filter dilation in the vertical dimension
1784 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1794 int *dilation_h, // filter dilation in the vertical dimension
1803 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
[all …]
Dcudnn_7_0.inc495 int dilation_h, /* filter dilation in the vertical dimension */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
513 int *dilation_h, /* filter dilation in the vertical dimension */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_3.inc495 int dilation_h, /* filter dilation in the vertical dimension */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
513 int *dilation_h, /* filter dilation in the vertical dimension */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_1.inc495 int dilation_h, /* filter dilation in the vertical dimension */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
513 int *dilation_h, /* filter dilation in the vertical dimension */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_4.inc495 int dilation_h, /* filter dilation in the vertical dimension */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
513 int *dilation_h, /* filter dilation in the vertical dimension */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_6.inc660 int dilation_h, /* filter dilation in the vertical dimension */
668 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
678 int *dilation_h, /* filter dilation in the vertical dimension */
686 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_8_0.inc1101 int dilation_h, /* filter dilation in the vertical dimension */
1109 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1119 int *dilation_h, /* filter dilation in the vertical dimension */
1127 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.h385 int dilation_h = GetTensorDim(dilations_, data_format_, 'H');
392 context, dilation_h > 0 && dilation_w > 0,
413 context, (dilation_h == 1 && dilation_w == 1),
Dconv_grad_filter_ops.cc291 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in Conv2DBackpropFilterOp() local
298 context, dilation_h > 0 && dilation_w > 0, in Conv2DBackpropFilterOp()
317 context, (dilation_h == 1 && dilation_w == 1), in Conv2DBackpropFilterOp()
Dconv_ops.cc530 const int64_t dilation_h = GetTensorDim(dilations, data_format, 'H'); in InitConv2DParameters() local
537 dilation_h > 0 && dilation_w > 0, in InitConv2DParameters()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h642 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvBackpropCommonOp() local
649 context, dilation_h > 0 && dilation_w > 0, in MklConvBackpropCommonOp()
Dmkl_conv_ops.cc597 const int64 dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvOp() local
604 context, dilation_h > 0 && dilation_w > 0, in MklConvOp()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc889 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
891 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
980 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
983 if (dilation_h != 1 || dilation_w != 1) return failure(); in matchAndRewrite()
Dlegalize_tfl.cc1037 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
1039 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
1269 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
1271 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
Dlegalize_common.cc3360 int64_t dilation_h = dilations_attr[1].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon() local
3362 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in convertTFConv2DCommon()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1235 auto dilation_h = op.dilation_h_factorAttr().getInt(); in inferReturnTypes() local
1260 input_ty.getDimSize(1), filter_ty.getDimSize(1), dilation_h, in inferReturnTypes()