Searched refs:root_computation (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_graph_dumper_test.cc | 61 HloComputation* root_computation = m.entry_computation(); in TEST_F() local
64 auto* outer_fusion = root_computation->CreateFusionInstruction( in TEST_F()
84 string graph, RenderGraph(*root_computation, /*label=*/"", DebugOptions(), in TEST_F()
87 {root_computation, // in TEST_F()
120 HloComputation* root_computation = m.AddEntryComputation(b.Build()); in TEST_F() local
122 string graph, RenderGraph(*root_computation, /*label=*/"an_empty_graph", in TEST_F()
139 HloComputation* root_computation = m.AddEntryComputation(b.Build(gte)); in TEST_F() local
141 string graph, RenderGraph(*root_computation, /*label=*/"tuple_constant", in TEST_F()
|