Endpoints — github.com/NVIDIA/apex

CodeHub MCPcopy Create free account

hub / github.com/NVIDIA/apex / endpoints

Endpoints2 in github.com/NVIDIA/apex

⨍ Functions 1,387◇ Types & classes 224↳ Endpoints 2

torch.amp.custom_fwd(cast_inputs=torch.half, device_type="cuda")

apex/contrib/conv_bias_relu/conv_bias_relu.py:None

Routetest_fp16_output

unittest.skip("No longer support output fp16 param")

tests/L0/run_optimizers/test_fused_optimizer.py:None