Preparation part of Invoke method before RunLoop. PrintInfoAndSetInputArgs (const VMFunction &func, const std::vector &args) Note: it sets set_outputs_enabled_ true and fill outputs_ but after invocation the first is switched off and the second is cleared. It is used in scenario when output tensors are allocated outside each invocation. It is native implementation of 'set_outputs' python method. Set pre-allocated output tensors to a function. SetOutputs (std::string name, TVMArgs args) Set one input tensor with index or name to a function. SetOneInput (std::string name, const TVMArgValue &tag, const TVMArgValue &tensor) SetInput (std::string name, TVMArgs args, int offset) Invoke a global setting up the VM state to execute. InvokeGlobal (const VMFunction &func, const std::vector &args) Get device from the device list based on a given device index. RunLoop (const std::vector &output_tensor_reg_indices=) Initialize the virtual machine for a set of (physical) devices. Init (const std::vector &physical_devices, const std::vector &alloc_types) InvokePacked ( Index packed_index, const PackedFunc &func, Index arg_count, Index output_size, const std::vector &args) Invoke (const VMFunction &func, const std::vector &input_args, const std::vector &output_args) Invoke (const std::string &name, const std::vector &args) Invoke (const VMFunction &func, const std::vector &args) Read a VM register and cast it to int32_t. WriteRegister ( RegName reg, const ObjectRef &obj) PushFrame ( Index arg_count, Index ret_pc, const VMFunction &vm_func) Public Member Functions inherited from tvm::runtime::Object TVM_DECLARE_FINAL_OBJECT_INFO ( ModuleNode, Object) If query_imports is true, also search in any imported modules. Returns true if this module has a definition for a function of name. ImplementsFunction (const String &name, bool query_imports=false) Returns true if this module is 'DSO exportable'. Get a function from current environment The environment includes all the imports as well as Global functions. Get packed function from current module by name. GetFunction (const std::string &name, bool query_imports=false) Get the format of the module, when available. Get the source code of module, when available. SaveToFile (const std::string &file_name, const std::string &format) Public Member Functions inherited from tvm::runtime::ModuleNode Load the executable for the virtual machine. GetFunction (const std::string &name, const ObjectPtr &sptr_to_self) This browser is not able to show SVG: try Firefox, Chrome, Safari, or Opera instead.
0 Comments
Leave a Reply. |