Esempio n. 1
0
 def setUp(self):
   super(FunMCMCTestTensorFlow, self).setUp()
   backend.set_backend(backend.TENSORFLOW, backend.MANUAL_TRANSFORMS)
Esempio n. 2
0
 def setUp(self):
   super(FunMCMCTestJAX, self).setUp()
   backend.set_backend(backend.JAX, backend.MANUAL_TRANSFORMS)
Esempio n. 3
0
 def setUp(self):
   super(FunMCMCTestTensorFlow, self).setUp()
   backend.set_backend(backend.TENSORFLOW)
Esempio n. 4
0
 def setUp(self):
   super(FunMCMCTestJAX, self).setUp()
   backend.set_backend(backend.JAX)
Esempio n. 5
0
 def setUp(self):
     super(UtilTFPTestTensorFlow32, self).setUp()
     backend.set_backend(backend.TENSORFLOW, backend.MANUAL_TRANSFORMS)
Esempio n. 6
0
 def setUp(self):
     super(UtilTFPTestJAX32, self).setUp()
     backend.set_backend(backend.JAX, backend.MANUAL_TRANSFORMS)