Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc1590 const HloInstruction* call_hlo = instruction; in BuildColocatedBufferSets() local
1591 const HloComputation* callee = call_hlo->to_apply(); in BuildColocatedBufferSets()
1593 for (int64 i = 0; i < call_hlo->operand_count(); i++) { in BuildColocatedBufferSets()
1595 const HloInstruction* call_operand = call_hlo->operand(i); in BuildColocatedBufferSets()
1609 call_hlo->shape(), in BuildColocatedBufferSets()
1610 [this, call_hlo, root_hlo, &points_to_analysis, in BuildColocatedBufferSets()
1615 AddBufferToColocatedSet(call_hlo, index, points_to_analysis, in BuildColocatedBufferSets()