Searched refs:entry_root (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_input_fusion_test.cc | 69 const HloInstruction* entry_root = in TEST_F() local
71 EXPECT_THAT(entry_root, op::Tuple((op::GetTupleElement(op::Fusion())), in TEST_F()
74 const HloInstruction* fusion = entry_root->operand(0)->operand(0); in TEST_F()
|
| D | horizontal_loop_fusion_test.cc | 76 const HloInstruction* entry_root = in TEST_F() local
78 EXPECT_THAT(entry_root, in TEST_F()
82 const HloInstruction* fusion = entry_root->operand(0)->operand(0)->operand(0); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_alias_analysis_test.cc | 1224 auto entry_root = in TEST_F() local
1245 schedule.set_sequence(entry, {init, xla_while, negate, entry_root}); in TEST_F()
1252 schedule.set_sequence(entry, {init, negate, xla_while, entry_root}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3665 auto entry_root = module->entry_computation()->root_instruction(); in Run() local
3666 CHECK(entry_root->has_sharding()) << "Missing sharding in entry root."; in Run()
3667 module->set_spmd_output_sharding(entry_root->sharding()); in Run()
3677 HloSharding root_sharding = entry_root->sharding(); in Run()
|