Searched defs:same_shape (Results 1 – 7 of 7) sorted by relevance
51 const bool same_shape = old_lhs.shape().IsSameSize(rhs.shape()); in AssignRefVariable() local
91 bool same_shape = HaveSameShapes(input_condition, input_x) && in SelectPrepare() local
137 auto same_shape = [](mlir::Value src, mlir::Value dst) { in matchAndRewrite() local
686 const int same_shape = state.range(0); in BM_BCastSetup() local
188 auto same_shape = [](mlir::Value src, mlir::Value dst) { in matchAndRewrite() local
188 bool same_shape = in verify() local
99 def same_shape(a: ShapeType, b: ShapeType, *, allow_rhs_unbacked=False) -> bool: function