Searched defs:shape_y (Results 1 – 7 of 7) sorted by relevance
21 def _broadcast_shape_helper(shape_x, shape_y): argument71 def is_broadcast_compatible(shape_x, shape_y): argument87 def broadcast_shape(shape_x, shape_y): argument
88 SmallVectorImpl<int64_t>& result_shape) { in RewriteBatchMatMulV2Op()
443 def broadcast_dynamic_shape(shape_x, shape_y): argument
512 def broadcast_dynamic_shape(shape_x, shape_y): argument546 def broadcast_static_shape(shape_x, shape_y): argument
1928 ShapeHandle shape_y = c->input(1); in __anon091a71b72902() local
2084 ShapeHandle shape_y, in BroadcastBinaryOpOutputShapeFnHelper()
312 auto shape_y = y.getShape(); in GetStaticBroadcastType() local736 auto shape_y = y_rankless.getShape(); in AreBroadcastCompatible() local