MLIR  20.0.0git
Namespaces | Enumerations | Functions
Merger.cpp File Reference
#include "mlir/Dialect/SparseTensor/Utils/Merger.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Complex/IR/Complex.h"
#include "mlir/Dialect/Math/IR/Math.h"
#include "mlir/Dialect/SparseTensor/IR/SparseTensor.h"
#include "mlir/IR/Operation.h"
#include "llvm/Support/Debug.h"
#include <optional>

Go to the source code of this file.

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::sparse_tensor
 

Enumerations

enum class  mlir::sparse_tensor::ExpArity { mlir::sparse_tensor::kNullary , mlir::sparse_tensor::kUnary , mlir::sparse_tensor::kBinary }
 

Functions

static ExpArity mlir::sparse_tensor::getExpArity (TensorExp::Kind k)
 
static const char * mlir::sparse_tensor::kindToOpSymbol (TensorExp::Kind kind)
 
static bool mlir::sparse_tensor::isCertainZero (Value val)
 Only returns true if we are certain this is a zero. More...
 
static bool mlir::sparse_tensor::isAdmissibleBranchExp (Operation *op, Block *block, Value v)
 Ensures that the sparsifier can generate code for expression. More...
 
static bool mlir::sparse_tensor::isAdmissibleBranch (Operation *op, Region &region)
 Ensures that the sparsifier can generate code for branch. More...
 
static bool mlir::sparse_tensor::isGreater (TensorExp::Kind kind, Attribute attr)
 
static Value mlir::sparse_tensor::insertYieldOp (RewriterBase &rewriter, Location loc, Region &region, ValueRange vals)
 
static Value mlir::sparse_tensor::buildUnaryPresent (RewriterBase &rewriter, Location loc, Operation *op, Value v0)
 
static Value mlir::sparse_tensor::buildBinaryOverlap (RewriterBase &rewriter, Location loc, Operation *op, Value v0, Value v1)
 
static Value mlir::sparse_tensor::buildRelu (RewriterBase &rewriter, Location loc, Value v0, Attribute attr)