Searched refs:refined_shape (Results 1 – 1 of 1) sorted by relevance
40 llvm::SmallVectorImpl<int64_t>* refined_shape) { in GetCastCompatibleShape() argument43 refined_shape->reserve(rank); in GetCastCompatibleShape()49 refined_shape->push_back(dim2); in GetCastCompatibleShape()53 refined_shape->push_back(dim1); in GetCastCompatibleShape()57 refined_shape->push_back(dim1); in GetCastCompatibleShape()340 llvm::SmallVector<int64_t, 8> refined_shape; in GetCastCompatibleType() local341 if (!GetCastCompatibleShape(a_tt.getShape(), b_tt.getShape(), &refined_shape)) in GetCastCompatibleType()344 return mlir::RankedTensorType::get(refined_shape, refined_element_ty); in GetCastCompatibleType()