tqchen commented on code in PR #17010:
URL: https://github.com/apache/tvm/pull/17010#discussion_r1606763675
##########
python/tvm/relax/op/index.py:
##########
@@ -134,3 +134,64 @@ def dynamic_strided_slice(
same length as rank of `data` tensor.
"""
return _ffi_api.dynamic_strided_slice(x, begin, end, strides) # type:
ignore
+
+
+@args_converter.auto
+def multinomial_from_uniform(
Review Comment:
move to sampling.py
##########
include/tvm/relax/attrs/index.h:
##########
@@ -51,6 +51,16 @@ struct StridedSliceAttrs : public
tvm::AttrsNode<StridedSliceAttrs> {
}
}; // struct StridedSliceAttrs
+struct MultinomialFromUniformAttrs : public
tvm::AttrsNode<MultinomialFromUniformAttrs> {
+ DataType dtype;
+
Review Comment:
sampling.h
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
