Home
last modified time | relevance | path

Searched refs:PolicyV1 (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dpolicy_test.py71 self.assertEqual(repr(mp_policy.PolicyV1(policy)),
73 self.assertEqual(repr(mp_policy.PolicyV1('float16', loss_scale=2)),
76 repr(mp_policy.PolicyV1('mixed_float16')),
123 policy = mp_policy.PolicyV1('float32')
126 policy = mp_policy.PolicyV1('float32', loss_scale=None)
130 policy = mp_policy.PolicyV1('float32', loss_scale=ls)
133 policy = mp_policy.PolicyV1('float32', loss_scale='dynamic')
136 policy = mp_policy.PolicyV1('mixed_float16')
139 policy = mp_policy.PolicyV1('mixed_float16', loss_scale=None)
142 policy = mp_policy.PolicyV1('mixed_bfloat16')
[all …]
Dpolicy.py321 class PolicyV1(Policy): class
367 super(PolicyV1, self).__init__(name)
Dkeras_test.py378 layer = mp_test_util.MultiplyLayer(dtype=policy.PolicyV1('mixed_float16',
398 layer = mp_test_util.MultiplyLayer(dtype=policy.PolicyV1('float64',
415 layer = mp_test_util.MultiplyLayer(dtype=policy.PolicyV1('_infer',
559 cls = policy.PolicyV1 if use_v1_policy else policy.Policy
828 p = policy.PolicyV1('mixed_float16', loss_scale=loss_scale)
906 with policy.policy_scope(policy.PolicyV1('float32', loss_scale=10.)):
918 with policy.policy_scope(policy.PolicyV1('mixed_float16',
930 with policy.policy_scope(policy.PolicyV1('float32', loss_scale=10.)):
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.mixed_precision.experimental.-policy.pbtxt3 is_instance: "<class \'tensorflow.python.keras.mixed_precision.policy.PolicyV1\'>"
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining.py595 if isinstance(self._dtype_policy, policy.PolicyV1):
Dtraining_v1.py1353 if isinstance(self._dtype_policy, policy.PolicyV1):