[GitHub] [incubator-tvm] zhiics commented on a change in pull request #5855: [RELAY][VM] Add shape_of instruction

zhiics commented on a change in pull request #5855: URL: https://github.com/apache/incubator-tvm/pull/5855#discussion_r446401587 ## File path: src/relay/op/vm/vm.cc ## @@ -0,0 +1,62 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor

[GitHub] [incubator-tvm] zhiics commented on a change in pull request #5855: [RELAY][VM] Add shape_of instruction

zhiics commented on a change in pull request #5855: URL: https://github.com/apache/incubator-tvm/pull/5855#discussion_r446250485 ## File path: python/tvm/relay/op/dialect/vm.py ## @@ -0,0 +1,35 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more

[GitHub] [incubator-tvm] zhiics commented on a change in pull request #5855: [RELAY][VM] Add shape_of instruction

zhiics commented on a change in pull request #5855: URL: https://github.com/apache/incubator-tvm/pull/5855#discussion_r444506282 ## File path: python/tvm/relay/op/memory/memory.py ## @@ -104,6 +104,21 @@ def shape_func(func, inputs, outputs, dependent=False): """

[GitHub] [incubator-tvm] zhiics commented on a change in pull request #5855: [RELAY][VM] Add shape_of instruction

zhiics commented on a change in pull request #5855: URL: https://github.com/apache/incubator-tvm/pull/5855#discussion_r93947 ## File path: python/tvm/relay/op/memory/memory.py ## @@ -104,6 +104,21 @@ def shape_func(func, inputs, outputs, dependent=False): """

[GitHub] [incubator-tvm] zhiics commented on a change in pull request #5855: [RELAY][VM] Add shape_of instruction

zhiics commented on a change in pull request #5855: URL: https://github.com/apache/incubator-tvm/pull/5855#discussion_r444355425 ## File path: tests/python/relay/test_vm_serialization.py ## @@ -332,6 +284,20 @@ def test_mobilenet(): run_network(mod, params) +def

[GitHub] [incubator-tvm] zhiics commented on a change in pull request #5855: [RELAY][VM] Add shape_of instruction

zhiics commented on a change in pull request #5855: URL: https://github.com/apache/incubator-tvm/pull/5855#discussion_r444325734 ## File path: src/relay/op/memory/memory.cc ## @@ -423,5 +427,24 @@ RELAY_REGISTER_OP("memory.shape_func") return