| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 58 dilation_w, dilation_h, padding, act): argument
63 dilations=[1, dilation_w, dilation_h, 1],
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.h | 385 int dilation_w = GetTensorDim(dilations_, data_format_, 'W');
391 context, dilation_h > 0 && dilation_w > 0,
412 context, (dilation_h == 1 && dilation_w == 1),
|
| D | conv_grad_filter_ops.cc | 288 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in Conv2DBackpropFilterOp() local
294 context, dilation_h > 0 && dilation_w > 0, in Conv2DBackpropFilterOp()
313 context, (dilation_h == 1 && dilation_w == 1), in Conv2DBackpropFilterOp()
|
| D | conv_ops.cc | 386 const int64 dilation_w = GetTensorDim(dilations, data_format, 'W'); in InitConv2DParameters() local
392 dilation_h > 0 && dilation_w > 0, in InitConv2DParameters()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cudnn_6_0.inc | 452 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 …]
|
| D | cudnn_7_0.inc | 496 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,
|
| D | cudnn_7_3.inc | 496 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,
|
| D | cudnn_7_1.inc | 496 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,
|
| D | cudnn_7_4.inc | 496 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,
|
| D | cudnn_8_0.inc | 1102 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,
|
| D | cudnn_7_6.inc | 661 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,
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 588 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvBackpropCommonOp() local
594 context, dilation_h > 0 && dilation_w > 0, in MklConvBackpropCommonOp()
|
| D | mkl_conv_ops.cc | 470 const int64 dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvOp() local
476 context, dilation_h > 0 && dilation_w > 0, in MklConvOp()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 85 absl::Status CheckDilation(int dilation_h, int dilation_w) { in CheckDilation() argument
86 if (dilation_h <= 0 || dilation_w <= 0) { in CheckDilation()
89 ", dilation_factor = ", dilation_w)); in CheckDilation()
95 int dilation_h, int dilation_w) { in CheckStridesAndDilation() argument
97 RETURN_IF_ERROR(CheckDilation(dilation_h, dilation_w)); in CheckStridesAndDilation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 887 int64_t dilation_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
888 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
978 int64_t dilation_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
979 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
|
| D | legalize_tfl.cc | 1137 int64_t dilation_w = tfl_conv2d_op.dilation_w_factor(); in matchAndRewrite() local
1138 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
1353 int64_t dilation_w = tfl_conv2d_op.dilation_w_factor(); in matchAndRewrite() local
1354 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
|
| D | legalize_common.cc | 2831 int64_t dilation_w = dilations_attr[2].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon() local
2832 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in convertTFConv2DCommon()
|