Searched refs:embedded_computations (Results 1 – 2 of 2) sorted by relevance
313 auto embedded_computations = comp->MakeEmbeddedComputationsList(); in DoInfoCommand() local314 std::cout << " " << embedded_computations.size() << " embedded computation" in DoInfoCommand()315 << (embedded_computations.size() != 1 ? "s" : "") in DoInfoCommand()316 << (!embedded_computations.empty() ? ":" : ".") << std::endl; in DoInfoCommand()317 for (const HloComputation* c : embedded_computations) { in DoInfoCommand()
113 auto embedded_computations = computation->MakeEmbeddedComputationsList(); in TEST_F() local114 EXPECT_EQ(3, embedded_computations.size()); in TEST_F()117 EXPECT_EQ(negate_computation, *embedded_computations.begin()); in TEST_F()118 EXPECT_THAT(embedded_computations, in TEST_F()