experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Improved code formatting

Browse Source
This commit is contained in:
Benoit Steiner 2016-11-02 20:53:57 -07:00
parent 8d26ac88ef
commit 3e678dc5f8
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/common_runtime/rendezvous_mgr.cc
View File

@ -65,10 +65,10 @@ void IntraProcessRendezvous::SameWorkerRecvDone(
StatusCallback done) { StatusCallback done) {
// Do a quick copy (sharing the underlying buffer) if both tensors // Do a quick copy (sharing the underlying buffer) if both tensors
// are on host memory. // are on host memory.
const bool src_host = (send_args.alloc_attrs.on_host() || const bool src_host =
parsed.src.type == "CPU"); (send_args.alloc_attrs.on_host() || parsed.src.type == "CPU");
const bool dst_host = (recv_args.alloc_attrs.on_host() || const bool dst_host =
parsed.dst.type == "CPU"); (recv_args.alloc_attrs.on_host() || parsed.dst.type == "CPU");
if (src_host && dst_host) { if (src_host && dst_host) {
*out = in; *out = in;
done(Status::OK()); done(Status::OK());

2
tensorflow/core/ops/math_ops.cc
View File

@ -976,7 +976,7 @@ REGISTER_OP("Select")
c->set_output(0, data); c->set_output(0, data);
return Status::OK(); return Status::OK();
}) })
.Doc(R"doc( .Doc(R"doc(
Selects elements from `t` or `e`, depending on `condition`. Selects elements from `t` or `e`, depending on `condition`.