Searched refs:EinsumShape (Results 1 – 4 of 4) sorted by relevance
117 Status EinsumShape(shape_inference::InferenceContext* c);
155 Status EinsumShape(shape_inference::InferenceContext* c) { in EinsumShape() function
830 return shape_inference::EinsumShape(context); in __anon3500048a0b02()
574 .SetShapeFn(shape_inference::EinsumShape);