| /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 | 384 int dilation_h = GetTensorDim(dilations_, data_format_, 'H');
391 context, dilation_h > 0 && dilation_w > 0,
412 context, (dilation_h == 1 && dilation_w == 1),
|
| D | conv_grad_filter_ops.cc | 287 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); 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 | 385 const int64 dilation_h = GetTensorDim(dilations, data_format, 'H'); in InitConv2DParameters() local
392 dilation_h > 0 && dilation_w > 0, in InitConv2DParameters()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cudnn_6_0.inc | 451 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 …]
|
| D | cudnn_7_0.inc | 495 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,
|
| D | cudnn_7_3.inc | 495 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,
|
| D | cudnn_7_1.inc | 495 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,
|
| D | cudnn_7_4.inc | 495 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,
|
| D | cudnn_8_0.inc | 1101 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,
|
| D | cudnn_7_6.inc | 660 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,
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 587 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvBackpropCommonOp() local
594 context, dilation_h > 0 && dilation_w > 0, in MklConvBackpropCommonOp()
|
| D | mkl_conv_ops.cc | 469 const int64 dilation_h = GetTensorDim(dilations_, data_format_, 'H'); 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()
88 "Incorrect dilation values: dilation_factor = ", dilation_h, 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 | 886 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
888 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
977 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
979 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
|
| D | legalize_tfl.cc | 1136 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
1138 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
1352 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
1354 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in matchAndRewrite()
|
| D | legalize_common.cc | 2830 int64_t dilation_h = dilations_attr[1].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon() local
2832 dilation = rewriter.getI64ArrayAttr({dilation_h, dilation_w}); in convertTFConv2DCommon()
|