Home
last modified time | relevance | path

Searched defs:CollectivePermuteParticipantData (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc141 struct CollectivePermuteParticipantData : xla::ParticipantData { struct
142 CollectivePermuteParticipantData(const xla::RendezvousKey& rendezvous_key_p, in CollectivePermuteParticipantData() function
149 xla::int64 device_ordinal;
150 se::Stream* stream;
151 int replica_id;
152 se::DeviceMemoryBase source_data;
153 se::DeviceMemoryBase destination_data;
154 xla::int64 byte_size;
155 std::vector<int> replica_ids_to_copy_to;
157 std::string ToString() const override { in ToString()
[all …]