| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cudnn_6_0.inc | 445 … int dilation_w, // filter dilation in the horizontal dimension
452 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
461 … int* dilation_w, // filter dilation in the horizontal dimension
468 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
1718 … int dilation_w, // filter dilation in the horizontal dimension
1723 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
1732 … int dilation_w, // filter dilation in the horizontal dimension
1739 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
1749 … int *dilation_w, // filter dilation in the horizontal dimension
1754 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
[all …]
|
| D | cudnn_7_0.inc | 488 … int dilation_w, /* filter dilation in the horizontal dimension */
495 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
504 … int* dilation_w, /* filter dilation in the horizontal dimension */
511 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
|
| D | cudnn_7_3.inc | 487 int dilation_w, /* filter dilation in the horizontal dimension */
493 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
503 int *dilation_w, /* filter dilation in the horizontal dimension */
509 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
|
| D | cudnn_7_1.inc | 488 … int dilation_w, /* filter dilation in the horizontal dimension */
495 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
504 … int* dilation_w, /* filter dilation in the horizontal dimension */
511 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
|
| D | cudnn_7_4.inc | 487 int dilation_w, /* filter dilation in the horizontal dimension */
493 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
503 int *dilation_w, /* filter dilation in the horizontal 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 | 581 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvBackpropCommonOp() local
587 context, dilation_h > 0 && dilation_w > 0, in MklConvBackpropCommonOp()
|
| D | conv_ops.cc | 319 const int64 dilation_w = GetTensorDim(dilations, data_format, 'W'); in InitConv2DParameters() local
325 dilation_h > 0 && dilation_w > 0, in InitConv2DParameters()
|
| D | conv_grad_filter_ops.cc | 471 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in Conv2DSlowBackpropFilterOp() local
477 context, dilation_h > 0 && dilation_w > 0, in Conv2DSlowBackpropFilterOp()
|
| D | conv_grad_input_ops.cc | 608 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in Conv2DSlowBackpropInputOp() local
614 context, dilation_h > 0 && dilation_w > 0, in Conv2DSlowBackpropInputOp()
|
| D | mkl_conv_ops.cc | 893 const int64 dilation_w = GetTensorDim(dilations_, data_format_, 'W'); 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()
301 tensorflow::hash<int>()(dilation_w)); in GetHashValue()
|