Searched refs:SetOutputShapeForReshape (Results 1 – 1 of 1) sorted by relevance
180 Status SetOutputShapeForReshape(InferenceContext* c) { in SetOutputShapeForReshape() function1299 return SetOutputShapeForReshape(c); in __anondb9326b21b02()1312 .SetShapeFn([](InferenceContext* c) { return SetOutputShapeForReshape(c); }) in __anondb9326b21c02()2835 TF_RETURN_IF_ERROR(SetOutputShapeForReshape(c)); in __anondb9326b24202()