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