Searched refs:use_cudnn_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.h | 400 OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_));
454 launch(context, use_cudnn_, cudnn_use_autotune_, out_backprop, filter,
466 bool use_cudnn_ = false;
|
| D | conv_ops_fused_impl.h | 723 OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_));
796 LaunchFusedConv2DOp<Device, T>()(context, use_cudnn_, cudnn_use_autotune_,
804 bool use_cudnn_;
|
| D | conv_grad_filter_ops.cc | 303 OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_)); in Conv2DBackpropFilterOp()
361 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, input, in Compute()
371 bool use_cudnn_; member in tensorflow::Conv2DBackpropFilterOp
|
| D | conv_ops.cc | 514 OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_)); in Conv2DOp()
583 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, filter, in Compute()
591 bool use_cudnn_; member in tensorflow::Conv2DOp
|