tensor.h File Reference

Dataflow tensor object. More...

#include <tvm/arith/bound.h>
#include <tvm/ffi/reflection/registry.h>
#include <tvm/ir/prim/expr.h>
#include <tvm/tirx/op.h>
#include <string>
#include <type_traits>
#include <utility>
#include <vector>

Include dependency graph for tensor.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::te::Operation
	Operation that produces tensors. More...
class	tvm::te::TensorNode
	Opaque construction-time node that represents a tensor. More...
class	tvm::te::Tensor
	Tensor structure representing a possible input, or intermediate computation result. More...
class	tvm::te::Tensor::Slice
	data structure to represent a slice that fixes first k coordinates. This is used to enable syntax sugar of Tensor[x][y][z] to get the element. More...

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::te
	Tensor expression language DSL.

Macros
#define	DEFINE_OVERLOAD_SLICE_UNARY_OP(Op)    inline PrimExpr operator Op(const Tensor::Slice& a) { return Op a.operator PrimExpr(); }
#define	DEFINE_OVERLOAD_SLICE_BINARY_OP(Op)

Functions
bool	tvm::te::IsTensorLoad (const Expr &expr)
	Return whether an expression is a Call whose callee is a TE Tensor.
Tensor	tvm::te::GetTensorFromLoad (const Call &call)
	Recover and validate the Tensor callee of a tensor-load Call.
ffi::Array< PrimExpr >	tvm::te::GetTensorLoadIndices (const Call &call)
	Recover and validate the primitive indices of a tensor-load Call.
PrimExpr	tvm::te::operator! (const Tensor::Slice &a)
PrimExpr	tvm::te::operator- (const Tensor::Slice &a)
template<typename T >
PrimExpr	tvm::te::operator+ (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator+ (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator+ (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator- (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator- (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator- (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator* (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator* (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator* (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator== (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator== (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator== (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator<= (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator<= (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator<= (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator>= (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator>= (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator>= (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator!= (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator!= (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator!= (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator&& (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator&& (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator&& (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator|| (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator|| (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator|| (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator>> (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator>> (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator>> (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator<< (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator<< (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator<< (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator> (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator> (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator> (const Tensor::Slice &a, const Tensor::Slice &b)
template<typename T >
PrimExpr	tvm::te::operator< (const Tensor::Slice &a, const T &b)
template<typename T >
PrimExpr	tvm::te::operator< (const T &a, const Tensor::Slice &b)
PrimExpr	tvm::te::operator< (const Tensor::Slice &a, const Tensor::Slice &b)

Detailed Description

Dataflow tensor object.

Macro Definition Documentation

DEFINE_OVERLOAD_SLICE_BINARY_OP

#define DEFINE_OVERLOAD_SLICE_BINARY_OP

(

Op

)

Value:

  template <typename T>                                                         \
  inline PrimExpr operator Op(const Tensor::Slice& a, const T& b) {             \
    return a.operator PrimExpr() Op b;                                          \
  }                                                                             \
  template <typename T>                                                         \
  inline PrimExpr operator Op(const T& a, const Tensor::Slice& b) {             \
    return a Op b.operator PrimExpr();                                          \
  }                                                                             \
  inline PrimExpr operator Op(const Tensor::Slice& a, const Tensor::Slice& b) { \
    return a.operator PrimExpr() Op b.operator PrimExpr();                      \
  }

DEFINE_OVERLOAD_SLICE_UNARY_OP

#define DEFINE_OVERLOAD_SLICE_UNARY_OP

(

Op

)

inline PrimExpr operator Op(const Tensor::Slice& a) { return Op a.operator PrimExpr(); }