[GitHub] [tvm] masahi commented on a change in pull request #8985: [ONNX] Add Einsum converter

2021-09-15 Thread GitBox
masahi commented on a change in pull request #8985: URL: https://github.com/apache/tvm/pull/8985#discussion_r709533999 ## File path: python/tvm/relay/op/strategy/cuda.py ## @@ -1210,3 +1210,16 @@ def invert_permutation_strategy_cuda(attrs, inputs, out_type, target): n

[GitHub] [tvm] masahi commented on a change in pull request #8985: [ONNX] Add Einsum converter

2021-09-15 Thread GitBox
masahi commented on a change in pull request #8985: URL: https://github.com/apache/tvm/pull/8985#discussion_r709112076 ## File path: python/tvm/relay/op/strategy/cuda.py ## @@ -1210,3 +1210,16 @@ def invert_permutation_strategy_cuda(attrs, inputs, out_type, target): n

[GitHub] [tvm] masahi commented on a change in pull request #8985: [ONNX] Add Einsum converter

2021-09-15 Thread GitBox
masahi commented on a change in pull request #8985: URL: https://github.com/apache/tvm/pull/8985#discussion_r709112076 ## File path: python/tvm/relay/op/strategy/cuda.py ## @@ -1210,3 +1210,16 @@ def invert_permutation_strategy_cuda(attrs, inputs, out_type, target): n

[GitHub] [tvm] masahi commented on a change in pull request #8985: [ONNX] Add Einsum converter

2021-09-10 Thread GitBox
masahi commented on a change in pull request #8985: URL: https://github.com/apache/tvm/pull/8985#discussion_r706557494 ## File path: src/relay/op/tensor/transform.cc ## @@ -2431,6 +2432,85 @@ RELAY_REGISTER_OP("broadcast_to_like") .set_attr("FTVMCompute", BroadCastToLikeCo

[GitHub] [tvm] masahi commented on a change in pull request #8985: [ONNX] Add Einsum converter

2021-09-10 Thread GitBox
masahi commented on a change in pull request #8985: URL: https://github.com/apache/tvm/pull/8985#discussion_r706557112 ## File path: python/tvm/relay/op/_transform.py ## @@ -182,6 +182,11 @@ def compute_unique(attrs, inputs, output_type): _reg.register_strategy("invert_permuta

[GitHub] [tvm] masahi commented on a change in pull request #8985: [ONNX] Add Einsum converter

2021-09-10 Thread GitBox
masahi commented on a change in pull request #8985: URL: https://github.com/apache/tvm/pull/8985#discussion_r706557112 ## File path: python/tvm/relay/op/_transform.py ## @@ -182,6 +182,11 @@ def compute_unique(attrs, inputs, output_type): _reg.register_strategy("invert_permuta

[GitHub] [tvm] masahi commented on a change in pull request #8985: [ONNX] Add Einsum converter

2021-09-10 Thread GitBox
masahi commented on a change in pull request #8985: URL: https://github.com/apache/tvm/pull/8985#discussion_r706556993 ## File path: python/tvm/topi/generic/search.py ## @@ -86,3 +86,19 @@ def schedule_unique(outs): The computation schedule for the op. """ retu