Home
last modified time | relevance | path

Searched refs:ShapeInferenceMethod (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/armnn/include/armnn/
H A DINetwork.hpp147 , m_shapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly) in OptimizerOptions()
162 , m_shapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly)
173ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly, in OptimizerOptions()
198 (m_shapeInferenceMethod == ShapeInferenceMethod::ValidateOnly in ToString()
238 ShapeInferenceMethod m_shapeInferenceMethod;
275ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly,
299 armnn::ShapeInferenceMethod GetShapeInferenceMethod() const;
313 void SetShapeInferenceMethod(armnn::ShapeInferenceMethod ShapeInferenceMethodType);
H A DTypes.hpp232 enum class ShapeInferenceMethod enum
/aosp_15_r20/external/armnn/src/armnn/
H A DLayer.cpp99 if (GetOwningLayer().GetShapeInferenceMethod() == ShapeInferenceMethod::InferAndValidate) in IsTensorInfoSet()
226 , m_ShapeInferenceMethod(ShapeInferenceMethod::ValidateOnly) in Layer()
424 const ShapeInferenceMethod shapeInferenceMethod, in ValidateAndCopyShape()
428 if (shapeInferenceMethod == ShapeInferenceMethod::ValidateOnly) in ValidateAndCopyShape()
491 void Layer::VerifyShapeInferenceType(const TensorShape& outputShape, ShapeInferenceMethod shapeInfe… in VerifyShapeInferenceType()
493 if (shapeInferenceMethod == ShapeInferenceMethod::ValidateOnly) in VerifyShapeInferenceType()
H A DNetwork.hpp305 , m_shapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly) in OptimizerOptionsOpaqueImpl()
321 , m_shapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly)
331 ShapeInferenceMethod shapeInferenceMethod, in OptimizerOptionsOpaqueImpl()
365 ShapeInferenceMethod m_shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly;
H A DLayer.hpp242 ShapeInferenceMethod GetShapeInferenceMethod() const { return m_ShapeInferenceMethod; }; in GetShapeInferenceMethod()
344 void SetShapeInferenceMethod(ShapeInferenceMethod shapeInferenceMethod) in SetShapeInferenceMethod()
388 const ShapeInferenceMethod shapeInferenceMethod,
392 …void VerifyShapeInferenceType(const TensorShape& outputShape, ShapeInferenceMethod shapeInferenceM…
427 ShapeInferenceMethod m_ShapeInferenceMethod;
H A DGraph.hpp101 , m_ShapeInferenceMethod(shapeInferenceMethod ? ShapeInferenceMethod::InferAndValidate : in Graph()
102 ShapeInferenceMethod::ValidateOnly) in Graph()
291 ShapeInferenceMethod m_ShapeInferenceMethod;
H A DNetwork.cpp70 ShapeInferenceMethod shapeInferenceMethod, in OptimizerOptionsOpaque()
140 void OptimizerOptionsOpaque::SetShapeInferenceMethod(armnn::ShapeInferenceMethod ShapeInferenceMeth… in SetShapeInferenceMethod()
200 armnn::ShapeInferenceMethod OptimizerOptionsOpaque::GetShapeInferenceMethod() const in GetShapeInferenceMethod()
214 (p_OptimizerOptionsImpl->m_shapeInferenceMethod == ShapeInferenceMethod::ValidateOnly ? in ToString()
1837 if(options.GetShapeInferenceMethod() == ShapeInferenceMethod::InferAndValidate) in Optimize()
1847 if(options.GetShapeInferenceMethod() == ShapeInferenceMethod::ValidateOnly) in Optimize()
H A DGraph.cpp593 if (layer->m_ShapeInferenceMethod == ShapeInferenceMethod::ValidateOnly) in InferTensorInfos()
/aosp_15_r20/external/armnn/tests/ExecuteNetwork/
H A DExecuteNetworkParams.cpp145 optimizerOptions.SetShapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly); in ToDelegateOptions()
148 optimizerOptions.SetShapeInferenceMethod(armnn::ShapeInferenceMethod::InferAndValidate); in ToDelegateOptions()
H A DArmNNExecutor.cpp526 armnn::ShapeInferenceMethod::InferAndValidate : in OptimizeNetwork()
527 armnn::ShapeInferenceMethod::ValidateOnly); in OptimizeNetwork()
/aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
H A Darmnn_network.i53ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly,
61 ShapeInferenceMethod m_shapeInferenceMethod;
/aosp_15_r20/external/armnn/src/armnn/test/
H A DShapeInferenceTests.cpp86 layer->SetShapeInferenceMethod(ShapeInferenceMethod::ValidateOnly); in RunShapeInferenceTest()
90 layer->SetShapeInferenceMethod(ShapeInferenceMethod::InferAndValidate); in RunShapeInferenceTest()
609 layer->SetShapeInferenceMethod(ShapeInferenceMethod::InferAndValidate);
/aosp_15_r20/external/armnn/tests/
H A DInferenceModel.hpp463 … armnn::ShapeInferenceMethod::InferAndValidate : armnn::ShapeInferenceMethod::ValidateOnly); in InferenceModel()
/aosp_15_r20/external/armnn/docs/
H A D05_05_runtimeoptions.dox72 ShapeInferenceMethod | infer-output-shape | (Not Available) | ["true"/"false"] | Infers output tens…