Searched refs:shape_v2 (Results 1 – 2 of 2) sorted by relevance
313 Node* shape_v2; in TEST_F() local316 .Finalize(root.graph(), &shape_v2)); in TEST_F()323 TF_ASSERT_OK(m.AddNode(shape_v2)); in TEST_F()328 shape_inference::InferenceContext* ctx = m.GetContext(shape_v2); in TEST_F()
285 def shape_v2(input, out_type=dtypes.int32, name=None): function