Home
last modified time | relevance | path

Searched refs:shapeInferenceMethod (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/armnn/src/armnn/
H A DLayer.hpp344 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 DLayer.cpp424 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 DNetwork.hpp331 ShapeInferenceMethod shapeInferenceMethod, in OptimizerOptionsOpaqueImpl()
338 , m_shapeInferenceMethod(shapeInferenceMethod) in OptimizerOptionsOpaqueImpl()
H A DGraph.hpp98 Graph(bool shapeInferenceMethod = false, bool allowExpandedDims = false) in Graph() argument
101 , m_ShapeInferenceMethod(shapeInferenceMethod ? ShapeInferenceMethod::InferAndValidate : in Graph()
H A DNetwork.cpp70 ShapeInferenceMethod shapeInferenceMethod, in OptimizerOptionsOpaque() argument
74shapeInferenceMethod, 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 DINetwork.hpp173 … 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 Darmnn_network.i53 … ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly,