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
tensorflow/core/common_runtime
rendezvous_mgr.cc

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());