HomeSort by relevance Sort by last modified time
    Searched refs:use_cudnn_on_gpu (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/cc/gradients/
nn_grad.cc 127 bool use_cudnn_on_gpu; local
132 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "use_cudnn_on_gpu", &use_cudnn_on_gpu));
135 input_attrs.UseCudnnOnGpu(use_cudnn_on_gpu);
142 filter_attrs.UseCudnnOnGpu(use_cudnn_on_gpu);
  /external/tensorflow/tensorflow/python/ops/
nn_grad.py 53 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
61 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
76 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
84 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
502 use_cudnn_on_gpu = op.get_attr("use_cudnn_on_gpu")
    [all...]
nn_ops.py     [all...]
  /external/tensorflow/tensorflow/contrib/quantize/python/
fold_batch_norms.py 386 use_cudnn_on_gpu=layer_op.get_attr('use_cudnn_on_gpu'),
684 use_cudnn_on_gpu=op.get_attr('use_cudnn_on_gpu'),
    [all...]
  /external/tensorflow/tensorflow/core/graph/
mkl_layout_pass.cc 1475 bool use_cudnn_on_gpu; local
    [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 273 milliseconds