Searched refs:get_layer_policy (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | get_layer_policy_test.py | 23 from tensorflow.python.keras.mixed_precision import get_layer_policy
32 self.assertEqual(get_layer_policy.get_layer_policy(layer).name, 'float32')
36 self.assertIs(get_layer_policy.get_layer_policy(layer), p)
39 self.assertEqual(get_layer_policy.get_layer_policy(layer).name, 'float64')
44 get_layer_policy.get_layer_policy(1)
|
| D | BUILD | 45 ":get_layer_policy",
106 name = "get_layer_policy",
107 srcs = ["get_layer_policy.py"],
119 ":get_layer_policy",
|
| D | get_layer_policy.py | 29 def get_layer_policy(layer): function
|
| D | keras_test.py | 48 from tensorflow.python.keras.mixed_precision import get_layer_policy
140 self.assertEqual(get_layer_policy.get_layer_policy(layer).name,
150 self.assertEqual(get_layer_policy.get_layer_policy(layer).name,
641 self.assertEqual(get_layer_policy.get_layer_policy(layer).name,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.mixed_precision.experimental.pbtxt | 12 name: "get_layer_policy"
|
| /external/tensorflow/tensorflow/python/keras/api/ |
| D | BUILD | 76 "tensorflow.python.keras.mixed_precision.get_layer_policy",
|