Exemple #1
0
 def setUp(self):
   super(FunMCMCTestTensorFlow, self).setUp()
   backend.set_backend(backend.TENSORFLOW, backend.MANUAL_TRANSFORMS)
Exemple #2
0
 def setUp(self):
   super(FunMCMCTestJAX, self).setUp()
   backend.set_backend(backend.JAX, backend.MANUAL_TRANSFORMS)
 def setUp(self):
   super(FunMCMCTestTensorFlow, self).setUp()
   backend.set_backend(backend.TENSORFLOW)
 def setUp(self):
   super(FunMCMCTestJAX, self).setUp()
   backend.set_backend(backend.JAX)
Exemple #5
0
 def setUp(self):
     super(UtilTFPTestTensorFlow32, self).setUp()
     backend.set_backend(backend.TENSORFLOW, backend.MANUAL_TRANSFORMS)
Exemple #6
0
 def setUp(self):
     super(UtilTFPTestJAX32, self).setUp()
     backend.set_backend(backend.JAX, backend.MANUAL_TRANSFORMS)