Hi, thanks for your for implementing and sharing. I am wondering how the backward pass of warping layer working. As far as I am concerned, Floor() operation is nondifferentiable. Do we need to use gradient_override_map to substitute gradient of Identity op for Floor ?