Home
last modified time | relevance | path

Searched refs:XLA_ENABLE_XCCL (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_collective_thunk.cc57 #if XLA_ENABLE_XCCL in NcclIsEnabled()
65 #if XLA_ENABLE_XCCL in ExecuteOnStream()
Dnccl_collective_thunk.h35 #define XLA_ENABLE_XCCL 1 macro
39 #if XLA_ENABLE_XCCL
Dnccl_all_gather_thunk.cc72 #if XLA_ENABLE_XCCL in RunNcclCollective()
Dnccl_all_to_all_thunk.cc64 #if XLA_ENABLE_XCCL in RunNcclCollective()
Dnccl_all_reduce_thunk.cc121 #if XLA_ENABLE_XCCL in RunNcclCollective()