Home
last modified time | relevance | path

Searched refs:CreateConstTensor (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/armnn/delegate/classic/src/
H A DConvolution.hpp187 CreateConstTensor(&tfLiteContext->tensors[tfLiteNode->inputs->data[1]], in VisitConv2dOperator()
200 auto biasTensor = CreateConstTensor(&tfLiteBiasTensor, biasTensorInfo); in VisitConv2dOperator()
212 CreateConstTensor(&tfLiteContext->tensors[tfLiteNode->inputs->data[0]], in VisitConv2dOperator()
373 auto filter = CreateConstTensor(&tfLiteFilterTensor, in VisitConv3dOperator()
388 auto biases = CreateConstTensor(&tfLiteBiasTensor, in VisitConv3dOperator()
403 CreateConstTensor(&tfLiteContext->tensors[tfLiteNode->inputs->data[0]], in VisitConv3dOperator()
596 auto filter = CreateConstTensor(&tfLiteFilterTensor, filterTensorInfo); in VisitDepthwiseConv2dOperator()
608 auto biasTensor = CreateConstTensor(&tfLiteBiasTensor, biasTensorInfo); in VisitDepthwiseConv2dOperator()
620 CreateConstTensor(&tfLiteContext->tensors[tfLiteNode->inputs->data[0]], in VisitDepthwiseConv2dOperator()
788 auto filterTensor = CreateConstTensor(&tfLiteFilterTensor, in VisitTransposeConv2dOperator()
[all …]
H A DFullyConnected.hpp176 auto weightsTensor = CreateConstTensor(&tfLiteWeightsTensor, in VisitFullyConnectedOperator()
190 auto biasTensor = CreateConstTensor(&tfLiteBiasTensor, in VisitFullyConnectedOperator()
205 CreateConstTensor(&tfLiteContext->tensors[tfLiteNode->inputs->data[0]], in VisitFullyConnectedOperator()
H A DClassicDelegateUtils.hpp448 armnn::ConstTensor CreateConstTensor(const TfLiteTensor* tfLiteTensor, in CreateConstTensor() function
505 auto constantInput = CreateConstTensor(&tfLiteInputTensor, in ProcessInputs()
/aosp_15_r20/external/armnn/src/armnnOnnxParser/
H A DOnnxParser.cpp601 OnnxParserImpl::CreateConstTensor(const std::string name, in CreateConstTensor() function in armnnOnnxParser::OnnxParserImpl
1099 auto weightTensor = CreateConstTensor(node.input(1)); in AddConvLayerWithDepthwiseConv()
1126 auto biasTensor = CreateConstTensor(node.input(2)); in AddConvLayerWithDepthwiseConv()
1259 …IConnectableLayer* weightsLayer = m_Network->AddConstantLayer(CreateConstTensor(weightName).first); in AddFullyConnected()
1268 … IConnectableLayer* biasLayer = m_Network->AddConstantLayer(CreateConstTensor(biasName).first); in AddFullyConnected()
1427 auto armnnTensor = CreateConstTensor(tensorName); in CreateConstantLayer()
1620 auto scaleTensor = CreateConstTensor(node.input(1)); in ParseBatchNormalization()
1621 auto biasTensor = CreateConstTensor(node.input(2)); in ParseBatchNormalization()
1622 auto meanTensor = CreateConstTensor(node.input(3)); in ParseBatchNormalization()
1623 auto varTensor = CreateConstTensor(node.input(4)); in ParseBatchNormalization()
[all …]
H A DOnnxParser.hpp164 CreateConstTensor(const std::string name,
/aosp_15_r20/external/armnn/delegate/opaque/src/
H A DOpaqueDelegateUtils.hpp528 armnn::ConstTensor CreateConstTensor(const TfLiteOpaqueTensor* tfLiteTensor, in CreateConstTensor() function
614 auto constantInput = CreateConstTensor(tfLiteInputTensor, inputTensorInfo); in ProcessInputs()
/aosp_15_r20/external/armnn/src/armnnSerializer/
H A DSerializer.cpp1926 …flatbuffers::Offset<serializer::ConstTensor> flatBufferConstTensor = serializer::CreateConstTensor( in CreateConstTensorInfo()
/aosp_15_r20/external/armnn/generated/
H A DArmnnSchema_generated.h2297 inline flatbuffers::Offset<ConstTensor> CreateConstTensor(