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.cc1323 EXPECT_EQ(m.GetExtendedContext(x2.node())->nested_inferences().size(), in TestSimpleFunctionInference()
1326 EXPECT_EQ(m.GetExtendedContext(x2.node())->nested_inferences().size(), 0); in TestSimpleFunctionInference()
1331 EXPECT_EQ(m.GetExtendedContext(x2.node())->nested_inferences().size(), 0); in TestSimpleFunctionInference()
1379 EXPECT_EQ(m.GetExtendedContext(x2.node())->nested_inferences().size(), 0); in TEST_F()
1408 EXPECT_EQ(m.GetExtendedContext(x16.node())->nested_inferences().size(), in TEST_F()
1411 m.GetExtendedContext(x16.node())->nested_inferences().at("x4").get(); in TEST_F()