Searched refs:CreateConstTensor (Results 1 – 7 of 7) sorted by relevance
| /external/armnn/delegate/classic/src/ |
| D | Convolution.hpp | 187 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 …]
|
| D | FullyConnected.hpp | 176 auto weightsTensor = CreateConstTensor(&tfLiteWeightsTensor, in VisitFullyConnectedOperator()
190 auto biasTensor = CreateConstTensor(&tfLiteBiasTensor, in VisitFullyConnectedOperator()
205 CreateConstTensor(&tfLiteContext->tensors[tfLiteNode->inputs->data[0]], in VisitFullyConnectedOperator()
|
| D | ClassicDelegateUtils.hpp | 448 armnn::ConstTensor CreateConstTensor(const TfLiteTensor* tfLiteTensor, in CreateConstTensor() function
505 auto constantInput = CreateConstTensor(&tfLiteInputTensor, in ProcessInputs()
|
| /external/armnn/src/armnnOnnxParser/ |
| D | OnnxParser.cpp | 601 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 …]
|
| D | OnnxParser.hpp | 164 CreateConstTensor(const std::string name,
|
| /external/armnn/delegate/opaque/src/ |
| D | OpaqueDelegateUtils.hpp | 528 armnn::ConstTensor CreateConstTensor(const TfLiteOpaqueTensor* tfLiteTensor, in CreateConstTensor() function
614 auto constantInput = CreateConstTensor(tfLiteInputTensor, inputTensorInfo); in ProcessInputs()
|
| /external/armnn/src/armnnSerializer/ |
| D | Serializer.cpp | 1926 …flatbuffers::Offset<serializer::ConstTensor> flatBufferConstTensor = serializer::CreateConstTensor( in CreateConstTensorInfo()
|