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) {
// Do a quick copy (sharing the underlying buffer) if both tensors
// are on host memory.
const bool src_host = (send_args.alloc_attrs.on_host() ||
parsed.src.type == "CPU");
const bool dst_host = (recv_args.alloc_attrs.on_host() ||
parsed.dst.type == "CPU");
const bool src_host =
(send_args.alloc_attrs.on_host() || parsed.src.type == "CPU");
const bool dst_host =
(recv_args.alloc_attrs.on_host() || parsed.dst.type == "CPU");
if (src_host && dst_host) {
*out = in;
done(Status::OK());

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

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