| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | infeed_thunk.cc | 41 ShapeTree<InfeedBuffer> source_buffers = in ExecuteOnStream() local
45 for (auto& source : source_buffers.leaves()) { in ExecuteOnStream()
50 ShapeUtil::GetSubshape(source_buffers.shape(), shape_index); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 164 std::vector<se::DeviceMemoryBase> source_buffers; member
182 absl::StrJoin(source_buffers, ", ", addr_formatter), in ToString()
334 CHECK(!participants_[0].source_buffers.empty()); in RunCollectiveOp()
335 int expected_buffer_size = participants_[0].source_buffers[0].size(); in RunCollectiveOp()
342 CHECK_EQ(p.source_buffers.size(), p.destination_buffers.size()); in RunCollectiveOp()
343 CHECK_EQ(p.source_buffers.size(), participants_.size()); in RunCollectiveOp()
344 for (int i = 0; i < p.source_buffers.size(); i++) { in RunCollectiveOp()
346 CHECK_EQ(p.source_buffers[i].size(), expected_buffer_size); in RunCollectiveOp()
372 sender.source_buffers[i].opaque(), expected_buffer_size); in RunCollectiveOp()
622 xla::int64 buffer_size, 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 | 588 const PointsToSet::BufferList& source_buffers) { in GatherBuffersDefinedByInstruction() argument
590 CHECK(!source_buffers.empty()); in GatherBuffersDefinedByInstruction()
591 if (source_buffers.size() == 1 && in GatherBuffersDefinedByInstruction()
592 source_buffers[0]->instruction() == instruction) { in GatherBuffersDefinedByInstruction()
595 DCHECK(source_buffers[0]->index() == index); in GatherBuffersDefinedByInstruction()
596 buffers->push_back(source_buffers[0]); in GatherBuffersDefinedByInstruction()
601 for (const LogicalBuffer* source_buffer : source_buffers) { in GatherBuffersDefinedByInstruction()
|
| /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/stm-issue-27/ |
| D | trace.ini | 9 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots-ete/ete_mem/ |
| 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/tme_tcancel/ |
| 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/ete_spec_3/ |
| 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_ip/ |
| 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/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_2/ |
| 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/tme_simple/ |
| 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/ts_marker/ |
| 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/a55-test-tpiu/ |
| D | trace.ini | 14 [source_buffers] section
|
| /external/OpenCSD/decoder/tests/snapshots/Snowball/ |
| D | trace.ini | 9 [source_buffers] section
|