Searched refs:nested_inferences (Results 1 – 3 of 3) sorted by relevance
187 nested_inferences; in InferShapesForFunction() local191 nested_inferences[node_name] = std::move(node_to_context_[node]); in InferShapesForFunction()196 nested_inferences[node_name]->get_context()->node_def_ = in InferShapesForFunction()200 outer_context->set_nested_inferences(std::move(nested_inferences)); in InferShapesForFunction()
1424 EXPECT_EQ(m.GetExtendedContext(x2.node())->nested_inferences().size(), in TestSimpleFunctionInference()1427 EXPECT_EQ(m.GetExtendedContext(x2.node())->nested_inferences().size(), 0); in TestSimpleFunctionInference()1432 EXPECT_EQ(m.GetExtendedContext(x2.node())->nested_inferences().size(), 0); in TestSimpleFunctionInference()1480 EXPECT_EQ(m.GetExtendedContext(x2.node())->nested_inferences().size(), 0); in TEST_F()1509 EXPECT_EQ(m.GetExtendedContext(x16.node())->nested_inferences().size(), in TEST_F()1512 m.GetExtendedContext(x16.node())->nested_inferences().at("x4").get(); in TEST_F()1515 auto* x2c = x4->nested_inferences().at("x2")->get_context(); in TEST_F()
51 nested_inferences() const { in nested_inferences() function