Home
last modified time | relevance | path

Searched refs:concatDescriptor (Results 1 – 20 of 20) sorted by relevance

/external/armnn/delegate/classic/src/
DControl.hpp25 armnn::OriginsDescriptor& concatDescriptor, in SetupConcatViewOrigin() argument
30 const uint32_t inputRank = concatDescriptor.GetNumDimensions(); in SetupConcatViewOrigin()
41 concatDescriptor.SetViewOriginCoord(inputIndex, j, 0); in SetupConcatViewOrigin()
44 concatDescriptor.SetViewOriginCoord(inputIndex, concatAxis, mergeDimOrigin); in SetupConcatViewOrigin()
49 concatDescriptor.SetViewOriginCoord(inputIndex, j, 0); in SetupConcatViewOrigin()
111 armnn::OriginsDescriptor concatDescriptor(static_cast<uint32_t>(numConcatView), inputRank); in VisitConcatenationOperator() local
112 concatDescriptor.SetConcatAxis(concatDimInput); in VisitConcatenationOperator()
121 …SetupConcatViewOrigin(inputTensorInfo, concatDescriptor, concatDimInput, viewIndex, mergeDimOrigin… in VisitConcatenationOperator()
149 concatDescriptor); in VisitConcatenationOperator()
159 …::IConnectableLayer* concatenationLayer = delegateData.m_Network->AddConcatLayer(concatDescriptor); in VisitConcatenationOperator()
/external/armnn/src/armnnUtils/
DParserHelper.cpp20 armnn::OriginsDescriptor& concatDescriptor, in ProcessConcatInputTensorInfo() argument
25 const uint32_t inputRank = concatDescriptor.GetNumDimensions(); in ProcessConcatInputTensorInfo()
40 concatDescriptor.SetViewOriginCoord(inputIndex, j, 0); in ProcessConcatInputTensorInfo()
43 concatDescriptor.SetViewOriginCoord(inputIndex, concatAxis, mergeDimOrigin); in ProcessConcatInputTensorInfo()
48 concatDescriptor.SetViewOriginCoord(inputIndex, j, 0); in ProcessConcatInputTensorInfo()
DParserHelper.hpp17 armnn::OriginsDescriptor& concatDescriptor,
/external/armnn/src/backends/tosaCommon/operatorMappings/
DConcatOperator.cpp11 const OriginsDescriptor* concatDescriptor) in ConvertConcatToTosaOperator() argument
44 auto axis = static_cast<int32_t>(concatDescriptor->GetConcatAxis()); in ConvertConcatToTosaOperator()
DConcatOperator.hpp20 const OriginsDescriptor* concatDescriptor);
/external/armnn/src/armnn/test/
DSubgraphViewTests.cpp528 OriginsDescriptor concatDescriptor(2); variable
529 Layer* const concatLayer = graph.AddLayer<ConcatLayer>(concatDescriptor, "concat");
575 OriginsDescriptor concatDescriptor(2); variable
576 Layer* const concatLayer = graph.AddLayer<ConcatLayer>(concatDescriptor, "concat");
629 OriginsDescriptor concatDescriptor(2); variable
630 Layer* const concatLayer = graph.AddLayer<ConcatLayer>(concatDescriptor, "concat");
683 OriginsDescriptor concatDescriptor(2); variable
684 IConnectableLayer* const concatLayer = graph.AddLayer<ConcatLayer>(concatDescriptor, "concat");
873 OriginsDescriptor concatDescriptor(2); variable
874 auto m6 = graph.AddLayer<ConcatLayer>(concatDescriptor, "m6");
[all …]
/external/armnn/src/backends/cl/workloads/
DClUnidirectionalSequenceLstmFloatWorkload.cpp421 …ConcatDescriptor concatDescriptor(maxTime, numberDimensions); // maxTime = num inputs (aka. numb… in ClUnidirectionalSequenceLstmFloatWorkload() local
424 concatDescriptor.SetViewOriginCoord(inputIdx, dimension, inputIdx); in ClUnidirectionalSequenceLstmFloatWorkload()
425 concatDescriptor.SetConcatAxis(dimension); in ClUnidirectionalSequenceLstmFloatWorkload()
429 unsigned int aclAxisConcat = CalcAclAxis(concatDescriptor.GetNumDimensions(), in ClUnidirectionalSequenceLstmFloatWorkload()
430 concatDescriptor.GetConcatAxis()); in ClUnidirectionalSequenceLstmFloatWorkload()
/external/armnn/src/backends/neon/workloads/
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp424 …ConcatDescriptor concatDescriptor(maxTime, numberDimensions); // maxTime = num inputs (aka. numb… in NeonUnidirectionalSequenceLstmFloatWorkload() local
427 concatDescriptor.SetViewOriginCoord(inputIdx, dimension, inputIdx); in NeonUnidirectionalSequenceLstmFloatWorkload()
428 concatDescriptor.SetConcatAxis(dimension); in NeonUnidirectionalSequenceLstmFloatWorkload()
432 …unsigned int aclAxisConcat = CalcAclAxis(concatDescriptor.GetNumDimensions(), concatDescriptor.Get… in NeonUnidirectionalSequenceLstmFloatWorkload()
DNeonUnidirectionalSequenceLstmWorkload.cpp404 …ConcatDescriptor concatDescriptor(maxTime, numberDimensions); // maxTime = num inputs (aka. numb… in NeonUnidirectionalSequenceLstmWorkload() local
408 concatDescriptor.SetViewOriginCoord(inputIdx, dimension, inputIdx); in NeonUnidirectionalSequenceLstmWorkload()
409 concatDescriptor.SetConcatAxis(dimension); in NeonUnidirectionalSequenceLstmWorkload()
413 …unsigned int aclAxisConcat = CalcAclAxis(concatDescriptor.GetNumDimensions(), concatDescriptor.Get… in NeonUnidirectionalSequenceLstmWorkload()
/external/armnn/src/armnnSerializer/
DSerializer.cpp861 const armnn::ConcatDescriptor& concatDescriptor, in SerializeConcatLayer() argument
869 for (unsigned int v = 0; v < concatDescriptor.GetNumViews(); ++v) in SerializeConcatLayer()
871 const uint32_t* origin = concatDescriptor.GetViewOrigin(v); in SerializeConcatLayer()
873 for (unsigned int d = 0; d < concatDescriptor.GetNumDimensions(); ++d) in SerializeConcatLayer()
883 concatDescriptor.GetConcatAxis(), in SerializeConcatLayer()
884 concatDescriptor.GetNumViews(), in SerializeConcatLayer()
885 concatDescriptor.GetNumDimensions(), in SerializeConcatLayer()
DSerializer.hpp141 const armnn::ConcatDescriptor& concatDescriptor,
/external/armnn/src/armnn/
DNetwork.hpp74 IConnectableLayer* AddConcatLayer(const ConcatDescriptor& concatDescriptor,
DNetwork.cpp264 IConnectableLayer* INetwork::AddConcatLayer(const ConcatDescriptor& concatDescriptor, in AddConcatLayer() argument
267 return pNetworkImpl->AddConcatLayer(concatDescriptor, name); in AddConcatLayer()
2108 IConnectableLayer* NetworkImpl::AddConcatLayer(const ConcatDescriptor& concatDescriptor, in AddConcatLayer() argument
2111 return m_Graph->AddLayer<ConcatLayer>(concatDescriptor, name); in AddConcatLayer()
/external/armnn/shim/sl/canonical/
DConverter.cpp847 armnn::OriginsDescriptor concatDescriptor; in ConvertConcatenation() local
853 concatDescriptor = armnn::CreateDescriptorForConcatenation(inputShapes.begin(), in ConvertConcatenation()
885 concatDescriptor); in ConvertConcatenation()
902 armnn::IConnectableLayer* layer = data.m_Network->AddConcatLayer(concatDescriptor); in ConvertConcatenation()
2453 ConcatDescriptor concatDescriptor; in ConvertGroupedConv2d() local
2455 concatDescriptor = ConcatDescriptor(weightsShape[0]); in ConvertGroupedConv2d()
2461 concatDescriptor.SetViewOriginCoord(index, channelsIndex, index); in ConvertGroupedConv2d()
2462 concatDescriptor.SetConcatAxis(channelsIndex); in ConvertGroupedConv2d()
2475 concatDescriptor); in ConvertGroupedConv2d()
2482 IConnectableLayer* concatLayer = data.m_Network->AddConcatLayer(concatDescriptor); in ConvertGroupedConv2d()
/external/android-nn-driver/
DConversionUtils_1_2.hpp1302 ConcatDescriptor concatDescriptor; in ConvertGroupedConv2d() local
1304 concatDescriptor = ConcatDescriptor(weightsShape[0]); in ConvertGroupedConv2d()
1310 concatDescriptor.SetViewOriginCoord(index, channelsIndex, index); in ConvertGroupedConv2d()
1311 concatDescriptor.SetConcatAxis(channelsIndex); in ConvertGroupedConv2d()
1324 concatDescriptor); in ConvertGroupedConv2d()
1331 IConnectableLayer* concatLayer = data.m_Network->AddConcatLayer(concatDescriptor); in ConvertGroupedConv2d()
DConversionUtils.hpp2170 armnn::OriginsDescriptor concatDescriptor; in ConvertConcatenation() local
2176 concatDescriptor = armnn::CreateDescriptorForConcatenation(inputShapes.begin(), in ConvertConcatenation()
2208 concatDescriptor); in ConvertConcatenation()
2225 armnn::IConnectableLayer* layer = data.m_Network->AddConcatLayer(concatDescriptor); in ConvertConcatenation()
/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
Darmnn_network.i487 armnn::IConnectableLayer* AddConcatLayer(const armnn::ConcatDescriptor& concatDescriptor,
/external/armnn/include/armnn/
DINetwork.hpp377 IConnectableLayer* AddConcatLayer(const ConcatDescriptor& concatDescriptor,
/external/armnn/src/armnnOnnxParser/
DOnnxParser.cpp1654 OriginsDescriptor concatDescriptor(numConcatView, inputRank); in ParseConcat() local
1655 concatDescriptor.SetConcatAxis(concatDimInput); in ParseConcat()
1671 inputTensorInfo, concatDescriptor, concatDimInput, viewIndex, mergeDimOrigin); in ParseConcat()
1674 IConnectableLayer* layer = m_Network->AddConcatLayer(concatDescriptor, node.name().c_str()); in ParseConcat()
/external/armnn/src/armnnTfLiteParser/
DTfLiteParser.cpp3239 OriginsDescriptor concatDescriptor(static_cast<uint32_t>(numConcatView), inputRank); in ParseConcatenation() local
3240 concatDescriptor.SetConcatAxis(concatDimInput); in ParseConcatenation()
3249 inputTensorInfo, concatDescriptor, concatDimInput, viewIndex, mergeDimOrigin); in ParseConcatenation()
3254 IConnectableLayer* layer = m_Network->AddConcatLayer(concatDescriptor, layerName.c_str()); in ParseConcatenation()