Home
last modified time | relevance | path

Searched refs:dilation_c (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.h579 int dilation_c = GetTensorDim(dilations_, data_format_, 'C'); in MklConvBackpropCommonOp() local
582 OP_REQUIRES(context, (dilation_n == 1 && dilation_c == 1), in MklConvBackpropCommonOp()
Dconv_ops.cc317 const int64 dilation_c = GetTensorDim(dilations, data_format, 'C'); in InitConv2DParameters() local
321 dilation_n == 1 && dilation_c == 1, in InitConv2DParameters()
Dconv_grad_filter_ops.cc469 int dilation_c = GetTensorDim(dilations_, data_format_, 'C'); in Conv2DSlowBackpropFilterOp() local
472 OP_REQUIRES(context, dilation_n == 1 && dilation_c == 1, in Conv2DSlowBackpropFilterOp()
Dconv_grad_input_ops.cc606 int dilation_c = GetTensorDim(dilations_, data_format_, 'C'); in Conv2DSlowBackpropInputOp() local
609 OP_REQUIRES(context, (dilation_n == 1 && dilation_c == 1), in Conv2DSlowBackpropInputOp()
Dmkl_conv_ops.cc891 const int64 dilation_c = GetTensorDim(dilations_, data_format_, 'C'); in MklConvOp() local
894 OP_REQUIRES(context, dilation_n == 1 && dilation_c == 1, in MklConvOp()