| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | infeed_thunk.cc | 38 ShapeTree<se::ScopedDeviceMemory<uint8_t>> source_buffers = in ExecuteOnStream() local
42 for (auto& source : source_buffers.leaves()) { in ExecuteOnStream()
47 ShapeUtil::GetSubshape(source_buffers.shape(), shape_index); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 206 std::vector<se::DeviceMemoryBase> source_buffers; member
229 absl::StrJoin(source_buffers, ", ", addr_formatter), in ToString()
398 CHECK(!participants_[0].source_buffers.empty()); in RunCollectiveOp()
399 int expected_buffer_size = participants_[0].source_buffers[0].size(); in RunCollectiveOp()
406 CHECK_EQ(p.source_buffers.size(), p.destination_buffers.size()); in RunCollectiveOp()
407 CHECK_EQ(p.source_buffers.size(), participants_.size()); in RunCollectiveOp()
408 for (int i = 0; i < p.source_buffers.size(); i++) { in RunCollectiveOp()
410 CHECK_EQ(p.source_buffers[i].size(), expected_buffer_size); in RunCollectiveOp()
436 sender.source_buffers[i].opaque(), expected_buffer_size); in RunCollectiveOp()
722 void** source_buffers, void** destination_buffers) { in __xla_cpu_runtime_AllToAll() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis.cc | 629 const PointsToSet::BufferList& source_buffers) { in GatherBuffersDefinedByInstruction() argument
631 CHECK(!source_buffers.empty()); in GatherBuffersDefinedByInstruction()
632 if (source_buffers.size() == 1 && in GatherBuffersDefinedByInstruction()
633 source_buffers[0]->instruction() == instruction) { in GatherBuffersDefinedByInstruction()
636 DCHECK(source_buffers[0]->index() == index); in GatherBuffersDefinedByInstruction()
637 buffers->push_back(source_buffers[0]); in GatherBuffersDefinedByInstruction()
642 for (const LogicalBuffer* source_buffer : source_buffers) { in GatherBuffersDefinedByInstruction()
|
| /external/OpenCSD/decoder/tests/snapshots/stm-issue-27/ |
| D | trace.ini | 9 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots/stm_only-juno/ |
| D | trace.ini | 9 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots/stm_only/ |
| D | trace.ini | 9 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots/stm_only-2/ |
| D | trace.ini | 9 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/tme_tcancel/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/001-ack_test/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/002-ack_test_scr/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/ete_spec_3/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/src_addr/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/tme_test/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/trace_file_cid_vmid/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/ts_bit64_set/ |
| D | trace.ini | 11 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/ete_ip/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/ete_mem/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots/juno-uname-001/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/ete-bc-instr/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/trace_file_vmid/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/ete_spec_1/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/ts_marker/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/event_test/ |
| D | trace.ini | 9 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/ete_spec_2/ |
| D | trace.ini | 10 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/tme_simple/ |
| D | trace.ini | 10 [source_buffers] section
|