Home
last modified time | relevance | path

Searched refs:global_policy (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dpolicy_test.py151 self.assertEqual(mp_policy.global_policy().name, default_policy)
154 self.assertEqual(mp_policy.global_policy().name, 'mixed_float16')
156 self.assertEqual(mp_policy.global_policy().name, 'mixed_float16')
158 self.assertEqual(mp_policy.global_policy().name, '_infer')
161 self.assertIs(mp_policy.global_policy(), policy)
225 self.assertEqual(mp_policy.global_policy().name, 'mixed_float16')
227 self.assertEqual(mp_policy.global_policy().name, '_infer')
228 self.assertEqual(mp_policy.global_policy().name, 'mixed_float16')
229 self.assertEqual(mp_policy.global_policy().name, default_policy)
Dpolicy.py421 def global_policy(): function
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.mixed_precision.pbtxt16 name: "global_policy"
Dtensorflow.keras.mixed_precision.experimental.pbtxt16 name: "global_policy"
/external/tensorflow/tensorflow/python/keras/distribute/
Dkeras_correctness_test_base.py303 if policy.global_policy().compute_dtype in ('float16', 'bfloat16'):
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc397 TFE_ContextDevicePlacementPolicy global_policy, in TensorHandleSilentCopy() argument
404 TFE_ContextOptionsSetDevicePlacementPolicy(opts, global_policy); in TensorHandleSilentCopy()
406 if (thread_policy != global_policy) { in TensorHandleSilentCopy()
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_v1.py1756 self._dtype_policy = policy.global_policy()
Dbase_layer.py2354 self._dtype_policy = policy.global_policy()