You could decorate the method with @amp.float_function 11 (maybe you would have to wrap it into another helper method), which should disable the casting.



 kkjh0723:

Can this issue be resolved in nightly version?


This might be the case. If you see that it’s not working in native amp, let us know, please.