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
namespace	mlir
	Include the generated interface declarations.
namespace	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.
static bool	mlir::sparse_tensor::isAdmissibleBranchExp (Operation *op, Block *block, Value v)
	Ensures that the sparsifier can generate code for expression.
static bool	mlir::sparse_tensor::isAdmissibleBranch (Operation *op, Region &region)
	Ensures that the sparsifier can generate code for branch.
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)