Searched refs:root_hlo (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 103 std::unique_ptr<HloInstruction> root_hlo; in TestElementwise2D() local
106 root_hlo = HloInstruction::CreateUnary(answer_shape, opcode, hlos[1]); in TestElementwise2D()
110 root_hlo = HloInstruction::CreateCompare(answer_shape, hlos[1], in TestElementwise2D()
113 root_hlo = HloInstruction::CreateBinary(answer_shape, opcode, hlos[1], in TestElementwise2D()
118 root_hlo = HloInstruction::CreateTernary(answer_shape, opcode, hlos[1], in TestElementwise2D()
124 hlos[0] = builder.AddInstruction(std::move(root_hlo)); in TestElementwise2D()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.cc | 1592 const HloInstruction* root_hlo = callee->root_instruction(); in BuildColocatedBufferSets() local
1610 [this, call_hlo, root_hlo, &points_to_analysis, in BuildColocatedBufferSets()
1618 AddBufferToColocatedSet(root_hlo, index, points_to_analysis, in BuildColocatedBufferSets()
|