MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeedExamples / forward

Method forward

Megatron-LM/fp16/fp16util.py:97–99  ·  view source on GitHub ↗
(self, *inputs)

Source from the content-addressed store, hash-verified

95 self.network = convert_network(network, dtype=torch.half)
96
97 def forward(self, *inputs):
98 inputs = tuple(t.half() for t in inputs)
99 return self.network(*inputs)
100
101
102def backwards_debug_hook(grad):

Callers

nothing calls this directly

Calls 1

halfMethod · 0.80

Tested by

no test coverage detected