Searched refs:SetOutputShapeForReshape (Results 1 – 1 of 1) sorted by relevance
187 Status SetOutputShapeForReshape(InferenceContext* c) { in SetOutputShapeForReshape() function1375 return SetOutputShapeForReshape(c); in __anondb9326b21a02()1388 .SetShapeFn([](InferenceContext* c) { return SetOutputShapeForReshape(c); }) in __anondb9326b21b02()3052 TF_RETURN_IF_ERROR(SetOutputShapeForReshape(c)); in __anondb9326b24302()