OnnxLstm.cpp 文件参考

#include "tensor/AffineQuant.h"
#include "tensor/OnnxGpuKernels.h"
#include "tensor/OnnxInternal.h"
#include <algorithm>
#include <cmath>
#include <limits>

浏览源代码.

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

函数
std::vector< RuntimeTensor >	eve::tensor::onnx_detail::executeQuantLstm (const Node &node, const std::vector< const RuntimeTensor * > &inputs, OnnxCompute *compute=nullptr)
	Execute quant lstm.

变量说明

directions

size_t directions

在文件 OnnxLstm.cpp 第 29 行定义.

被这些函数引用 eve::daynight::DayNight::applyReflectionProbeSky(), eve::fluids::buildVolumeFluidMeshDistribution(), eve::procgen::road::RoadNetwork::connectAllTurns(), eve::housegen::HouseGenerator::generate(), eve::procgen::generateRockMesh(), importRpgMakerMap(), eve::ui_editing::UiDocumentTarget::makeSetStyle(), eve::rts::ArtilleryRelocationSystem::select() , 以及 eve::fluids::VolumeFluidShadowCaster::update().

gates

size_t gates

在文件 OnnxLstm.cpp 第 29 行定义.

被这些函数引用 eve::tensor::onnx_detail::executeQuantLstm().

rows

size_t rows

在文件 OnnxLstm.cpp 第 29 行定义.

scales

std::vector<float> scales

在文件 OnnxLstm.cpp 第 27 行定义.

被这些函数引用 eve::animation::AnimClip::addScaleKey(), eve::tensor::affine::dequantize(), eve::tensor::q::dequantizeAll(), eve::tensor::q::dequantValue(), eve::tensor::onnx_detail::executeNeural(), eve::tensor::onnx_detail::executeQuant(), eve::animation::AnimClip::getScaleKeyCount(), eve::animation::AnimClip::getScaleKeyTime(), eve::animation::AnimClip::getScaleKeyX(), eve::animation::AnimClip::getScaleKeyY(), eve::animation::AnimClip::getScaleKeyZ(), eve::animation::AnimClip::removeScaleKey(), eve::animation::AnimClip::retargetWithProfile() , 以及 eve::animation::AnimClip::setScaleKey().

tensor

const RuntimeTensor& tensor

在文件 OnnxLstm.cpp 第 26 行定义.

zeros

std::vector<int64_t> zeros

在文件 OnnxLstm.cpp 第 28 行定义.

被这些函数引用 eve::tensor::affine::dequantize() , 以及 eve::tensor::onnx_detail::executeQuant().