Home
last modified time | relevance | path

Searched defs:CudaStreamForMPI (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/
Dmpi_ops.cc856 cudaStream_t CudaStreamForMPI() { return mpi_global.stream; } in CudaStreamForMPI() function
/external/tensorflow/tensorflow/contrib/mpi_collectives/
Dmpi_ops.cc857 cudaStream_t CudaStreamForMPI() { return mpi_global.stream; } in CudaStreamForMPI() function