OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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