Home
last modified time | relevance | path

Searched refs:GetExtendedContext (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.h138 ExtendedInferenceContext* GetExtendedContext(const Node* node) const { in GetExtendedContext() function
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()