Searched defs:use_dnn_ (Results 1 - 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ | ||
H A D | maxpooling_op.cc | 359 use_dnn_ = CanUseCudnn(); 405 if (use_dnn_) { 423 bool use_dnn_; member in class:tensorflow::MaxPoolingGradOp 730 bool use_dnn_; member in class:tensorflow::MaxPoolingGradGradOp 1053 use_dnn_ = CanUseCudnn(); 1079 if (!is_int8x4 && use_dnn_ && data_format_ == FORMAT_NCHW) { 1110 bool use_dnn_; member in class:tensorflow::MaxPoolingNoMaskOp 1140 use_dnn_ = CanUseCudnn(); 1183 if (use_dnn_ && data_format_ == FORMAT_NCHW) { 1203 bool use_dnn_; member in class:tensorflow::MaxPoolingNoMaskV2Op [all...] |
Completed in 48 milliseconds