#include "vtkSMPTools.h"
#include <algorithm>
#include <cstdint>
#include <iostream>
#include <numeric>
#include <vector>
Go to the source code of this file.
|
| int64_t | vtkONNXInternalUtils::TensorNumberOfElements (const std::vector< int64_t > &shape) |
| | Helper to find the total number of elements given the list of dimensions of a 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.
|
| std::vector< int > | vtkONNXInternalUtils::FindMatchingPermutation (int64_t numTuples, int64_t numComponents, const std::vector< int64_t > modelShape) |
| | This function tries to find the permutation required to match the dimensions of a VTK array to any N dimensional array.
|