Home
last modified time | relevance | path

Searched refs:CreateConstTensorInfo (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/armnn/src/armnnSerializer/
H A DSerializer.cpp298 auto fbMeanConstTensorInfo = CreateConstTensorInfo(mean); in SerializeBatchNormalizationLayer()
299 auto fbVarianceConstTensorInfo = CreateConstTensorInfo(variance); in SerializeBatchNormalizationLayer()
300 auto fbBetaConstTensorInfo = CreateConstTensorInfo(beta); in SerializeBatchNormalizationLayer()
301 auto fbGammaConstTensorInfo = CreateConstTensorInfo(gamma); in SerializeBatchNormalizationLayer()
363 auto flatBufferConstTensorInfo = CreateConstTensorInfo(input); in SerializeConstantLayer()
516 …flatbuffers::Offset<serializer::ConstTensor> fbAnchorsConstTensorInfo = CreateConstTensorInfo(anch… in SerializeDetectionPostProcessLayer()
719 auto inputToForgetWeights = CreateConstTensorInfo(constants[i++]); //InputToForgetWeights in SerializeLstmLayer()
720 auto inputToCellWeights = CreateConstTensorInfo(constants[i++]); //InputToCellWeights in SerializeLstmLayer()
721 auto inputToOutputWeights = CreateConstTensorInfo(constants[i++]); //InputToOutputWeights in SerializeLstmLayer()
722 … auto recurrentToForgetWeights = CreateConstTensorInfo(constants[i++]); //RecurrentToForgetWeights in SerializeLstmLayer()
[all …]
H A DSerializer.hpp66 flatbuffers::Offset<armnnSerializer::ConstTensor> CreateConstTensorInfo(