Home
last modified time | relevance | path

Searched refs:use_dnn_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op.cc373 use_dnn_ = CanUseCudnn(); in MaxPoolingGradOp()
419 if (use_dnn_) { in Compute()
437 bool use_dnn_; member in tensorflow::MaxPoolingGradOp
744 bool use_dnn_; member in tensorflow::MaxPoolingGradGradOp
1140 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::MaxPoolingNoMaskOp
1233 use_dnn_ = CanUseCudnn(); in MaxPoolingNoMaskV2Op()
1276 if (use_dnn_ && data_format_ == FORMAT_NCHW) { in Compute()
[all …]