Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_collective_thunk.cc50 NcclCollectiveConfig::NcclCollectiveConfig() = default;
51 NcclCollectiveConfig::NcclCollectiveConfig(NcclCollectiveConfig&&) = default;
52 NcclCollectiveConfig::~NcclCollectiveConfig() = default;
53 NcclCollectiveConfig& NcclCollectiveConfig::operator=(NcclCollectiveConfig&&) =
Dnccl_collective_thunk.h61 struct NcclCollectiveConfig { struct
62 NcclCollectiveConfig();
63 NcclCollectiveConfig(NcclCollectiveConfig&&);
64 ~NcclCollectiveConfig();
66 NcclCollectiveConfig& operator=(NcclCollectiveConfig&&); argument
78 NcclCollectiveConfig GetNcclCollectiveConfigForMlir(OpT op, in GetNcclCollectiveConfigForMlir() argument
80 NcclCollectiveConfig config; in GetNcclCollectiveConfigForMlir()
128 virtual const NcclCollectiveConfig& config() const = 0;
Dnccl_all_reduce_thunk.h31 NcclCollectiveConfig config;
51 const NcclCollectiveConfig& config() const override { return config_.config; } in config()
Dnccl_all_to_all_thunk.h31 NcclCollectiveConfig config;
51 const NcclCollectiveConfig& config() const override { return config_.config; } in config()
Dnccl_all_gather_thunk.h31 NcclCollectiveConfig config;
53 const NcclCollectiveConfig& config() const override { return config_.config; } in config()