HomeSort by relevance Sort by last modified time
    Searched refs:GetPeakMemoryUsage (Results 1 - 12 of 12) sorted by null

  /external/tensorflow/tensorflow/core/grappler/costs/
graph_memory_test.cc 55 memory.GetPeakMemoryUsage("/CPU:0");
85 memory.GetPeakMemoryUsage("/CPU:0");
110 const GraphMemory::MemoryUsage& cpu_mem = memory.GetPeakMemoryUsage("/CPU:0");
123 const GraphMemory::MemoryUsage& gpu_mem = memory.GetPeakMemoryUsage("/GPU:0");
150 memory.GetPeakMemoryUsage("/GPU:0");
177 memory.GetPeakMemoryUsage("/GPU:0");
213 const GraphMemory::MemoryUsage& mem = memory.GetPeakMemoryUsage("/CPU:0");
graph_memory.h 55 const MemoryUsage& GetPeakMemoryUsage(const string& device) const {
virtual_scheduler.h 292 const std::unordered_map<string, int64> GetPeakMemoryUsage() const;
virtual_scheduler.cc     [all...]
  /external/tensorflow/tensorflow/core/grappler/clusters/
cluster.h 115 virtual Status GetPeakMemoryUsage(
118 "GetPeakMemoryUsage is not implemented for this type of cluster.");
single_machine.h 51 Status GetPeakMemoryUsage(
virtual_cluster.cc 91 estimator_->GetScheduler()->GetPeakMemoryUsage();
single_machine_test.cc 546 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory_before));
556 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory));
564 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory_after));
582 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory));
593 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory));
615 Status s = cluster.GetPeakMemoryUsage(&device_peak_memory);
single_machine.cc 215 Status SingleMachine::GetPeakMemoryUsage(
  /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
trt_optimization_pass.cc 99 auto status = cluster->GetPeakMemoryUsage(&peak_mem);
  /external/tensorflow/tensorflow/python/grappler/
cluster.i 411 memory.GetPeakMemoryUsage(device.first);
  /external/tensorflow/tensorflow/core/grappler/optimizers/
memory_optimizer.cc 552 const GraphMemory::MemoryUsage& mem_usage = memory.GetPeakMemoryUsage(name);
    [all...]

Completed in 134 milliseconds