We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 837b1e5 commit eb938adCopy full SHA for eb938ad
tests/python/relay/test_adt.py
@@ -738,11 +738,11 @@ def run(dtype):
738
run('int32')
739
740
def vmobj_to_list(o):
741
- if isinstance(o, tvm.relay.backend.vmobj.TensorObject):
+ if isinstance(o, tvm.relay.backend.vmobj.Tensor):
742
return [o.asnumpy().tolist()]
743
elif isinstance(o, tvm.relay.backend.interpreter.TensorValue):
744
return [o.asnumpy()]
745
- elif isinstance(o, tvm.relay.backend.vmobj.DatatypeObject):
+ elif isinstance(o, tvm.relay.backend.vmobj.Datatype):
746
result = []
747
for f in o:
748
result.extend(vmobj_to_list(f))
0 commit comments