Home
last modified time | relevance | path

Searched refs:m_TargetShape (Results 1 – 25 of 29) sorted by relevance

12

/external/armnn/src/armnn/layers/
DReshapeLayer.hpp45 … m_Param.m_TargetShape == PolymorphicDowncast<const ReshapeLayer*>(&other)->m_Param.m_TargetShape; in IsEqual()
DReshapeLayer.cpp38 return std::vector<TensorShape>({ m_Param.m_TargetShape }); in InferOutputShapes()
/external/armnn/src/backends/backendsCommon/test/
DReshapeEndToEndTestImpl.hpp51 descriptor.m_TargetShape = outputShape; in ReshapeEndToEnd()
77 descriptor.m_TargetShape = outputShape; in ReshapeEndToEndFloat16()
/external/armnn/src/backends/tosaCommon/operatorMappings/
DReshapeOperator.cpp29 TosaReshapeAttribute attribute(GetTosaTensorShape(reshapeDescriptor->m_TargetShape)); in ConvertReshapeToTosaOperator()
/external/armnn/delegate/classic/src/
DRedefine.hpp113 reshapeDesc.m_TargetShape = outputShape; in CreateOutputTensorShape()
211 if (reshapeDesc.m_TargetShape.GetNumElements() != inputTensorInfo0.GetNumElements()) in VisitReshapeOperator()
DUnpack.hpp156 reshapeDescriptor.m_TargetShape = outputTensorInfos[0].get().GetShape(); in VisitUnpackOperator()
DFullyConnected.hpp221 reshapeDescriptor.m_TargetShape = reshapedTensorInfo.GetShape(); in VisitFullyConnectedOperator()
DClassicDelegateUtils.hpp279 desc.m_TargetShape = outputTensorInfo.GetShape(); in AddReshapeLayer()
/external/armnn/src/armnn/test/optimizations/
DPermuteAsReshapeTests.cpp50 (reshapeLayer->GetParameters().m_TargetShape == infoOut.GetShape()) && in __anonef200eee0102()
DTransposeAsReshapeTests.cpp50 (reshapeLayer->GetParameters().m_TargetShape == infoOut.GetShape()) && in __anon3d6af22b0102()
DOptimizeConsecutiveReshapesTests.cpp52 (reshapeLayer->GetParameters().m_TargetShape == info2.GetShape()) && in __anon76a812040102()
/external/armnn/include/armnn/
DDescriptors.hpp993 : m_TargetShape() in ReshapeDescriptor()
997 : m_TargetShape(shape) in ReshapeDescriptor()
1002 return m_TargetShape == rhs.m_TargetShape; in operator ==()
1006 TensorShape m_TargetShape; member
/external/armnn/src/backends/tosaReference/test/
DTosaRefLayerSupportTests.cpp341 desc.m_TargetShape = {12};
363 desc.m_TargetShape = {12};
/external/armnn/src/backends/tosaCommon/test/
DTosaTestUtils.hpp113 CHECK(GetTosaTensorShape(reshapeDesc->m_TargetShape) == shapeAttrib); in VerifyTosaAttribute()
DOneToOneMappingTests.cpp505 descriptor.m_TargetShape = { 6 };
527 descriptor.m_TargetShape = { 6 };
/external/armnn/delegate/opaque/src/
DOpaqueDelegateUtils.hpp331 desc.m_TargetShape = outputTensorInfo.GetShape(); in AddReshapeLayer()
/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
Darmnn_descriptors.i843 armnn::TensorShape m_TargetShape; member
/external/android-nn-driver/
DConversionUtils.hpp286 reshapeDescriptor.m_TargetShape = reshapeInfo.GetShape(); in AddReshapeLayer()
2074 reshapeDescriptor.m_TargetShape = reshapeInfo.GetShape(); in ConvertConcatenation()
2312 reshapeDescriptor.m_TargetShape = afterConcatInfo.GetShape(); in ConvertConcatenation()
3217 reshapeDescriptor.m_TargetShape = reshapedInfo.GetShape(); in ConvertFullyConnected()
3636 reshapeDescriptor.m_TargetShape = armnn::TensorShape(requestedShape.dimensions.size(), in ConvertReshape()
3745 reshapeDesc.m_TargetShape = outputInfo.GetShape(); in ConvertSqueeze()
/external/armnn/python/pyarmnn/test/
Dtest_descriptors.py307 assert desc.m_TargetShape.GetNumDimensions() == 0
/external/armnn/src/armnn/
DSerializeLayerParameters.cpp422 ss << desc.m_TargetShape; in Serialize()
/external/armnn/src/armnnTfLiteParser/
DTfLiteParser.cpp1494 reshapeDesc.m_TargetShape = outputTensorInfo.GetShape(); in ParseExpandDims()
1527 …reshapeDesc.m_TargetShape = TensorShape(static_cast<unsigned int>(inputDimSize + 1), shape.data()); in ParseExpandDims()
1533 …reshapeDesc.m_TargetShape = OutputTensorInfoFromInputs(subgraphIndex, operatorIndex, layer, 0, {0}… in ParseExpandDims()
1534 outputTensorInfo.SetShape(reshapeDesc.m_TargetShape); in ParseExpandDims()
2298 reshapeDesc.m_TargetShape = outputTensorInfo.GetShape(); in ParseSqueeze()
3128 reshapeDesc.m_TargetShape = reshapeOutputTensorInfo.GetShape(); in ParseReshape()
3361 reshapeDescriptor.m_TargetShape = reshapedTensorInfo.GetShape(); in ParseFullyConnected()
3901 desc.m_TargetShape = reshapeOutputShape; in ParseUnpack()
4650 desc.m_TargetShape = outputShape.GetShape(); in AddReshapeLayer()
/external/armnn/src/armnn/test/
DShapeInferenceTests.cpp626 descriptor.m_TargetShape = { 1, 1, 1, 8 };
/external/armnn/shim/sl/canonical/
DConverter.cpp755 reshapeDescriptor.m_TargetShape = reshapeInfo.GetShape(); in ConvertConcatenation()
978 reshapeDescriptor.m_TargetShape = afterConcatInfo.GetShape(); in ConvertConcatenation()
1730 reshapeDescriptor.m_TargetShape = targetShape; in ConvertExpandDims()
2028 reshapeDescriptor.m_TargetShape = reshapedInfo.GetShape(); in ConvertFullyConnected()
4868 reshapeDescriptor.m_TargetShape = armnn::TensorShape(requestedShape.dimensions.size(), in ConvertReshape()
5698 reshapeDesc.m_TargetShape = outputInfo.GetShape(); in ConvertSqueeze()
DConversionUtils.hpp233 reshapeDescriptor.m_TargetShape = reshapeInfo.GetShape(); in AddReshapeLayer()
/external/armnn/src/armnnOnnxParser/
DOnnxParser.cpp1236 reshapeDescriptor.m_TargetShape = reshapedTensorInfo.GetShape(); in AddFullyConnected()
1295 desc.m_TargetShape = outputInfo[0].GetShape(); in AddFullyConnected()
1447 reshapeDesc.m_TargetShape = outputTensorInfo.GetShape(); in CreateReshapeLayer()

12