| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column.py | 57 use_safe_embedding_lookup=True): argument
148 use_safe_embedding_lookup=use_safe_embedding_lookup)
164 use_safe_embedding_lookup=True): argument
285 use_safe_embedding_lookup=use_safe_embedding_lookup)
372 use_safe_embedding_lookup=True, argument
389 use_safe_embedding_lookup=use_safe_embedding_lookup)
402 use_safe_embedding_lookup=True, argument
519 use_safe_embedding_lookup=True): argument
531 use_safe_embedding_lookup=use_safe_embedding_lookup)
545 use_safe_embedding_lookup=True): argument
|
| D | feature_column_v2.py | 59 use_safe_embedding_lookup=True): argument
192 use_safe_embedding_lookup=use_safe_embedding_lookup)
203 use_safe_embedding_lookup=use_safe_embedding_lookup)
216 use_safe_embedding_lookup=True): argument
399 use_safe_embedding_lookup=use_safe_embedding_lookup)
411 use_safe_embedding_lookup=use_safe_embedding_lookup)
427 use_safe_embedding_lookup=True, argument
441 use_safe_embedding_lookup=use_safe_embedding_lookup)
446 self.use_safe_embedding_lookup, self._bypass_scope_validation)
459 use_safe_embedding_lookup=True, argument
[all …]
|
| D | feature_column_v2_test.py | 124 def test_feature_layer_cpu(self, use_safe_embedding_lookup): argument
182 use_safe_embedding_lookup=use_safe_embedding_lookup)
188 use_safe_embedding_lookup=use_safe_embedding_lookup)
210 if use_safe_embedding_lookup:
242 use_safe_embedding_lookup=True,
269 use_safe_embedding_lookup=True,
291 use_safe_embedding_lookup=False,
298 self.assertFalse(embedding_column_copy.use_safe_embedding_lookup)
375 def test_feature_layer_cpu(self, use_safe_embedding_lookup): argument
433 use_safe_embedding_lookup=use_safe_embedding_lookup)
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.tpu.experimental.pbtxt | 41 …ate_fn\', \'embedding_lookup_device\', \'tensor_core_shape\', \'use_safe_embedding_lookup\'], vara…
49 …ate_fn\', \'embedding_lookup_device\', \'tensor_core_shape\', \'use_safe_embedding_lookup\'], vara…
|
| D | tensorflow.feature_column.pbtxt | 29 …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/ |
| D | tensorflow.feature_column.pbtxt | 29 …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/ |
| D | feature_column_v2.py | 523 use_safe_embedding_lookup=True): argument
625 use_safe_embedding_lookup=use_safe_embedding_lookup)
638 use_safe_embedding_lookup=True): argument
807 use_safe_embedding_lookup=use_safe_embedding_lookup))
822 use_safe_embedding_lookup=True): argument
976 use_safe_embedding_lookup)
2905 use_safe_embedding_lookup=True): argument
2916 use_safe_embedding_lookup=use_safe_embedding_lookup)
2992 if (not self.use_safe_embedding_lookup and sparse_id_rank is not None and
3173 use_safe_embedding_lookup=True): argument
[all …]
|
| D | feature_column.py | 826 use_safe_embedding_lookup=True): argument
939 use_safe_embedding_lookup=use_safe_embedding_lookup)
2490 use_safe_embedding_lookup=True): argument
2501 use_safe_embedding_lookup=use_safe_embedding_lookup)
2548 if (not self.use_safe_embedding_lookup and sparse_id_rank is not None and
2698 if (not self.use_safe_embedding_lookup and sparse_id_rank is not None and
|
| D | feature_column_v2_test.py | 5734 def test_get_dense_tensor(self, use_safe_embedding_lookup, argument
5779 if use_safe_embedding_lookup:
5807 use_safe_embedding_lookup=use_safe_embedding_lookup)
5832 if use_safe_embedding_lookup:
5849 def test_get_dense_tensor_valid(self, use_safe_embedding_lookup): argument
5902 use_safe_embedding_lookup=use_safe_embedding_lookup)
5922 if use_safe_embedding_lookup:
5939 def test_get_dense_tensor_placeholder_inputs(self, use_safe_embedding_lookup): argument
5990 use_safe_embedding_lookup=use_safe_embedding_lookup)
5997 if use_safe_embedding_lookup:
|
| D | feature_column_test.py | 4903 def test_get_dense_tensor(self, use_safe_embedding_lookup, argument
4961 use_safe_embedding_lookup=use_safe_embedding_lookup)
4982 if use_safe_embedding_lookup:
5760 def test_get_dense_tensor(self, use_safe_embedding_lookup, argument
5804 if use_safe_embedding_lookup:
5833 use_safe_embedding_lookup=use_safe_embedding_lookup))
5857 if use_safe_embedding_lookup:
|