Variational inference for Conv2D layers

[haniehmsv]

Hello, I have been trying to use variational inference to the MLP and CNN (conv2D) layers for my research. Unlike a comprehensive discussion and examples for probabilistic MLP layers, I could not find any resources implying whether tensorflow_probability has also released VI for the conv2D layers. My network is built of a combination of MLP-CNN layers and I need to define probabilistic layers for all types of layers in my code. I would appreciate it if you could guide me through finding VI for CNN.