MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / StartStep

Method StartStep

tensorflow/core/common_runtime/eager/context.cc:348–359  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

346}
347
348void EagerContext::StartStep() {
349 mutex_lock ml(metadata_mu_);
350 num_active_steps_++;
351 if (step_container_ == nullptr) {
352 step_container_.reset(
353 new ScopedStepContainer(0, [this](const string& name) {
354 for (Device* device : devices_) {
355 device->resource_manager()->Cleanup(name).IgnoreError();
356 }
357 }));
358 }
359}
360
361void EagerContext::EndStep() {
362 mutex_lock ml(metadata_mu_);

Callers 1

TFE_ContextStartStepFunction · 0.45

Calls 4

resetMethod · 0.45
IgnoreErrorMethod · 0.45
CleanupMethod · 0.45
resource_managerMethod · 0.45

Tested by

no test coverage detected