tfra.dynamic_embedding.enable_inference_mode View source on GitHub set inference mode. tfra.dynamic_embedding.enable_inference_mode()