Searched refs:global_policy (Results 1 – 6 of 6) sorted by relevance
42 self.assertEqual(mp_policy.global_policy().name, 'infer')43 default_policy = mp_policy.global_policy()46 self.assertEqual(mp_policy.global_policy().name, 'infer_float32_vars')47 self.assertEqual(mp_policy.global_policy().default_variable_dtype,50 self.assertEqual(mp_policy.global_policy().name, 'infer_float32_vars')52 self.assertEqual(mp_policy.global_policy().name, 'infer')53 self.assertEqual(mp_policy.global_policy().default_variable_dtype, None)56 self.assertIs(mp_policy.global_policy(), policy)62 self.assertEqual(mp_policy.global_policy().name, 'infer_float32_vars')64 self.assertEqual(mp_policy.global_policy().name, 'infer')[all …]
21 from tensorflow.python.keras.mixed_precision.experimental.policy import global_policy
133 def global_policy(): function
12 name: "global_policy"
1357 self._mixed_precision_policy = policy.global_policy()