Home
last modified time | relevance | path

Searched refs:source_buffers (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc508 const PointsToSet::BufferList& source_buffers) { in GatherBuffersDefinedByInstruction() argument
510 CHECK(!source_buffers.empty()); in GatherBuffersDefinedByInstruction()
511 if (source_buffers.size() == 1 && in GatherBuffersDefinedByInstruction()
512 source_buffers[0]->instruction() == instruction) { in GatherBuffersDefinedByInstruction()
515 DCHECK(source_buffers[0]->index() == index); in GatherBuffersDefinedByInstruction()
516 buffers->push_back(source_buffers[0]); in GatherBuffersDefinedByInstruction()
521 for (const LogicalBuffer* source_buffer : source_buffers) { in GatherBuffersDefinedByInstruction()
Dlayout_assignment.cc1500 const auto& source_buffers = in InferArrayLayout() local
1502 TF_RET_CHECK(!source_buffers.empty()); in InferArrayLayout()
1507 for (const LogicalBuffer* source_buffer : source_buffers) { in InferArrayLayout()
1528 source_buffers[0]->ToString(), source_buffer->ToString()); in InferArrayLayout()