HomeSort by relevance Sort by last modified time
    Searched defs:use_cudnn (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
depthwise_conv_op.cc 38 #include "tensorflow/core/util/use_cudnn.h"
375 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); variable
384 << ", Use cuDNN: " << use_cudnn; variable
386 if (use_cudnn) {
depthwise_conv_grad_op.cc 37 #include "tensorflow/core/util/use_cudnn.h"
604 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); variable
613 << ", Use cuDNN: " << use_cudnn; variable
615 if (use_cudnn) {
1086 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); variable
1095 << ", Use cuDNN: " << use_cudnn; variable
    [all...]

Completed in 424 milliseconds