Searched defs:use_safe_embedding_lookup (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | feature_column.py | 57 use_safe_embedding_lookup=True): argument
164 use_safe_embedding_lookup=True): argument
372 use_safe_embedding_lookup=True, argument
402 use_safe_embedding_lookup=True, argument
519 use_safe_embedding_lookup=True): argument
545 use_safe_embedding_lookup=True): argument
|
| H A D | feature_column_v2.py | 59 use_safe_embedding_lookup=True): argument
216 use_safe_embedding_lookup=True): argument
427 use_safe_embedding_lookup=True, argument
459 use_safe_embedding_lookup=True, argument
628 use_safe_embedding_lookup=True): argument
656 use_safe_embedding_lookup=True): argument
|
| H A D | feature_column_v2_test.py | 124 def test_feature_layer_cpu(self, use_safe_embedding_lookup): argument
375 def test_feature_layer_cpu(self, use_safe_embedding_lookup): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/ |
| H A D | feature_column_v2.py | 523 use_safe_embedding_lookup=True): argument
638 use_safe_embedding_lookup=True): argument
822 use_safe_embedding_lookup=True): argument
2905 use_safe_embedding_lookup=True): argument
3173 use_safe_embedding_lookup=True): argument
3231 use_safe_embedding_lookup=True): argument
|
| H A D | feature_column.py | 826 use_safe_embedding_lookup=True): argument
2490 use_safe_embedding_lookup=True): argument
|
| H A D | feature_column_v2_test.py | 5734 def test_get_dense_tensor(self, use_safe_embedding_lookup, argument
5849 def test_get_dense_tensor_valid(self, use_safe_embedding_lookup): argument
5939 def test_get_dense_tensor_placeholder_inputs(self, use_safe_embedding_lookup): argument
|
| H A D | feature_column_test.py | 4903 def test_get_dense_tensor(self, use_safe_embedding_lookup, argument
5760 def test_get_dense_tensor(self, use_safe_embedding_lookup, argument
|