Searched refs:EnablePeakMemoryStats (Results 1 – 4 of 4) sorted by relevance
108 virtual Status EnablePeakMemoryStats() { in EnablePeakMemoryStats() function
48 Status EnablePeakMemoryStats() override;
205 Status SingleMachine::EnablePeakMemoryStats() { in EnablePeakMemoryStats() function in tensorflow::grappler::SingleMachine
54 TF_CHECK_OK(cluster_->EnablePeakMemoryStats()); in SetUp()