Home
last modified time | relevance | path

Searched refs:use_safe_embedding_lookup (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column.py61 use_safe_embedding_lookup=True): argument
149 use_safe_embedding_lookup=use_safe_embedding_lookup)
165 use_safe_embedding_lookup=True): argument
282 use_safe_embedding_lookup=use_safe_embedding_lookup)
369 use_safe_embedding_lookup=True): argument
383 use_safe_embedding_lookup=use_safe_embedding_lookup)
396 use_safe_embedding_lookup=True): argument
504 use_safe_embedding_lookup=True): argument
516 use_safe_embedding_lookup=use_safe_embedding_lookup)
530 use_safe_embedding_lookup=True): argument
Dfeature_column_v2.py60 use_safe_embedding_lookup=True): argument
187 use_safe_embedding_lookup=use_safe_embedding_lookup)
198 use_safe_embedding_lookup=use_safe_embedding_lookup)
211 use_safe_embedding_lookup=True): argument
386 use_safe_embedding_lookup=use_safe_embedding_lookup)
398 use_safe_embedding_lookup=use_safe_embedding_lookup)
414 use_safe_embedding_lookup=True): argument
425 use_safe_embedding_lookup=use_safe_embedding_lookup)
442 use_safe_embedding_lookup=True): argument
600 use_safe_embedding_lookup=True): argument
[all …]
Dfeature_column_v2_test.py89 def test_feature_layer_cpu(self, use_safe_embedding_lookup): argument
147 use_safe_embedding_lookup=use_safe_embedding_lookup)
153 use_safe_embedding_lookup=use_safe_embedding_lookup)
175 if use_safe_embedding_lookup:
270 def test_feature_layer_cpu(self, use_safe_embedding_lookup): argument
328 use_safe_embedding_lookup=use_safe_embedding_lookup)
349 if use_safe_embedding_lookup:
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.tpu.experimental.pbtxt25 …ate_fn\', \'embedding_lookup_device\', \'tensor_core_shape\', \'use_safe_embedding_lookup\'], vara…
33 …ate_fn\', \'embedding_lookup_device\', \'tensor_core_shape\', \'use_safe_embedding_lookup\'], vara…
Dtensorflow.feature_column.pbtxt29 …from\', \'tensor_name_in_ckpt\', \'max_norm\', \'trainable\', \'use_safe_embedding_lookup\'], vara…
73 …from\', \'tensor_name_in_ckpt\', \'max_norm\', \'trainable\', \'use_safe_embedding_lookup\'], vara…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.feature_column.pbtxt29 …from\', \'tensor_name_in_ckpt\', \'max_norm\', \'trainable\', \'use_safe_embedding_lookup\'], vara…
65 …from\', \'tensor_name_in_ckpt\', \'max_norm\', \'trainable\', \'use_safe_embedding_lookup\'], vara…
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py854 use_safe_embedding_lookup=True): argument
956 use_safe_embedding_lookup=use_safe_embedding_lookup)
969 use_safe_embedding_lookup=True): argument
1138 use_safe_embedding_lookup=use_safe_embedding_lookup))
1153 use_safe_embedding_lookup=True): argument
1307 use_safe_embedding_lookup)
3125 use_safe_embedding_lookup=True): argument
3136 use_safe_embedding_lookup=use_safe_embedding_lookup)
3212 if (not self.use_safe_embedding_lookup and sparse_id_rank is not None and
3388 use_safe_embedding_lookup=True): argument
[all …]
Dfeature_column.py825 use_safe_embedding_lookup=True): argument
938 use_safe_embedding_lookup=use_safe_embedding_lookup)
2469 use_safe_embedding_lookup=True): argument
2480 use_safe_embedding_lookup=use_safe_embedding_lookup)
2527 if (not self.use_safe_embedding_lookup and sparse_id_rank is not None and
2677 if (not self.use_safe_embedding_lookup and sparse_id_rank is not None and
Dfeature_column_v2_test.py6283 def test_dense_features(self, use_safe_embedding_lookup): argument
6328 use_safe_embedding_lookup=use_safe_embedding_lookup)
6350 if use_safe_embedding_lookup:
6987 def test_get_dense_tensor(self, use_safe_embedding_lookup): argument
7022 if use_safe_embedding_lookup:
7044 use_safe_embedding_lookup=use_safe_embedding_lookup)
7065 if use_safe_embedding_lookup:
7082 def test_get_dense_tensor_valid(self, use_safe_embedding_lookup): argument
7133 use_safe_embedding_lookup=use_safe_embedding_lookup)
7153 if use_safe_embedding_lookup:
[all …]