Ejemplos de _compute_colocation_summary_from_dict en Python

Ejemplo n.º 1

Archivo: error_interpolation_test.py Proyecto: jrse/tensorflow-raspberry-pi

 def testCorrectFormatWithActiveColocations(self):
   t_obj_1 = traceable_stack.TraceableObject(
       None, filename="test_1.py", lineno=27)
   t_obj_2 = traceable_stack.TraceableObject(
       None, filename="test_2.py", lineno=38)
   colocation_dict = {
       "test_node_1": t_obj_1,
       "test_node_2": t_obj_2,
   }
   summary = error_interpolation._compute_colocation_summary_from_dict(
       "node_name", colocation_dict, prefix="  ")
   self.assertIn("node_name", summary)
   self.assertIn("colocate_with(test_node_1)", summary)
   self.assertIn("<test_1.py:27>", summary)
   self.assertIn("colocate_with(test_node_2)", summary)
   self.assertIn("<test_2.py:38>", summary)

Ejemplo n.º 2

Archivo: error_interpolation_test.py Proyecto: zyy2020/tensorflow

 def testCorrectFormatWithActiveColocations(self):
   t_obj_1 = traceable_stack.TraceableObject(None,
                                             filename="test_1.py",
                                             lineno=27)
   t_obj_2 = traceable_stack.TraceableObject(None,
                                             filename="test_2.py",
                                             lineno=38)
   colocation_dict = {
       "test_node_1": t_obj_1,
       "test_node_2": t_obj_2,
   }
   summary = error_interpolation._compute_colocation_summary_from_dict(
       colocation_dict, prefix="  ")
   assert_node_in_colocation_summary(self,
                                     summary,
                                     name="test_node_1",
                                     filename="test_1.py",
                                     lineno=27)
   assert_node_in_colocation_summary(self, summary,
                                     name="test_node_2",
                                     filename="test_2.py",
                                     lineno=38)

Ejemplo n.º 3

Archivo: error_interpolation_test.py Proyecto: wwjiang007/tensorflow

 def testCorrectFormatWhenNoColocationsWereActive(self):
     colocation_dict = {}
     summary = error_interpolation._compute_colocation_summary_from_dict(
         "node_name", colocation_dict, prefix="  ")
     self.assertIn("node_name", summary)
     self.assertIn("No node-device colocations", summary)

Ejemplo n.º 4

Archivo: error_interpolation_test.py Proyecto: mbrukman/tensorflow

 def testCorrectFormatWhenNoColocationsWereActive(self):
   colocation_dict = {}
   summary = error_interpolation._compute_colocation_summary_from_dict(
       "node_name", colocation_dict, prefix="  ")
   self.assertIn("node_name", summary)
   self.assertIn("No node-device colocations", summary)