Python _compute_device_summary_from_list примеры использования

Пример #1

Файл: error_interpolation_test.py Проект: jrse/tensorflow-raspberry-pi

  def testCorrectFormatWithActiveDeviceAssignments(self):
    assignments = []
    assignments.append(
        traceable_stack.TraceableObject(
            "/cpu:0", filename="hope.py", lineno=24))
    assignments.append(
        traceable_stack.TraceableObject(
            "/gpu:2", filename="please.py", lineno=42))

    summary = error_interpolation._compute_device_summary_from_list(
        "nodename", assignments, prefix="  ")

    self.assertIn("nodename", summary)
    self.assertIn("tf.device(/cpu:0)", summary)
    self.assertIn("<hope.py:24>", summary)
    self.assertIn("tf.device(/gpu:2)", summary)
    self.assertIn("<please.py:42>", summary)

Пример #2

Файл: error_interpolation_test.py Проект: wwjiang007/tensorflow

 def testCorrectFormatWhenNoColocationsWereActive(self):
     device_assignment_list = []
     summary = error_interpolation._compute_device_summary_from_list(
         "nodename", device_assignment_list, prefix="  ")
     self.assertIn("nodename", summary)
     self.assertIn("No device assignments", summary)

Пример #3

Файл: error_interpolation_test.py Проект: mbrukman/tensorflow

 def testCorrectFormatWhenNoColocationsWereActive(self):
   device_assignment_list = []
   summary = error_interpolation._compute_device_summary_from_list(
       "nodename", device_assignment_list, prefix="  ")
   self.assertIn("nodename", summary)
   self.assertIn("No device assignments", summary)