Searched refs:shapeInferenceMethod (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/armnn/src/armnn/ |
| H A D | Layer.hpp | 344 void SetShapeInferenceMethod(ShapeInferenceMethod shapeInferenceMethod) in SetShapeInferenceMethod() argument
346 m_ShapeInferenceMethod = shapeInferenceMethod; in SetShapeInferenceMethod()
388 const ShapeInferenceMethod shapeInferenceMethod,
392 …erifyShapeInferenceType(const TensorShape& outputShape, ShapeInferenceMethod shapeInferenceMethod);
|
| H A D | Layer.cpp | 424 const ShapeInferenceMethod shapeInferenceMethod, in ValidateAndCopyShape() argument
428 if (shapeInferenceMethod == ShapeInferenceMethod::ValidateOnly) in ValidateAndCopyShape()
491 …VerifyShapeInferenceType(const TensorShape& outputShape, ShapeInferenceMethod shapeInferenceMethod) in VerifyShapeInferenceType() argument
493 if (shapeInferenceMethod == ShapeInferenceMethod::ValidateOnly) in VerifyShapeInferenceType()
|
| H A D | Network.hpp | 331 ShapeInferenceMethod shapeInferenceMethod, in OptimizerOptionsOpaqueImpl()
338 , m_shapeInferenceMethod(shapeInferenceMethod) in OptimizerOptionsOpaqueImpl()
|
| H A D | Graph.hpp | 98 Graph(bool shapeInferenceMethod = false, bool allowExpandedDims = false) in Graph() argument
101 , m_ShapeInferenceMethod(shapeInferenceMethod ? ShapeInferenceMethod::InferAndValidate : in Graph()
|
| H A D | Network.cpp | 70 ShapeInferenceMethod shapeInferenceMethod, in OptimizerOptionsOpaque() argument
74 … shapeInferenceMethod, importEnabled, in OptimizerOptionsOpaque()
2016 bool shapeInferenceMethod = false; in GetShapeInferenceMethod() local
2022 shapeInferenceMethod |= value.AsBool(); in GetShapeInferenceMethod()
2025 return shapeInferenceMethod; in GetShapeInferenceMethod()
|
| /aosp_15_r20/external/armnn/include/armnn/ |
| H A D | INetwork.hpp | 173 … ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly, in OptimizerOptions()
180 , m_shapeInferenceMethod(shapeInferenceMethod)
275 … ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly,
|
| /aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
| H A D | armnn_network.i | 53 … ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly,
|