Home
last modified time | relevance | path

Searched refs:device_peak_memory (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc555 std::unordered_map<string, uint64> device_peak_memory; in TEST_F() local
556 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()
557 EXPECT_EQ(device_peak_memory.size(), 1); in TEST_F()
558 EXPECT_GT(device_peak_memory.begin()->second, 0); in TEST_F()
581 std::unordered_map<string, uint64> device_peak_memory; in TEST_F() local
582 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()
584 device_peak_memory.find("/job:localhost/replica:0/task:0/device:CPU:0"), in TEST_F()
585 device_peak_memory.end()); in TEST_F()
587 device_peak_memory["/job:localhost/replica:0/task:0/device:CPU:0"]; in TEST_F()
592 device_peak_memory.clear(); in TEST_F()
[all …]
Dsingle_machine.cc216 std::unordered_map<string, uint64>* device_peak_memory) const { in GetPeakMemoryUsage()
228 device_peak_memory->clear(); in GetPeakMemoryUsage()
236 (*device_peak_memory)[device->name()] = in GetPeakMemoryUsage()
Dcluster.h116 std::unordered_map<string, uint64>* device_peak_memory) const { in GetPeakMemoryUsage() argument
Dsingle_machine.h52 std::unordered_map<string, uint64>* device_peak_memory) const override;