Searched defs:use_dnn_ (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dmaxpooling_op.cc359 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