Searched defs:shape_y (Results 1 – 8 of 8) sorted by relevance
25 def _broadcast_shape_helper(shape_x, shape_y): argument73 def is_broadcast_compatible(shape_x, shape_y): argument89 def broadcast_shape(shape_x, shape_y): argument
86 SmallVectorImpl<int64_t>& result_shape) { in RewriteBatchMatMulV2Op()
444 def broadcast_dynamic_shape(shape_x, shape_y): argument
77 auto shape_y = y_ranked.getShape(); in GetBroadcastType() local
506 def broadcast_dynamic_shape(shape_x, shape_y): argument540 def broadcast_static_shape(shape_x, shape_y): argument
1895 ShapeHandle shape_y = c->input(1); in __anon42d741192902() local
1993 ShapeHandle shape_y, in BroadcastBinaryOpOutputShapeFnHelper()
383 auto shape_y = y.getShape(); in GetStaticBroadcastType() local863 auto shape_y = y_rankless.getShape(); in AreBroadcastCompatible() local