Searched defs:broadcasted_shape (Results 1 – 6 of 6) sorted by relevance
173 llvm::SmallVector<int64_t, 4> broadcasted_shape; in RewriteOp() local
459 const auto broadcasted_shape = infer_size(x.sizes(), y.sizes()); variable
238 const auto broadcasted_shape = infer_size(sparse.sizes(), scalar.sizes()); in intersection_binary_op_with_wrapped_scalar_() local
847 SmallVector<int64_t, 4> broadcasted_shape; in matchAndRewrite() local
4652 SmallVector<int64_t, 8> broadcasted_shape; in matchAndRewrite() local
724 const Shape& broadcasted_shape = ShapeUtil::MakeShape( in BinaryOp() local