Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.h578 int dilation_n = GetTensorDim(dilations_, data_format_, 'N'); in MklConvBackpropCommonOp() local
582 OP_REQUIRES(context, (dilation_n == 1 && dilation_c == 1), in MklConvBackpropCommonOp()
Dconv_ops.cc316 const int64 dilation_n = GetTensorDim(dilations, data_format, 'N'); in InitConv2DParameters() local
321 dilation_n == 1 && dilation_c == 1, in InitConv2DParameters()
Dconv_grad_filter_ops.cc468 int dilation_n = GetTensorDim(dilations_, data_format_, 'N'); in Conv2DSlowBackpropFilterOp() local
472 OP_REQUIRES(context, dilation_n == 1 && dilation_c == 1, in Conv2DSlowBackpropFilterOp()
Dconv_grad_input_ops.cc605 int dilation_n = GetTensorDim(dilations_, data_format_, 'N'); in Conv2DSlowBackpropInputOp() local
609 OP_REQUIRES(context, (dilation_n == 1 && dilation_c == 1), in Conv2DSlowBackpropInputOp()
Dmkl_conv_ops.cc890 const int64 dilation_n = GetTensorDim(dilations_, data_format_, 'N'); in MklConvOp() local
894 OP_REQUIRES(context, dilation_n == 1 && dilation_c == 1, in MklConvOp()