Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/clusters/
Dcluster.h108 virtual Status EnablePeakMemoryStats(bool enable) { in EnablePeakMemoryStats() function
Dsingle_machine.h48 Status EnablePeakMemoryStats(bool enable) override;
Dsingle_machine.cc208 Status SingleMachine::EnablePeakMemoryStats(bool enable) { in EnablePeakMemoryStats() function in tensorflow::grappler::SingleMachine
Dsingle_machine_test.cc48 TF_CHECK_OK(cluster_->EnablePeakMemoryStats(true)); in SetUp()