MCPcopy
hub / github.com/tensorflow/models / pop

Method pop

official/projects/bigbird/recompute_grad.py:68–71  ·  view source on GitHub ↗
(self, context: RecomputeContext)

Source from the content-addressed store, hash-verified

66 return context
67
68 def pop(self, context: RecomputeContext):
69 if self._stack[-1] is not context:
70 raise AssertionError('Nesting violated for RecomputeContext.')
71 self._stack.pop()
72
73
74_context_stack = _ContextStack()

Callers 15

train_loop_endMethod · 0.80
from_configMethod · 0.80
from_configMethod · 0.80
set_cudnn_batchnorm_modeFunction · 0.80
from_nameMethod · 0.80
_replicated_stepFunction · 0.80
_truncate_seq_pairFunction · 0.80
parserFunction · 0.80
parserFunction · 0.80
_get_sorted_inputsFunction · 0.80
_run_and_report_benchmarkMethod · 0.80
test_movinet_export_a0_stream_with_tfliteMethod · 0.80

Calls

no outgoing calls

Tested by 5

test_movinet_export_a0_stream_with_tfliteMethod · 0.64
test_self_attention_widening_factorMethod · 0.64
test_cross_attention_widening_factorMethod · 0.64
testForwardMethod · 0.64
test_hidden_cls_listMethod · 0.64