From dec51c5795298c149bc5defcbb177e733da56072 Mon Sep 17 00:00:00 2001 From: Josh Fromm Date: Sat, 10 Apr 2021 18:02:50 -0700 Subject: [PATCH] Add logical_not shape registration. (#7820) Co-authored-by: Ubuntu --- python/tvm/relay/op/_tensor.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/tvm/relay/op/_tensor.py b/python/tvm/relay/op/_tensor.py index 5f68be84d46a..d7d99c017b2b 100644 --- a/python/tvm/relay/op/_tensor.py +++ b/python/tvm/relay/op/_tensor.py @@ -282,3 +282,4 @@ def elemwise_shape_func(attrs, inputs, _): register_shape_func("log2", False, elemwise_shape_func) register_shape_func("sigmoid", False, elemwise_shape_func) register_shape_func("tanh", False, elemwise_shape_func) +register_shape_func("logical_not", False, elemwise_shape_func)