Equivalent of tf.nn.dilation2d

vision
2

I don’t think a dilation is implemented yet.
You could unfold the image tensor and apply the max operation on each patch.
Have a look at this example to see how to create the patches.