Add logical_not shape registration. (apache#7820)

Co-authored-by: Ubuntu <jwfromm@jwfromm-cpu-dev.itxhlkosmouevgkdrmwxfbs5qh.xx.internal.cloudapp.net>
jinqiua · Apr 11, 2021 · dec51c5 · dec51c5
1 parent 461d06e
commit dec51c5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)