Searched refs:ShapeInferenceMethod (Results 1 – 8 of 8) sorted by relevance
70 if (GetOwningLayer().GetShapeInferenceMethod() == ShapeInferenceMethod::InferAndValidate) in IsTensorInfoSet()197 , m_ShapeInferenceMethod(ShapeInferenceMethod::ValidateOnly) in Layer()394 const ShapeInferenceMethod shapeInferenceMethod, in ValidateAndCopyShape()398 if (shapeInferenceMethod == ShapeInferenceMethod::ValidateOnly) in ValidateAndCopyShape()432 void Layer::VerifyShapeInferenceType(const TensorShape& outputShape, ShapeInferenceMethod shapeInfe… in VerifyShapeInferenceType()434 if (shapeInferenceMethod == ShapeInferenceMethod::ValidateOnly) in VerifyShapeInferenceType()
232 ShapeInferenceMethod GetShapeInferenceMethod() const { return m_ShapeInferenceMethod; }; in GetShapeInferenceMethod()333 void SetShapeInferenceMethod(ShapeInferenceMethod shapeInferenceMethod) in SetShapeInferenceMethod()370 const ShapeInferenceMethod shapeInferenceMethod,374 …void VerifyShapeInferenceType(const TensorShape& outputShape, ShapeInferenceMethod shapeInferenceM…405 ShapeInferenceMethod m_ShapeInferenceMethod;
97 , m_ShapeInferenceMethod(shapeInferenceMethod ? ShapeInferenceMethod::InferAndValidate : in Graph()98 ShapeInferenceMethod::ValidateOnly) in Graph()266 ShapeInferenceMethod m_ShapeInferenceMethod;
552 if (layer->m_ShapeInferenceMethod == ShapeInferenceMethod::ValidateOnly) in InferTensorInfos()
622 , m_shapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly) in OptimizerOptions()632 , m_shapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly)643 … ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly, in OptimizerOptions()668 ShapeInferenceMethod m_shapeInferenceMethod;
169 enum class ShapeInferenceMethod enum
28 if (m_ShapeInferenceMethod == ShapeInferenceMethod::ValidateOnly) in InferOutputShapes()32 else if (m_ShapeInferenceMethod == ShapeInferenceMethod::InferAndValidate && in InferOutputShapes()
85 layer->SetShapeInferenceMethod(ShapeInferenceMethod::ValidateOnly); in RunShapeInferenceTest()89 layer->SetShapeInferenceMethod(ShapeInferenceMethod::InferAndValidate); in RunShapeInferenceTest()623 layer->SetShapeInferenceMethod(ShapeInferenceMethod::InferAndValidate); in BOOST_AUTO_TEST_CASE()