einsum.h File Reference

Einstein summation op. More...

#include <tvm/te/operation.h>
#include <tvm/topi/detail/constant_utils.h>
#include <tvm/topi/detail/ravel_unravel.h>
#include <tvm/topi/detail/tensor_utils.h>
#include <tvm/topi/tags.h>
#include <algorithm>
#include <bitset>
#include <iterator>
#include <string>
#include <tuple>
#include <unordered_set>
#include <vector>

Include dependency graph for einsum.h:

Go to the source code of this file.

Classes
struct	tvm::topi::EinsumEquation

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::topi

Macros
#define	LABELRANGE   128
#define	NPY_MAXDIMS   16
#define	NPY_MAXARGS   16

Functions
ffi::Array< PrimExpr >	tvm::topi::InferEinsumShape (const std::string &subscripts, const std::vector< ffi::Array< PrimExpr > > &operands)
	Compute the shape of the output.
Tensor	tvm::topi::einsum (const std::string &subscripts_str, const ffi::Array< Tensor > inputs, std::string name="T_einsum", std::string tag=kEinsum)
	Evaluates the Einstein summation convention on the operands.

Detailed Description

Einstein summation op.

Macro Definition Documentation

LABELRANGE

#define LABELRANGE   128

NPY_MAXARGS

#define NPY_MAXARGS   16

NPY_MAXDIMS

#define NPY_MAXDIMS   16