Searched refs:BroadcastBinaryOpShapeFn (Results 1 – 5 of 5) sorted by relevance
378 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn);387 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)402 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)420 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)437 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn);444 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)453 shape_inference::BroadcastBinaryOpShapeFn);460 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn);468 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)477 shape_inference::BroadcastBinaryOpShapeFn);[all …]
34 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)42 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn)
156 .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn);
280 return shape_inference::BroadcastBinaryOpShapeFn(c); in __anon4354d5f10702()
239 inline Status BroadcastBinaryOpShapeFn(InferenceContext* c) { in BroadcastBinaryOpShapeFn() function