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

Method inc

axlearn/common/module_test.py:365–369  ·  view source on GitHub ↗
(self, x: Tensor)

Source from the content-addressed store, hash-verified

363 return x
364
365 def inc(self, x: Tensor) -> Tensor:
366 self.add_summary("x", x)
367 y = x + self.state["inc"]
368 self.add_module_output("y", y)
369 return y
370
371 def invoke_self_multiple_times_causing_conflict(self, x: Tensor, n: int) -> Tensor:
372 for _ in range(n):

Callers 4

forwardMethod · 0.95
forward2Method · 0.95

Calls 2

add_summaryMethod · 0.45
add_module_outputMethod · 0.45

Tested by

no test coverage detected