Home
last modified time | relevance | path

Searched refs:RunCollectiveOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.h244 virtual StatusOr<O> RunCollectiveOp(const I& participant) = 0;
297 TF_ASSIGN_OR_RETURN(O output, RunCollectiveOp(participant)); in SubmitParticipant()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc326 xla::StatusOr<std::nullptr_t> RunCollectiveOp( in RunCollectiveOp() function in __anon95669c100311::CpuAllToAllRendezvous
387 xla::StatusOr<std::nullptr_t> RunCollectiveOp( in RunCollectiveOp() function in __anon95669c100311::CpuCollectivePermuteRendezvous
429 xla::StatusOr<std::nullptr_t> RunCollectiveOp( in RunCollectiveOp() function in __anon95669c100311::CpuAllReduceRendezvous
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_utils.cc227 StatusOr<LockedNcclClique> RunCollectiveOp( in RunCollectiveOp() function in xla::gpu::__anonf15227f90111::NcclCliqueRendezvous