Ort::Custom Namespace Reference

Classes
struct	CudaContext
struct	RocmContext
class	ArgBase
class	TensorBase
struct	Span
class	Tensor
class	Tensor< std::string >
class	Tensor< std::string_view >
struct	TensorArray
struct	OrtLiteCustomOp
struct	OrtLiteCustomFunc
struct	OrtLiteCustomStruct

Typedefs
using	ArgPtr = std::unique_ptr< Custom::ArgBase >
using	ArgPtrs = std::vector< ArgPtr >
using	TensorPtr = std::unique_ptr< Custom::TensorBase >
using	TensorPtrs = std::vector< TensorPtr >
using	Variadic = TensorArray

Functions
template<typename... Args>
OrtLiteCustomOp *	CreateLiteCustomOp (const char *op_name, const char *execution_provider, void(*custom_compute_fn)(Args...), Status(*shape_infer_fn)(ShapeInferContext &)={}, int start_ver=1, int end_ver=MAX_CUSTOM_OP_END_VER)
template<typename... Args>
OrtLiteCustomOp *	CreateLiteCustomOp (const char *op_name, const char *execution_provider, Status(*custom_compute_fn_v2)(Args...), Status(*shape_infer_fn)(ShapeInferContext &)={}, int start_ver=1, int end_ver=MAX_CUSTOM_OP_END_VER)
template<typename CustomOp >
OrtLiteCustomOp *	CreateLiteCustomOp (const char *op_name, const char *execution_provider, int start_ver=1, int end_ver=MAX_CUSTOM_OP_END_VER)

Typedef Documentation

using Ort::Custom::ArgPtr = typedef std::unique_ptr<Custom::ArgBase>

Definition at line 40 of file onnxruntime_lite_custom_op.h.

using Ort::Custom::ArgPtrs = typedef std::vector<ArgPtr>

Definition at line 41 of file onnxruntime_lite_custom_op.h.

using Ort::Custom::TensorPtr = typedef std::unique_ptr<Custom::TensorBase>

Definition at line 310 of file onnxruntime_lite_custom_op.h.

using Ort::Custom::TensorPtrs = typedef std::vector<TensorPtr>

Definition at line 311 of file onnxruntime_lite_custom_op.h.

using Ort::Custom::Variadic = typedef TensorArray

Definition at line 400 of file onnxruntime_lite_custom_op.h.

Function Documentation

template<typename... Args>

OrtLiteCustomOp* Ort::Custom::CreateLiteCustomOp	(	const char *	op_name,
		const char *	execution_provider,
		void(*)(Args...)	custom_compute_fn,
		Status(*)(ShapeInferContext &)	shape_infer_fn = {},
		int	start_ver = 1,
		int	end_ver = MAX_CUSTOM_OP_END_VER
	)

Definition at line 1088 of file onnxruntime_lite_custom_op.h.

template<typename... Args>

OrtLiteCustomOp* Ort::Custom::CreateLiteCustomOp	(	const char *	op_name,
		const char *	execution_provider,
		Status(*)(Args...)	custom_compute_fn_v2,
		Status(*)(ShapeInferContext &)	shape_infer_fn = {},
		int	start_ver = 1,
		int	end_ver = MAX_CUSTOM_OP_END_VER
	)

Definition at line 1099 of file onnxruntime_lite_custom_op.h.

template<typename CustomOp >

OrtLiteCustomOp* Ort::Custom::CreateLiteCustomOp	(	const char *	op_name,
		const char *	execution_provider,
		int	start_ver = 1,
		int	end_ver = MAX_CUSTOM_OP_END_VER
	)

Definition at line 1110 of file onnxruntime_lite_custom_op.h.