Does TVM currently support TensorFlow Probability 
(https://github.com/tensorflow/probability), specifically tfp.layers 
(https://github.com/tensorflow/probability/tree/main/tensorflow_probability/python/layers)?

I am working with Bayesian Neural Network models 
(https://github.com/tensorflow/probability/blob/main/tensorflow_probability/examples/bayesian_neural_network.py)
 and would love to use TVM for the same. 

If not TensorFlow Probability, then does TVM support Bayesian Neural networks 
from other frameworks (MxNet, Pytorch etc.)?

My specific doubt is related to importing pre-trained Bayesian models into TVM 
where weights are not fixed point values but rather probability distributions.

Thanks for your help on this.





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/tvm-support-for-tensorflow-probability/10379/1)
 to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.apache.org/email/unsubscribe/2f7bef4f0fcb67d53d029bc85dc4d3bfb1ce3073c038ee3885a88b867de464c0).

Reply via email to