Searched refs:body_root (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation.cc | 173 auto body_root = body->root_instruction(); in DetermineWhileComputationsPrecision() local
177 body_root->shape(), [this, while_hlo, body_root]( in DetermineWhileComputationsPrecision()
183 AddToOrRemoveFromBF16ChangeSet(body_root, index, BF16); in DetermineWhileComputationsPrecision()
184 VLOG(2) << "While body root " << body_root->ToString() in DetermineWhileComputationsPrecision()
|
| D | hlo_alias_analysis_test.cc | 1063 auto body_root = body_builder.AddInstruction( in TEST_F() local
1090 schedule.set_sequence(body, {body_param, body_root}); in TEST_F()
|
| D | buffer_assignment_test.cc | 895 HloInstruction* body_root = body_computation->root_instruction(); in TEST_F() local
896 EXPECT_EQ(HloOpcode::kTuple, body_root->opcode()); in TEST_F()
902 [this, &buffers, while_op, body_root](const Shape& /*subshape*/, in TEST_F()
905 auto body_root_allocation = GetAllocation(*buffers, body_root, index); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 495 auto body_root = body_builder.AddInstruction( in TEST_P() local
542 UnorderedElementsAre(HloUse{add, 0, {}}, HloUse{body_root, 0, {}}, in TEST_P()
|