MCPcopy Create free account
hub / github.com/apple/axlearn / invoke_shared

Method invoke_shared

axlearn/common/module_test.py:405–407  ·  view source on GitHub ↗
(self, *args)

Source from the content-addressed store, hash-verified

403 return x * shared_module.state["weight"]
404
405 def invoke_shared(self, *args):
406 shared_module = self.get_shared_module("shared")
407 return shared_module.module(*args)
408
409
410class ModuleProvidingSharedChild(Module):

Callers 1

invoke_sharedMethod · 0.45

Calls 1

get_shared_moduleMethod · 0.80

Tested by

no test coverage detected