Searched defs:shape_x (Results 1 – 8 of 8) 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
141 const size_t shape_x = reshaped_weights_shape.x(); in compute_target() local
512 def broadcast_dynamic_shape(shape_x, shape_y): argument546 def broadcast_static_shape(shape_x, shape_y): argument
1927 ShapeHandle shape_x = c->input(0); in __anon091a71b72902() local
2083 ShapeHandle shape_x, in BroadcastBinaryOpOutputShapeFnHelper()
311 auto shape_x = x.getShape(); in GetStaticBroadcastType() local735 auto shape_x = x_rankless.getShape(); in AreBroadcastCompatible() local