Home
last modified time | relevance | path

Searched refs:m_ShapeInferenceMethod (Results 1 – 25 of 68) sorted by relevance

123

/aosp_15_r20/external/armnn/src/armnn/layers/
H A DDetectionPostProcessLayer.cpp45 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
62 …ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "DetectionPostProcess… in ValidateTensorShapesFromInputs()
66 m_ShapeInferenceMethod, in ValidateTensorShapesFromInputs()
71 m_ShapeInferenceMethod, in ValidateTensorShapesFromInputs()
76 m_ShapeInferenceMethod, in ValidateTensorShapesFromInputs()
H A DSwitchLayer.cpp38 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
49 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "SwitchLayer"); in ValidateTensorShapesFromInputs()
52 …GetOutputSlot(1).GetTensorInfo().GetShape(), inferredShapes[1], m_ShapeInferenceMethod, "SwitchLay… in ValidateTensorShapesFromInputs()
H A DLstmLayer.cpp174 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
213 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "LstmLayer"); in ValidateTensorShapesFromInputs()
224 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "LstmLayer"); in ValidateTensorShapesFromInputs()
248 …GetOutputSlot(1).GetTensorInfo().GetShape(), inferredShapes[1], m_ShapeInferenceMethod, "LstmLayer… in ValidateTensorShapesFromInputs()
250 …GetOutputSlot(2).GetTensorInfo().GetShape(), inferredShapes[2], m_ShapeInferenceMethod, "LstmLayer… in ValidateTensorShapesFromInputs()
252 …GetOutputSlot(3).GetTensorInfo().GetShape(), inferredShapes[3], m_ShapeInferenceMethod, "LstmLayer… in ValidateTensorShapesFromInputs()
H A DQLstmLayer.cpp176 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
216 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "QLstmLayer"); in ValidateTensorShapesFromInputs()
228 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "QLstmLayer"); in ValidateTensorShapesFromInputs()
252 …GetOutputSlot(1).GetTensorInfo().GetShape(), inferredShapes[1], m_ShapeInferenceMethod, "QLstmLaye… in ValidateTensorShapesFromInputs()
254 …GetOutputSlot(2).GetTensorInfo().GetShape(), inferredShapes[2], m_ShapeInferenceMethod, "QLstmLaye… in ValidateTensorShapesFromInputs()
H A DCastLayer.cpp40 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
46 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "CastLayer"); in ValidateTensorShapesFromInputs()
H A DChannelShuffleLayer.cpp41 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
47 …ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "ChannelShuffleLayer"… in ValidateTensorShapesFromInputs()
H A DConvertFp16ToFp32Layer.cpp41 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
47 …ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "ConvertFp16ToFp32Lay… in ValidateTensorShapesFromInputs()
H A DFloorLayer.cpp40 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
45 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "FloorLayer"); in ValidateTensorShapesFromInputs()
H A DDequantizeLayer.cpp39 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
46 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "DequantizeLayer"); in ValidateTensorShapesFromInputs()
H A DMemImportLayer.cpp43 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
49 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "MemImportLayer"); in ValidateTensorShapesFromInputs()
H A DConvertFp32ToFp16Layer.cpp41 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
47 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "LayerName"); in ValidateTensorShapesFromInputs()
H A DQuantizeLayer.cpp39 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
43 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "QuantizeLayer"); in ValidateTensorShapesFromInputs()
H A DMemCopyLayer.cpp43 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
49 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "MemCopyLayer"); in ValidateTensorShapesFromInputs()
H A DL2NormalizationLayer.cpp40 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
46 …ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "L2NormalizationLayer… in ValidateTensorShapesFromInputs()
H A DLogSoftmaxLayer.cpp40 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
45 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "LogSoftmaxLayer"); in ValidateTensorShapesFromInputs()
H A DActivationLayer.cpp39 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
45 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "ActivationLayer"); in ValidateTensorShapesFromInputs()
H A DSoftmaxLayer.cpp40 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
46 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "SoftmaxLayer"); in ValidateTensorShapesFromInputs()
H A DFakeQuantizationLayer.cpp40 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
46 …ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "FakeQuantizationLaye… in ValidateTensorShapesFromInputs()
H A DAbsLayer.cpp41 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
47 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "AbsLayer"); in ValidateTensorShapesFromInputs()
H A DRsqrtLayer.cpp42 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
48 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "RsqrtLayer"); in ValidateTensorShapesFromInputs()
H A DNormalizationLayer.cpp40 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
46 …ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "NormalizationLayer"); in ValidateTensorShapesFromInputs()
H A DInstanceNormalizationLayer.cpp40 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
46 …ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "InstanceNormalizatio… in ValidateTensorShapesFromInputs()
H A DRankLayer.cpp46 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
47 ValidateAndCopyShape(outputShape, inferredShape, m_ShapeInferenceMethod, "RankLayer"); in ValidateTensorShapesFromInputs()
H A DMergeLayer.cpp36 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod); in ValidateTensorShapesFromInputs()
45 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "MergeLayer"); in ValidateTensorShapesFromInputs()
/aosp_15_r20/external/armnn/src/armnn/
H A DGraph.hpp101 , m_ShapeInferenceMethod(shapeInferenceMethod ? ShapeInferenceMethod::InferAndValidate : in Graph()
123 m_ShapeInferenceMethod = other.m_ShapeInferenceMethod; in operator =()
291 ShapeInferenceMethod m_ShapeInferenceMethod; member in armnn::Graph
462 layer->SetShapeInferenceMethod(m_ShapeInferenceMethod); in AddLayer()

123