diff options
| author |  Olivia Nordquist <nolivia@google.com> | 2017-07-18 11:34:24 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-18 11:40:59 -0700 |
| commit | 9867119831afa821aa53206cc5b7ae773db4284d (patch) | |
| tree | a50ffe4e4e06d8b932721b7d501c39e7b3c6b559 /tensorflow/core/graph/graph.cc | |
| parent | 1558c10c8d07c78a58d74d0fcf53c1d2e2505d5e (diff) | |
Implementing set_device for the C API
PiperOrigin-RevId: 162379684
Diffstat (limited to 'tensorflow/core/graph/graph.cc')
| -rw-r--r-- | tensorflow/core/graph/graph.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/graph/graph.cc b/tensorflow/core/graph/graph.cc index a840ef39d2..f6586f0519 100644 --- a/tensorflow/core/graph/graph.cc +++ b/tensorflow/core/graph/graph.cc @@ -184,6 +184,11 @@ void Node::ClearAttr(const string& name) { (*props_->node_def.mutable_attr()).erase(name); } +void Node::set_requested_device(const string& device) { + MaybeCopyOnWrite(); + props_->node_def.set_device(device); +} + Status Node::input_edge(int idx, const Edge** e) const { if (idx < 0 || idx >= num_inputs()) { return errors::InvalidArgument("Invalid input_edge index: ", idx, ", Node ", |