diff options
| author |  Justin Lebar <jlebar@google.com> | 2018-04-23 17:16:55 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-23 17:19:22 -0700 |
| commit | 80fc661853f9a0844faf95eb68438dc85a5879e3 (patch) | |
| tree | e7328a367c4f4db472063f96c983b70e63aa3f26 /tensorflow/contrib/mpi_collectives | |
| parent | ecd837fd0ab69cf54d920eae3b1c73602be6c626 (diff) | |
Use tensorflow::se instead of perftools::gputools for StreamExecutor.
PiperOrigin-RevId: 194010749
Diffstat (limited to 'tensorflow/contrib/mpi_collectives')
| -rw-r--r-- | tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc | 2 | ||||
| -rw-r--r-- | tensorflow/contrib/mpi_collectives/mpi_ops.cc | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc b/tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc index 8dca90a1e3..ed22ee667f 100644 --- a/tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc +++ b/tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc @@ -73,7 +73,7 @@ limitations under the License. */ template <class T> -using StatusOr = perftools::gputools::port::StatusOr<T>; +using StatusOr = se::port::StatusOr<T>; using CPUDevice = Eigen::ThreadPoolDevice; using GPUDevice = Eigen::GpuDevice; diff --git a/tensorflow/contrib/mpi_collectives/mpi_ops.cc b/tensorflow/contrib/mpi_collectives/mpi_ops.cc index a051ab0004..475297ca92 100644 --- a/tensorflow/contrib/mpi_collectives/mpi_ops.cc +++ b/tensorflow/contrib/mpi_collectives/mpi_ops.cc @@ -74,7 +74,7 @@ limitations under the License. */ template <class T> -using StatusOr = perftools::gputools::port::StatusOr<T>; +using StatusOr = se::port::StatusOr<T>; using CPUDevice = Eigen::ThreadPoolDevice; using GPUDevice = Eigen::GpuDevice; |