Searched defs:shape_x (Results 1 – 9 of 9) 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
76 auto shape_x = x_ranked.getShape(); in GetBroadcastType() local
2236 auto shape_x = x_ranked.getShape(); in UpdateResultElementType() local
506 def broadcast_dynamic_shape(shape_x, shape_y): argument540 def broadcast_static_shape(shape_x, shape_y): argument
1894 ShapeHandle shape_x = c->input(0); in __anon42d741192902() local
1992 ShapeHandle shape_x, in BroadcastBinaryOpOutputShapeFnHelper()
382 auto shape_x = x.getShape(); in GetStaticBroadcastType() local862 auto shape_x = x_rankless.getShape(); in AreBroadcastCompatible() local