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,
|