From 956dd34a1ebd7938e4b2bf7fe24482b6ee63a5b3 Mon Sep 17 00:00:00 2001 From: daquexian Date: Fri, 1 Jul 2022 18:10:45 +0800 Subject: [PATCH] add is_constant_initializer Signed-off-by: daquexian --- onnx/common/ir.h | 3 +++ 1 file changed, 3 insertions(+) diff --git a/onnx/common/ir.h b/onnx/common/ir.h index 5be163cbcad..e066001a945 100644 --- a/onnx/common/ir.h +++ b/onnx/common/ir.h @@ -994,6 +994,9 @@ struct Graph final { } return initializers_.end(); } + bool is_constant_initializer(const Value* value) const { + return value->node() == initializer_node_; + } ArrayRef inputs() { return input_->outputs(); }