| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cudnn_6_0.inc | 444 … int dilation_h, // filter dilation in the vertical dimension
452 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
460 … int* dilation_h, // filter dilation in the vertical dimension
468 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
1717 … int dilation_h, // filter dilation in the vertical dimension
1723 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
1731 … int dilation_h, // filter dilation in the vertical dimension
1739 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
1748 … int *dilation_h, // filter dilation in the vertical dimension
1754 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
[all …]
|
| D | cudnn_7_0.inc | 487 … int dilation_h, /* filter dilation in the vertical dimension */
495 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
503 … int* dilation_h, /* filter dilation in the vertical dimension */
511 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
|
| D | cudnn_7_3.inc | 486 int dilation_h, /* filter dilation in the vertical dimension */
493 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
502 int *dilation_h, /* filter dilation in the vertical dimension */
509 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
|
| D | cudnn_7_1.inc | 487 … int dilation_h, /* filter dilation in the vertical dimension */
495 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
503 … int* dilation_h, /* filter dilation in the vertical dimension */
511 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
|
| D | cudnn_7_4.inc | 486 int dilation_h, /* filter dilation in the vertical dimension */
493 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
502 int *dilation_h, /* filter dilation in the vertical dimension */
509 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.h | 580 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvBackpropCommonOp() local
587 context, dilation_h > 0 && dilation_w > 0, in MklConvBackpropCommonOp()
|
| D | conv_ops.cc | 318 const int64 dilation_h = GetTensorDim(dilations, data_format, 'H'); in InitConv2DParameters() local
325 dilation_h > 0 && dilation_w > 0, in InitConv2DParameters()
|
| D | conv_grad_filter_ops.cc | 470 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in Conv2DSlowBackpropFilterOp() local
477 context, dilation_h > 0 && dilation_w > 0, in Conv2DSlowBackpropFilterOp()
|
| D | conv_grad_input_ops.cc | 607 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in Conv2DSlowBackpropInputOp() local
614 context, dilation_h > 0 && dilation_w > 0, in Conv2DSlowBackpropInputOp()
|
| D | mkl_conv_ops.cc | 892 const int64 dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvOp() local
899 context, dilation_h > 0 && dilation_w > 0, in MklConvOp()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 285 int pad_h = 0, pad_w = 0, u = 0, v = 0, dilation_h = 0, dilation_w = 0; in GetHashValue() local
287 &v, &dilation_h, &dilation_w); in GetHashValue()
299 tensorflow::hash<int>()(dilation_h)); in GetHashValue()
|