Searched refs:destination_buffers (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 165 std::vector<se::DeviceMemoryBase> destination_buffers; member
183 absl::StrJoin(destination_buffers, ", ", addr_formatter)); in ToString()
342 CHECK_EQ(p.source_buffers.size(), p.destination_buffers.size()); in RunCollectiveOp()
345 CHECK_EQ(p.destination_buffers[i].size(), expected_buffer_size); in RunCollectiveOp()
371 std::memcpy(receiver.destination_buffers[rank].opaque(), in RunCollectiveOp()
622 xla::int64 buffer_size, void** source_buffers, void** destination_buffers) { in __xla_cpu_runtime_AllToAll() argument
644 participant.destination_buffers.emplace_back(destination_buffers[i], in __xla_cpu_runtime_AllToAll()
|
| D | cpu_runtime.h | 190 xla::int64 buffer_size, void** source_buffers, void** destination_buffers);
|
| /external/python/cpython3/Lib/test/ |
| D | test_clinic.py | 48 self.destination_buffers = {}
|
| /external/python/cpython3/Tools/clinic/ |
| D | clinic.py | 1341 for name, destination in clinic.destination_buffers.items():
1874 self.destination_buffers = collections.OrderedDict((
1908 for name in self.destination_buffers:
1912 assert name in self.destination_buffers
3988 fd = self.clinic.destination_buffers
|