Searched refs:use_cudnn_grouped_conv_ (Results 1 – 2 of 2) sorted by relevance
310 use_cudnn_grouped_conv_ = in DepthwiseConv2dNativeOp()319 use_cudnn_grouped_conv_ = dtype_ == DT_HALF && data_format_ == FORMAT_NCHW; in DepthwiseConv2dNativeOp()321 use_cudnn_grouped_conv_ = false; in DepthwiseConv2dNativeOp()410 (use_cudnn_grouped_conv_ && in Compute()474 bool use_cudnn_grouped_conv_; member in tensorflow::DepthwiseConv2dNativeOp521 this->use_cudnn_grouped_conv_ = true; in DepthwiseConv2dGroupedConvOp()
591 use_cudnn_grouped_conv_ = in DepthwiseConv2dNativeBackpropInputOp()600 use_cudnn_grouped_conv_ = dtype_ == DT_HALF && in DepthwiseConv2dNativeBackpropInputOp()604 use_cudnn_grouped_conv_ = false; in DepthwiseConv2dNativeBackpropInputOp()641 (use_cudnn_grouped_conv_ && in Compute()691 bool use_cudnn_grouped_conv_; member in tensorflow::DepthwiseConv2dNativeBackpropInputOp742 this->use_cudnn_grouped_conv_ = true; in DepthwiseConv2dGroupedConvBackpropInputOp()1100 use_cudnn_grouped_conv_ = dtype_ == DT_HALF; in DepthwiseConv2dNativeBackpropFilterOp()1102 use_cudnn_grouped_conv_ = false; in DepthwiseConv2dNativeBackpropFilterOp()1138 (use_cudnn_grouped_conv_ && in Compute()1222 bool use_cudnn_grouped_conv_; member in tensorflow::DepthwiseConv2dNativeBackpropFilterOp[all …]