Python check_onnx_broadcast Exemples

Exemple #1

Fichier : test_utils.py Projet : thiagocrepaldi/pytorch

 def try_check_onnx_broadcast(dims1, dims2, expect_broadcast, expect_fail):
     broadcast = True
     fail = False
     try:
         broadcast = check_onnx_broadcast(dims1, dims2)
     except ValueError:
         fail = True
     self.assertEqual(broadcast, expect_broadcast)
     self.assertEqual(fail, expect_fail)

Exemple #2

Fichier : test_utils.py Projet : xiongyw/pytorch

 def try_check_onnx_broadcast(dims1, dims2, expect_broadcast, expect_fail):
     broadcast = True
     fail = False
     try:
         broadcast = check_onnx_broadcast(dims1, dims2)
     except ValueError:
         fail = True
     self.assertEqual(broadcast, expect_broadcast)
     self.assertEqual(fail, expect_fail)