Add support for custom message in Assert.
Add assert in the conv2d_fft graph.