| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_scheduling_test.cc | 53 HloComputation* cond_computation = local
75 tuple_shape, cond_computation, body_computation, tuple));
84 module_sequence[cond_computation] = {cond_param, cond_iter, cond_data,
|
| flatten_call_graph_test.cc | 112 HloComputation* cond_computation = local
127 kScalarShape, cond_computation, b_computation, call));
137 kScalarShape, cond_computation, a_computation, param0));
153 HloComputation* cond_computation; local
164 cond_computation = module->AddEmbeddedComputation(builder.Build());
173 ShapeUtil::MakeShape(PRED, {}), cond_computation, cond_computation,
180 const CallGraphNode& cond_node = call_graph->GetNode(cond_computation);
188 const CallGraphNode& cond_node = call_graph->GetNode(cond_computation);
|
| call_graph_test.cc | 302 HloComputation* cond_computation = local
317 kScalarShape, cond_computation, b_computation, call));
327 kScalarShape, cond_computation, a_computation, param0));
336 // (cond_computation and a_computation).
342 UnorderedElementsAre(cond_computation, a_computation));
372 EXPECT_LT(index_of(cond_computation), index_of(a_computation));
383 EXPECT_TRUE(call_graph->Dominates(entry_computation, cond_computation));
387 EXPECT_FALSE(call_graph->Dominates(cond_computation, entry_computation));
395 EXPECT_FALSE(call_graph->Dominates(a_computation, cond_computation));
399 EXPECT_FALSE(call_graph->Dominates(b_computation, cond_computation));
422 HloComputation* cond_computation = local
[all...] |
| hlo_dce_test.cc | 230 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build()); local
250 shape, cond_computation, body_computation, param));
|
| liveness_util_test.cc | 401 HloComputation* cond_computation = local
410 data_shape, cond_computation, body_computation, data));
|
| heap_simulator_test.cc | 430 HloComputation* cond_computation = local
443 tuple_shape, cond_computation, body_computation, param));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| while_test.cc | [all...] |
