OnnxInternal.h 文件参考

#include <cstring>
#include <map>
#include <optional>
#include <stdexcept>
#include <unordered_map>
#include "tensor/OnnxByteStorage.h"
#include "tensor/OnnxModel.h"

浏览源代码.

类
struct	eve::tensor::onnx_detail::Failure
	Failure public API. 更多...
struct	eve::tensor::onnx_detail::Attribute
	Attribute public API. 更多...
struct	eve::tensor::onnx_detail::Node
	Node public API. 更多...
struct	eve::tensor::onnx_detail::Input
	Input public API. 更多...
struct	eve::tensor::onnx_detail::ModelData
	ModelData public API. 更多...

命名空间
namespace	eve
	Build metadata (engine git commit, build time, third-party version).
namespace	eve::tensor
namespace	eve::tensor::onnx_detail

函数
size_t	eve::tensor::onnx_detail::elementSize (OnnxElement e)
	Element size.
size_t	eve::tensor::onnx_detail::count (const std::vector< int64_t > &shape)
	Returns the number of count.
void	eve::tensor::onnx_detail::validate (const RuntimeTensor &v)
	Validate.
template<class T >
T	eve::tensor::onnx_detail::read (const RuntimeTensor &v, size_t i)
	Reads read.
template<class T >
RuntimeTensor	eve::tensor::onnx_detail::make (OnnxElement e, std::vector< int64_t > shape, const std::vector< T > &values)
	Make.
int64_t	eve::tensor::onnx_detail::integer (const RuntimeTensor &v, size_t i=0)
	Integer.
std::vector< float >	eve::tensor::onnx_detail::floats (const RuntimeTensor &v)
	Floats.
std::vector< int64_t >	eve::tensor::onnx_detail::ints (const RuntimeTensor &v)
	Ints.
int64_t	eve::tensor::onnx_detail::attr (const Node &n, const char *key, int64_t fallback)
	Attr.
std::vector< int64_t >	eve::tensor::onnx_detail::attrs (const Node &n, const char *key, std::vector< int64_t > fallback)
	Attrs.
const RuntimeTensor &	eve::tensor::onnx_detail::required (const std::vector< const RuntimeTensor * > &in, size_t i)
	Required.
int	eve::tensor::onnx_detail::axis (int64_t a, size_t rank)
	Axis.
std::vector< int64_t >	eve::tensor::onnx_detail::broadcast (const std::vector< int64_t > &a, const std::vector< int64_t > &b)
	Broadcast.
size_t	eve::tensor::onnx_detail::broadcastIndex (size_t i, const std::vector< int64_t > &shape, const std::vector< int64_t > &output)
	Broadcast index.
std::optional< std::vector< RuntimeTensor > >	eve::tensor::onnx_detail::executeIndex (const Node &, const std::vector< const RuntimeTensor * > &)
	Execute index.
std::optional< RuntimeTensor >	eve::tensor::onnx_detail::executeShape (const Node &, const std::vector< const RuntimeTensor * > &)
	Execute shape.
RuntimeTensor	eve::tensor::onnx_detail::dispatchFloat (OnnxCompute &, const std::vector< const RuntimeTensor * > &, const std::vector< int64_t > &, const std::string &, size_t work=0)
	Dispatches float.
std::optional< RuntimeTensor >	eve::tensor::onnx_detail::executeMisc (const Node &, const std::vector< const RuntimeTensor * > &, OnnxCompute *)
	Execute misc.
std::optional< RuntimeTensor >	eve::tensor::onnx_detail::executeNeural (const Node &, const std::vector< const RuntimeTensor * > &, OnnxCompute *)
	Execute neural.
std::optional< RuntimeTensor >	eve::tensor::onnx_detail::executeNumeric (const Node &, const std::vector< const RuntimeTensor * > &, OnnxCompute *)
	Execute numeric.
std::vector< OnnxNamedTensor >	eve::tensor::onnx_detail::evaluate (const ModelData &, std::span< const OnnxNamedTensor >, const std::vector< std::string > &, OnnxCompute *, OnnxRunOptions options)
	Evaluate.
ModelData	eve::tensor::onnx_detail::parse (std::span< const uint8_t > bytes)
	Parse.
bool	eve::tensor::onnx_detail::isSupported (const Node &node)
	True when supported.
std::vector< RuntimeTensor >	eve::tensor::onnx_detail::execute (const Node &node, const std::vector< const RuntimeTensor * > &inputs, OnnxCompute *compute=nullptr)
	Execute.
std::vector< RuntimeTensor >	eve::tensor::onnx_detail::executeQuant (const Node &node, const std::vector< const RuntimeTensor * > &inputs, OnnxCompute *compute=nullptr)
	Execute quant.
std::vector< RuntimeTensor >	eve::tensor::onnx_detail::executeQuantLstm (const Node &node, const std::vector< const RuntimeTensor * > &inputs, OnnxCompute *compute=nullptr)
	Execute quant lstm.