The function fails when one of the intermediate tensors is not float32
The function fails when one of the intermediate tensors is not float32