Searched refs:use_cudnn_grouped_conv_ (Results 1 – 2 of 2) sorted by relevance
295 use_cudnn_grouped_conv_ = false; in DepthwiseConv2dNativeOp()375 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute()435 bool use_cudnn_grouped_conv_; member in tensorflow::DepthwiseConv2dNativeOp482 this->use_cudnn_grouped_conv_ = true; in DepthwiseConv2dGroupedConvOp()
569 use_cudnn_grouped_conv_ = false; in DepthwiseConv2dNativeBackpropInputOp()604 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute()650 bool use_cudnn_grouped_conv_; member in tensorflow::DepthwiseConv2dNativeBackpropInputOp701 this->use_cudnn_grouped_conv_ = true; in DepthwiseConv2dGroupedConvBackpropInputOp()1043 use_cudnn_grouped_conv_ = false; in DepthwiseConv2dNativeBackpropFilterOp()1086 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute()1133 bool use_cudnn_grouped_conv_; member in tensorflow::DepthwiseConv2dNativeBackpropFilterOp1183 this->use_cudnn_grouped_conv_ = true; in DepthwiseConv2dGroupedConvBackpropFilterOp()