on_device.h File Reference

Attribute for the "on_device" annotation (ie operator). More...

#include <tvm/ir/attrs.h>
#include <tvm/target/virtual_device.h>
#include <string>

Include dependency graph for on_device.h:

Go to the source code of this file.

Classes
struct	tvm::relay::OnDeviceAttrs
	Attributes for the "on_device" annotation (ie operator). More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::relay
	Relay: a high level functional IR for TVM.

Detailed Description

Attribute for the "on_device" annotation (ie operator).