//
// Copyright (C) 2024-2026 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

#pragma once

#include "vpux/compiler/dialect/core/interfaces/type_interfaces.hpp"

namespace vpux::VPU {

mlir::RankedTensorType inferNCEMatmulOutputType(vpux::NDTypeInterface input1Type, vpux::NDTypeInterface input2Type,
                                                vpux::NDTypeInterface origOutputType);

}  // namespace vpux::VPU
