MCPcopy Create free account
hub / github.com/ModalityDance/Omni-R1 / reset

Method reset

src/transformers/src/transformers/cache_utils.py:1655–1657  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

1653 return self.ssm_states[layer_idx]
1654
1655 def reset(self):
1656 self.conv_states.zero_()
1657 self.ssm_states.zero_()

Callers 9

resetMethod · 0.45
test_generate_compile_fullgraphMethod · 0.45
test_torchdynamo_full_evalMethod · 0.45
test_torchdynamo_memoryMethod · 0.45
test_compile_static_cacheMethod · 0.45
create_and_check_falcon_mamba_cached_slow_forward_and_backwardsMethod · 0.45
create_and_check_mamba_cached_slow_forward_and_backwardsMethod · 0.45
run_decision_transformer.pyFile · 0.45
reset_metricMethod · 0.45

Calls

no outgoing calls

Tested by 6

test_generate_compile_fullgraphMethod · 0.36
test_torchdynamo_full_evalMethod · 0.36
test_torchdynamo_memoryMethod · 0.36
test_compile_static_cacheMethod · 0.36
create_and_check_falcon_mamba_cached_slow_forward_and_backwardsMethod · 0.36
create_and_check_mamba_cached_slow_forward_and_backwardsMethod · 0.36