Home
last modified time | relevance | path

Searched refs:nested_inferences (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc187 nested_inferences; in InferShapesForFunction() local
191 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()
Dshape_refiner_test.cc1424 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()
Dshape_refiner.h51 nested_inferences() const { in nested_inferences() function