Searched refs:BroadcastBinaryOpShapeFn (Results 1 – 5 of 5) sorted by relevance
382 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn);393 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)407 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)416 shape_inference::BroadcastBinaryOpShapeFn);423 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)432 shape_inference::BroadcastBinaryOpShapeFn);440 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn);448 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)457 shape_inference::BroadcastBinaryOpShapeFn);464 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn);[all …]
35 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)
146 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn);
256 return shape_inference::BroadcastBinaryOpShapeFn(c); in __anonb4ab91930702()
305 inline Status BroadcastBinaryOpShapeFn(InferenceContext* c) { in BroadcastBinaryOpShapeFn() function