diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-08-28 22:33:32 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-08-28 22:37:40 -0700 |
commit | c32d401c51ce004dcd33811e7c9027bf259f38a0 (patch) | |
tree | 6f30efd50f4d393697e3da77c9b5e6f1333d6056 /tensorflow/contrib/nccl | |
parent | 5396e8ee172da2d3b5e20fae775facf91cd0490c (diff) |
Optimize convert_to_tensor.
PiperOrigin-RevId: 166805061
Diffstat (limited to 'tensorflow/contrib/nccl')
-rw-r--r-- | tensorflow/contrib/nccl/kernels/nccl_manager.h | 2 | ||||
-rw-r--r-- | tensorflow/contrib/nccl/kernels/nccl_ops.cc | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/nccl/kernels/nccl_manager.h b/tensorflow/contrib/nccl/kernels/nccl_manager.h index 1a661e8f7f..6e2f8e953a 100644 --- a/tensorflow/contrib/nccl/kernels/nccl_manager.h +++ b/tensorflow/contrib/nccl/kernels/nccl_manager.h @@ -20,7 +20,7 @@ limitations under the License. #include <unordered_map> #include <vector> -#include "external/nccl_archive/src/nccl.h" +#include "src/nccl.h" #include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h" #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/platform/mutex.h" diff --git a/tensorflow/contrib/nccl/kernels/nccl_ops.cc b/tensorflow/contrib/nccl/kernels/nccl_ops.cc index 3c532e3d73..d4455483f7 100644 --- a/tensorflow/contrib/nccl/kernels/nccl_ops.cc +++ b/tensorflow/contrib/nccl/kernels/nccl_ops.cc @@ -18,7 +18,7 @@ limitations under the License. #include <unordered_map> #include <vector> -#include "external/nccl_archive/src/nccl.h" +#include "src/nccl.h" #include "tensorflow/contrib/nccl/kernels/nccl_manager.h" #include "tensorflow/core/framework/op_kernel.h" |