vtkONNXInternalUtils.h File Reference

#include "vtkSMPTools.h"
#include <algorithm>
#include <numeric>
#include <onnxruntime_cxx_api.h>
#include <vector>

Include dependency graph for vtkONNXInternalUtils.h:

Go to the source code of this file.

Namespaces
namespace	vtkONNXInternalUtils

Functions
Ort::Value	vtkONNXInternalUtils::RawToTensor (float *data, const std::vector< int64_t > &shape)
	Wraps a raw float buffer into a ONNX Runtime tensor.
bool	vtkONNXInternalUtils::IsPermutation (const std::vector< int > &permutation)
	This checks if a sequence actually represents a permutation.
std::vector< int >	vtkONNXInternalUtils::InversePermutation (const std::vector< int > &permutation)
	Computes the inverse of the input permutation, in other words the permutation you need to apply after the input to retrieve the original sequence.
void	vtkONNXInternalUtils::Permute (float *data, const std::vector< int64_t > &outputShape, const std::vector< int > &permutation)
	This reorders the memory pointed by data so that it matches the layout defined by outputShape and permutation.