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

Method AddCPUFreeVisitor

tensorflow/core/common_runtime/process_state.cc:134–140  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

132}
133
134void ProcessState::AddCPUFreeVisitor(SubAllocator::Visitor visitor) {
135 mutex_lock lock(mu_);
136 CHECK_EQ(0, cpu_allocators_.size()) // Crash OK
137 << "AddCPUFreeVisitor must be called prior to first call to "
138 "ProcessState::GetCPUAllocator";
139 cpu_free_visitors_.push_back(std::move(visitor));
140}
141
142void ProcessState::TestOnlyReset() {
143 mutex_lock lock(mu_);

Callers 2

InitMethod · 0.80
RegMemVisitorsMethod · 0.80

Calls 2

sizeMethod · 0.45
push_backMethod · 0.45

Tested by

no test coverage detected