Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
Dtf_to_xla_attribute_utils.cc97 const int dilation_w = dilations[2].cast<IntegerAttr>().getInt(); in PadForDynamicShapedInputSamePadding() local
109 GetSamePaddingValues(builder, loc, input_size_w, filter_w, dilation_w, in PadForDynamicShapedInputSamePadding()
173 const int dilation_w = dilations[2].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()
193 ", dilation_width = ", dilation_w)); 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.cc65 const int dilation_w = dilations[2].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.inc452 int dilation_w, // filter dilation in the horizontal dimension
459 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
470 int *dilation_w, // filter dilation in the horizontal dimension
477 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1758 int dilation_w, // filter dilation in the horizontal dimension
1766 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
1776 int dilation_w, // filter dilation in the horizontal dimension
1784 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1795 int *dilation_w, // filter dilation in the horizontal dimension
1803 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
[all …]
Dcudnn_7_0.inc496 int dilation_w, /* filter dilation in the horizontal dimension */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
514 int *dilation_w, /* filter dilation in the horizontal dimension */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_3.inc496 int dilation_w, /* filter dilation in the horizontal dimension */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
514 int *dilation_w, /* filter dilation in the horizontal dimension */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_1.inc496 int dilation_w, /* filter dilation in the horizontal dimension */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
514 int *dilation_w, /* filter dilation in the horizontal dimension */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_4.inc496 int dilation_w, /* filter dilation in the horizontal dimension */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
514 int *dilation_w, /* filter dilation in the horizontal dimension */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_6.inc661 int dilation_w, /* filter dilation in the horizontal dimension */
668 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
679 int *dilation_w, /* filter dilation in the horizontal dimension */
686 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_8_0.inc1102 int dilation_w, /* filter dilation in the horizontal dimension */
1109 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1120 int *dilation_w, /* filter dilation in the horizontal 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.h386 int dilation_w = GetTensorDim(dilations_, data_format_, 'W');
392 context, dilation_h > 0 && dilation_w > 0,
413 context, (dilation_h == 1 && dilation_w == 1),
Dconv_grad_filter_ops.cc292 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); 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.cc531 const int64_t dilation_w = GetTensorDim(dilations, data_format, 'W'); in InitConv2DParameters() local
537 dilation_h > 0 && dilation_w > 0, in InitConv2DParameters()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h643 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvBackpropCommonOp() local
649 context, dilation_h > 0 && dilation_w > 0, in MklConvBackpropCommonOp()
Dmkl_conv_ops.cc598 const int64 dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvOp() local
604 context, dilation_h > 0 && dilation_w > 0, in MklConvOp()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc890 int64_t dilation_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
891 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
981 int64_t dilation_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
983 if (dilation_h != 1 || dilation_w != 1) return failure(); in matchAndRewrite()
Dlegalize_tfl.cc1038 int64_t dilation_w = tfl_conv2d_op.dilation_w_factor(); in matchAndRewrite() local
1039 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
1270 int64_t dilation_w = tfl_conv2d_op.dilation_w_factor(); in matchAndRewrite() local
1271 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
Dlegalize_common.cc3361 int64_t dilation_w = dilations_attr[2].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.cc1236 auto dilation_w = op.dilation_w_factorAttr().getInt(); in inferReturnTypes() local
1271 input_ty.getDimSize(2), filter_ty.getDimSize(2), dilation_w, in inferReturnTypes()