reduction.h File Reference

Tensor-expression reduction constructors. More...

#include <tvm/s_tir/iter_var.h>

Include dependency graph for reduction.h:

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

Go to the source code of this file.

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

Functions
PrimExpr	tvm::prim::sum (PrimExpr source, ffi::Array< s_tir::IterVar > axis, ffi::Array< PrimExpr > init={}, Location loc=UnknownLoc())
	sum of source expression over axis
PrimExpr	tvm::prim::all (PrimExpr source, ffi::Array< s_tir::IterVar > axis, ffi::Array< PrimExpr > init={}, Location loc=UnknownLoc())
	logical And of source expression over axis
PrimExpr	tvm::prim::any (PrimExpr source, ffi::Array< s_tir::IterVar > axis, ffi::Array< PrimExpr > init={}, Location loc=UnknownLoc())
	logical Or of source expression over axis
PrimExpr	tvm::prim::prod (PrimExpr source, ffi::Array< s_tir::IterVar > axis, ffi::Array< PrimExpr > init={}, Location loc=UnknownLoc())
	product of source expression over axis
PrimExpr	tvm::max (PrimExpr source, ffi::Array< s_tir::IterVar > axis, ffi::Array< PrimExpr > init={}, Location loc=UnknownLoc())
	max of source expression over axis
PrimExpr	tvm::min (PrimExpr source, ffi::Array< s_tir::IterVar > axis, ffi::Array< PrimExpr > init={}, Location loc=UnknownLoc())
	max of source expression over axis

Detailed Description

Tensor-expression reduction constructors.