HomeSort by relevance Sort by last modified time
    Searched defs:a_computation (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
flatten_call_graph_test.cc 120 HloComputation* a_computation; local
129 a_computation = module->AddEmbeddedComputation(builder.Build());
138 kScalarShape, cond_computation, a_computation, param0));
call_graph_test.cc 319 HloComputation* a_computation; local
328 a_computation = module->AddEmbeddedComputation(builder.Build());
337 kScalarShape, cond_computation, a_computation, param0));
346 const CallGraphNode& a_node = call_graph->GetNode(a_computation);
359 // (cond_computation and a_computation).
364 UnorderedElementsAre(cond_computation, a_computation));
369 UnorderedElementsAre(a_computation, b_computation));
393 EXPECT_LT(index_of(cond_computation), index_of(a_computation));
395 EXPECT_LT(index_of(b_computation), index_of(a_computation));
401 EXPECT_TRUE(call_graph->Dominates(entry_computation, a_computation));
451 HloComputation* a_computation; local
    [all...]

Completed in 75 milliseconds