Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_collective_thunk.h95 config.collective_op_kind = RendezvousKey::kCrossReplica; in GetNcclCollectiveConfigForMlir()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.h78 kCrossReplica, enumerator
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc604 : xla::RendezvousKey::kCrossReplica; in GetRendezvousKey()