25 #ifndef NNVM_COMPILER_OP_ATTR_TYPES_H_ 26 #define NNVM_COMPILER_OP_ATTR_TYPES_H_ 29 #include <tvm/tensor.h> 30 #include <tvm/schedule.h> 31 #include <tvm/packed_func_ext.h> 32 #include <tvm/runtime/registry.h> 45 using ::tvm::Schedule;
80 const Array<Tensor>& inputs,
81 const Array<Tensor>& out_info)>;
93 const Array<Tensor>& outs,
94 const std::string& target)>;
108 const Array<Tensor>& tinfos,
120 #endif // NNVM_COMPILER_OP_ATTR_TYPES_H_
Definition: op_attr_types.h:57
The attributes of the current operation node. Usually are additional parameters like axis...
Definition: node.h:120
Definition: op_attr_types.h:62
std::function< bool(const NodeAttrs &attrs, const Symbol &inputs, const Array< Tensor > &tinfos, Symbol *ret)> FTVMAlterOpLayout
Modify the op node to alter its input layout. it is invoked in AlterOpLayout pass.
Definition: op_attr_types.h:109
std::function< Array< Tensor >(const NodeAttrs &attrs, const Array< Tensor > &inputs, const Array< Tensor > &out_info)> FTVMCompute
Computation description interface.
Definition: op_attr_types.h:81
Definition: op_attr_types.h:50
Data structures that can appear in graph attributes.
Definition: op_attr_types.h:64
Extension to enable packed functionn for nnvm types.
Configuation of nnvm as well as basic data structure.
int TOpPattern
the operator pattern
Definition: op_attr_types.h:68
Definition: op_attr_types.h:54
Definition: op_attr_types.h:59
std::function< Schedule(const NodeAttrs &attrs, const Array< Tensor > &outs, const std::string &target)> FTVMSchedule
Build the computation schedule for op whose root is at current op.
Definition: op_attr_types.h:94
OpPatternKind
operator pattern used in graph fusion
Definition: op_attr_types.h:48
std::function< nnvm::NodePtr(const nnvm::Node *node)> FTVMVectorizedOp
Transform from normal operator to vectorized operator.
Definition: op_attr_types.h:116
Data structures that can appear in operator attributes.
Symbol is help class used to represent the operator node in Graph.
Definition: symbolic.h:51