dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r431854463
##
File path: topi/include/topi/transform.h
##
@@ -1309,5 +1309,57 @@ inline Tensor one_hot(const Tensor& indices, const
PrimExpr on_value,
dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r431854463
##
File path: topi/include/topi/transform.h
##
@@ -1309,5 +1309,57 @@ inline Tensor one_hot(const Tensor& indices, const
PrimExpr on_value,
dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r430402869
##
File path: topi/include/topi/transform.h
##
@@ -1309,5 +1309,57 @@ inline Tensor one_hot(const Tensor& indices, const
PrimExpr on_value,
dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r416614304
##
File path: tests/python/relay/test_op_level3.py
##
@@ -780,4 +826,5 @@ def verify_unravel_index(indices, shape, dtype):
test_gather_nd()
dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r416524977
##
File path: topi/python/topi/transform.py
##
@@ -676,3 +676,30 @@ def unravel_index(indices, shape):
"""
return
dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r416524442
##
File path: topi/include/topi/transform.h
##
@@ -1369,5 +1369,57 @@ inline Tensor one_hot(const Tensor& indices,
}, name, tag);
}
+/*!
+
dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r416524653
##
File path: topi/tests/python/test_topi_transform.py
##
@@ -595,6 +595,41 @@ def check_device(device):
for device in get_all_backend():
dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r416523256
##
File path: src/relay/op/tensor/transform.cc
##
@@ -2732,5 +2732,87 @@ Example::
.set_attr("FTVMCompute", UnRavelIndexCompute)
dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r416522950
##
File path: python/tvm/relay/op/transform.py
##
@@ -884,3 +884,29 @@ def unravel_index(indices, shape):
"""
return
dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r416523113
##
File path: src/relay/op/tensor/transform.cc
##
@@ -2732,5 +2732,87 @@ Example::
.set_attr("FTVMCompute", UnRavelIndexCompute)
dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r416522181
##
File path: python/tvm/relay/frontend/tflite.py
##
@@ -2075,6 +2077,29 @@ def convert_space_to_depth(self, op):
return out
+
dhruvaray commented on a change in pull request #5447:
URL: https://github.com/apache/incubator-tvm/pull/5447#discussion_r416519580
##
File path: python/tvm/relay/frontend/tflite.py
##
@@ -2075,6 +2077,29 @@ def convert_space_to_depth(self, op):
return out
+
12 matches
Mail list logo