HomeSort by relevance Sort by last modified time
    Searched refs:b_computation (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
call_graph_test.cc 306 HloComputation* b_computation = module->AddEmbeddedComputation( local
317 kScalarShape, cond_computation, b_computation, call));
348 UnorderedElementsAre(a_computation, b_computation));
373 EXPECT_LT(index_of(c_computation), index_of(b_computation));
374 EXPECT_LT(index_of(b_computation), index_of(a_computation));
381 EXPECT_TRUE(call_graph->Dominates(entry_computation, b_computation));
385 EXPECT_FALSE(call_graph->Dominates(b_computation, entry_computation));
391 EXPECT_TRUE(call_graph->Dominates(a_computation, b_computation));
393 EXPECT_FALSE(call_graph->Dominates(b_computation, a_computation));
397 EXPECT_TRUE(call_graph->Dominates(b_computation, b_computation))
426 HloComputation* b_computation = module->AddEmbeddedComputation( local
    [all...]
flatten_call_graph_test.cc 116 HloComputation* b_computation = module->AddEmbeddedComputation( local
127 kScalarShape, cond_computation, b_computation, call));
208 HloComputation* b_computation = module->AddEmbeddedComputation( local
212 MakeCallingComputation(b_computation, /*callsites=*/2, ".Entry"));
222 const CallGraphNode& b_node = call_graph->GetNode(b_computation);
buffer_assignment_test.cc 1280 auto b_computation = b_builder.Build(); local
    [all...]

Completed in 114 milliseconds