HomeSort by relevance Sort by last modified time
    Searched refs:RingAllreduce (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/
ring.cc 35 // Generate all necessary specializations for RingAllreduce.
36 template Status RingAllreduce<CPUDevice, int>(OpKernelContext*, const Tensor*,
38 template Status RingAllreduce<CPUDevice, long long>(OpKernelContext*,
41 template Status RingAllreduce<CPUDevice, float>(OpKernelContext*, const Tensor*,
ring.cu.cc 56 // Generate all necessary specializations for RingAllreduce.
57 template Status RingAllreduce<GPUDevice, int>(OpKernelContext*, const Tensor*,
59 template Status RingAllreduce<GPUDevice, long long>(OpKernelContext*,
62 template Status RingAllreduce<GPUDevice, float>(OpKernelContext*, const Tensor*,
mpi_ops.cc 54 * RingAllreduce(). The background thread which facilitates MPI operations is
87 extern template Status RingAllreduce<GPUDevice, int>(OpKernelContext*,
90 extern template Status RingAllreduce<GPUDevice, long long>(OpKernelContext*,
93 extern template Status RingAllreduce<GPUDevice, float>(OpKernelContext*,
104 extern template Status RingAllreduce<CPUDevice, int>(OpKernelContext*,
107 extern template Status RingAllreduce<CPUDevice, long long>(OpKernelContext*,
110 extern template Status RingAllreduce<CPUDevice, float>(OpKernelContext*,
490 status = on_gpu ? RingAllreduce<GPU_DEVICE_IF_CUDA, float>(
492 : RingAllreduce<CPUDevice, float>(
495 status = on_gpu ? RingAllreduce<GPU_DEVICE_IF_CUDA, int>
    [all...]
ring.h 156 Status RingAllreduce(OpKernelContext* context, const Tensor* input,

Completed in 131 milliseconds