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

Method MemoryPlannerFactory

tensorflow/core/common_runtime/memory_planner.cc:475–486  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

473}
474
475MemoryPlannerFactory::MemoryPlannerFactory() {
476 // Enable Memory Optimization by default
477 Status s = ReadBoolFromEnvVar("ENABLE_MEMORY_OPTIMIZATION",
478 true,
479 &enable_memory_opt_);
480 if (enable_memory_opt_) {
481 //LOG(INFO_DEV) << "Enable Memory Optimization!";
482 memory_planner_ = new MemoryPlanner();
483 } else {
484 memory_planner_ = new NullableMemoryPlanner();
485 }
486}
487
488} // tensorflow

Callers

nothing calls this directly

Calls 1

ReadBoolFromEnvVarFunction · 0.50

Tested by

no test coverage detected