Searched refs:use_dnn_ (Results 1 – 1 of 1) sorted by relevance
373 use_dnn_ = CanUseCudnn(); in MaxPoolingGradOp()419 if (use_dnn_) { in Compute()437 bool use_dnn_; member in tensorflow::MaxPoolingGradOp744 bool use_dnn_; member in tensorflow::MaxPoolingGradGradOp1140 use_dnn_ = CanUseCudnn(); in MaxPoolingNoMaskOp()1166 if (use_dnn_) { in Compute()1172 if (!is_int8x4 && use_dnn_ && data_format_ == FORMAT_NCHW) { in Compute()1203 bool use_dnn_; in Compute() member in tensorflow::MaxPoolingNoMaskOp1233 use_dnn_ = CanUseCudnn(); in MaxPoolingNoMaskV2Op()1276 if (use_dnn_ && data_format_ == FORMAT_NCHW) { in Compute()[all …]