Operator implementation class. More...
#include <op_strategy.h>
Inheritance diagram for tvm::relay::OpImplementation:
Collaboration diagram for tvm::relay::OpImplementation:
Public Member Functions | |
| Array< te::Tensor > | Compute (const Attrs &attrs, const Array< te::Tensor > &inputs, const Type &out_type) |
| Invoke the operator compute function. More... | |
| te::Schedule | Schedule (const Attrs &attrs, const Array< te::Tensor > &outs, const Target &target) |
| Build the computation schedule. More... | |
| TVM_DEFINE_OBJECT_REF_METHODS (OpImplementation, ObjectRef, OpImplementationNode) | |
 Public Member Functions inherited from tvm::runtime::ObjectRef | |
| ObjectRef ()=default | |
| default constructor More... | |
| ObjectRef (ObjectPtr< Object > data) | |
| Constructor from existing object ptr. More... | |
| bool | same_as (const ObjectRef &other) const |
| Comparator. More... | |
| bool | operator== (const ObjectRef &other) const |
| Comparator. More... | |
| bool | operator!= (const ObjectRef &other) const |
| Comparator. More... | |
| bool | operator< (const ObjectRef &other) const |
| Comparator. More... | |
| bool | defined () const |
| const Object * | get () const |
| const Object * | operator-> () const |
| bool | unique () const |
| int | use_count () const |
| template<typename ObjectType > | |
| const ObjectType * | as () const |
| Try to downcast the internal Object to a raw pointer of a corresponding type. More... | |
Additional Inherited Members | |
| Public Types inherited from tvm::runtime::ObjectRef | |
| using | ContainerType = Object |
| type indicate the container type. More... | |
| Static Public Attributes inherited from tvm::runtime::ObjectRef | |
| static constexpr bool | _type_is_nullable = true |
| Protected Member Functions inherited from tvm::runtime::ObjectRef | |
| Object * | get_mutable () const |
| Static Protected Member Functions inherited from tvm::runtime::ObjectRef | |
| template<typename T > | |
| static T | DowncastNoCheck (ObjectRef ref) |
| Internal helper function downcast a ref without check. More... | |
| static void | FFIClearAfterMove (ObjectRef *ref) |
| Clear the object ref data field without DecRef after we successfully moved the field. More... | |
| template<typename ObjectType > | |
| static ObjectPtr< ObjectType > | GetDataPtr (const ObjectRef &ref) |
| Internal helper function get data_ as ObjectPtr of ObjectType. More... | |
| Protected Attributes inherited from tvm::runtime::ObjectRef | |
| ObjectPtr< Object > | data_ |
| Internal pointer that backs the reference. More... | |
Detailed Description
Operator implementation class.
Member Function Documentation
◆ Compute()
| Array<te::Tensor> tvm::relay::OpImplementation::Compute | ( | const Attrs & | attrs, |
| const Array< te::Tensor > & | inputs, | ||
| const Type & | out_type | ||
| ) |
Invoke the operator compute function.
- Parameters
-
attrs The attribute of the primitive inputs The input tensors. out_type The output type information.
- Returns
- The output compute description of the operator.
◆ Schedule()
| te::Schedule tvm::relay::OpImplementation::Schedule | ( | const Attrs & | attrs, |
| const Array< te::Tensor > & | outs, | ||
| const Target & | target | ||
| ) |
Build the computation schedule.
- Parameters
-
attrs The attribute of the node. outs The output tensors. target The build target.
- Returns
- The computation schedule.
◆ TVM_DEFINE_OBJECT_REF_METHODS()
| tvm::relay::OpImplementation::TVM_DEFINE_OBJECT_REF_METHODS | ( | OpImplementation | , |
| ObjectRef | , | ||
| OpImplementationNode | |||
| ) |
The documentation for this class was generated from the following file:
- include/tvm/relay/op_strategy.h
