llvm/mlir/lib/Interfaces/InferTypeOpInterface.cpp

//===- InferTypeOpInterface.cpp - Infer Type Interfaces ---------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// This file contains the definitions of the infer op interfaces defined in
// `InferTypeOpInterface.td`.
//
//===----------------------------------------------------------------------===//

#include "mlir/Interfaces/InferTypeOpInterface.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/Matchers.h"
#include "llvm/Support/FormatVariadic.h"

usingnamespacemlir;

namespace mlir {
#include "mlir/Interfaces/InferTypeOpInterface.cpp.inc"
} // namespace mlir

LogicalResult
mlir::reifyResultShapes(OpBuilder &b, Operation *op,
                        ReifiedRankedShapedTypeDims &reifiedReturnShapes) { … }

bool ShapeAdaptor::hasRank() const { … }

Type ShapeAdaptor::getElementType() const { … }

void ShapeAdaptor::getDims(SmallVectorImpl<int64_t> &res) const { … }

void ShapeAdaptor::getDims(ShapedTypeComponents &res) const { … }

int64_t ShapeAdaptor::getDimSize(int index) const { … }

int64_t ShapeAdaptor::getRank() const { … }

bool ShapeAdaptor::hasStaticShape() const { … }

int64_t ShapeAdaptor::getNumElements() const { … }

void ShapeAdaptor::dump() const { … }

ShapeAdaptor ValueShapeRange::getValueAsShape(int index) { … }

ShapeAdaptor ValueShapeRange::getShape(Value val) const { … }

ShapeAdaptor ValueShapeRange::getShape(int index) const { … }

LogicalResult mlir::detail::inferReturnTensorTypes(
    ArrayRef<ShapedTypeComponents> retComponents,
    SmallVectorImpl<Type> &inferredReturnTypes) { … }

LogicalResult mlir::detail::verifyInferredResultTypes(Operation *op) { … }