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

Пример #1

 def testPrintREFTypeTensor(self):
   ref_tensor_info = meta_graph_pb2.TensorInfo()
   ref_tensor_info.dtype = types_pb2.DT_FLOAT_REF
   with captured_output() as (out, err):
     saved_model_cli._print_tensor_info(ref_tensor_info)
   self.assertTrue('DT_FLOAT_REF' in out.getvalue().strip())
   self.assertEqual(err.getvalue().strip(), '')

Пример #2

Файл: saved_model_cli_test.py Проект: DILASSS/tensorflow

 def testPrintREFTypeTensor(self):
   ref_tensor_info = meta_graph_pb2.TensorInfo()
   ref_tensor_info.dtype = types_pb2.DT_FLOAT_REF
   with captured_output() as (out, err):
     saved_model_cli._print_tensor_info(ref_tensor_info)
   self.assertTrue('DT_FLOAT_REF' in out.getvalue().strip())
   self.assertEqual(err.getvalue().strip(), '')