Searched refs:ShapeInferenceMethod (Results 1 – 14 of 14) sorted by relevance
/aosp_15_r20/external/armnn/include/armnn/ |
H A D | INetwork.hpp | 147 , m_shapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly) in OptimizerOptions() 162 , m_shapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly) 173 … ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly, in OptimizerOptions() 198 (m_shapeInferenceMethod == ShapeInferenceMethod::ValidateOnly in ToString() 238 ShapeInferenceMethod m_shapeInferenceMethod; 275 … ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly, 299 armnn::ShapeInferenceMethod GetShapeInferenceMethod() const; 313 void SetShapeInferenceMethod(armnn::ShapeInferenceMethod ShapeInferenceMethodType);
|
H A D | Types.hpp | 232 enum class ShapeInferenceMethod enum
|
/aosp_15_r20/external/armnn/src/armnn/ |
H A D | Layer.cpp | 99 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 D | Network.hpp | 305 , 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 D | Layer.hpp | 242 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 D | Graph.hpp | 101 , m_ShapeInferenceMethod(shapeInferenceMethod ? ShapeInferenceMethod::InferAndValidate : in Graph() 102 ShapeInferenceMethod::ValidateOnly) in Graph() 291 ShapeInferenceMethod m_ShapeInferenceMethod;
|
H A D | Network.cpp | 70 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 D | Graph.cpp | 593 if (layer->m_ShapeInferenceMethod == ShapeInferenceMethod::ValidateOnly) in InferTensorInfos()
|
/aosp_15_r20/external/armnn/tests/ExecuteNetwork/ |
H A D | ExecuteNetworkParams.cpp | 145 optimizerOptions.SetShapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly); in ToDelegateOptions() 148 optimizerOptions.SetShapeInferenceMethod(armnn::ShapeInferenceMethod::InferAndValidate); in ToDelegateOptions()
|
H A D | ArmNNExecutor.cpp | 526 armnn::ShapeInferenceMethod::InferAndValidate : in OptimizeNetwork() 527 armnn::ShapeInferenceMethod::ValidateOnly); in OptimizeNetwork()
|
/aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
H A D | armnn_network.i | 53 … ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly, 61 ShapeInferenceMethod m_shapeInferenceMethod;
|
/aosp_15_r20/external/armnn/src/armnn/test/ |
H A D | ShapeInferenceTests.cpp | 86 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 D | InferenceModel.hpp | 463 … armnn::ShapeInferenceMethod::InferAndValidate : armnn::ShapeInferenceMethod::ValidateOnly); in InferenceModel()
|
/aosp_15_r20/external/armnn/docs/ |
H A D | 05_05_runtimeoptions.dox | 72 ShapeInferenceMethod | infer-output-shape | (Not Available) | ["true"/"false"] | Infers output tens…
|