Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column.py61 use_safe_embedding_lookup=True): argument
152 use_safe_embedding_lookup=use_safe_embedding_lookup)
168 use_safe_embedding_lookup=True): argument
288 use_safe_embedding_lookup=use_safe_embedding_lookup)
375 use_safe_embedding_lookup=True, argument
391 use_safe_embedding_lookup=use_safe_embedding_lookup)
404 use_safe_embedding_lookup=True, argument
521 use_safe_embedding_lookup=True): argument
533 use_safe_embedding_lookup=use_safe_embedding_lookup)
547 use_safe_embedding_lookup=True): argument
Dfeature_column_v2.py63 use_safe_embedding_lookup=True): argument
196 use_safe_embedding_lookup=use_safe_embedding_lookup)
207 use_safe_embedding_lookup=use_safe_embedding_lookup)
220 use_safe_embedding_lookup=True): argument
402 use_safe_embedding_lookup=use_safe_embedding_lookup)
414 use_safe_embedding_lookup=use_safe_embedding_lookup)
430 use_safe_embedding_lookup=True, argument
443 use_safe_embedding_lookup=use_safe_embedding_lookup)
460 use_safe_embedding_lookup=True, argument
629 use_safe_embedding_lookup=True): argument
[all …]
Dfeature_column_v2_test.py126 def test_feature_layer_cpu(self, use_safe_embedding_lookup): argument
184 use_safe_embedding_lookup=use_safe_embedding_lookup)
190 use_safe_embedding_lookup=use_safe_embedding_lookup)
212 if use_safe_embedding_lookup:
244 use_safe_embedding_lookup=True,
271 use_safe_embedding_lookup=True,
357 def test_feature_layer_cpu(self, use_safe_embedding_lookup): argument
415 use_safe_embedding_lookup=use_safe_embedding_lookup)
436 if use_safe_embedding_lookup:
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.tpu.experimental.pbtxt37 …ate_fn\', \'embedding_lookup_device\', \'tensor_core_shape\', \'use_safe_embedding_lookup\'], vara…
45 …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.py499 use_safe_embedding_lookup=True): argument
601 use_safe_embedding_lookup=use_safe_embedding_lookup)
614 use_safe_embedding_lookup=True): argument
783 use_safe_embedding_lookup=use_safe_embedding_lookup))
798 use_safe_embedding_lookup=True): argument
952 use_safe_embedding_lookup)
2838 use_safe_embedding_lookup=True): argument
2849 use_safe_embedding_lookup=use_safe_embedding_lookup)
2925 if (not self.use_safe_embedding_lookup and sparse_id_rank is not None and
3105 use_safe_embedding_lookup=True): argument
[all …]
Dfeature_column.py830 use_safe_embedding_lookup=True): argument
943 use_safe_embedding_lookup=use_safe_embedding_lookup)
2494 use_safe_embedding_lookup=True): argument
2505 use_safe_embedding_lookup=use_safe_embedding_lookup)
2552 if (not self.use_safe_embedding_lookup and sparse_id_rank is not None and
2702 if (not self.use_safe_embedding_lookup and sparse_id_rank is not None and
Dfeature_column_v2_test.py5705 def test_get_dense_tensor(self, use_safe_embedding_lookup, argument
5750 if use_safe_embedding_lookup:
5778 use_safe_embedding_lookup=use_safe_embedding_lookup)
5803 if use_safe_embedding_lookup:
5820 def test_get_dense_tensor_valid(self, use_safe_embedding_lookup): argument
5873 use_safe_embedding_lookup=use_safe_embedding_lookup)
5893 if use_safe_embedding_lookup:
5910 def test_get_dense_tensor_placeholder_inputs(self, use_safe_embedding_lookup): argument
5961 use_safe_embedding_lookup=use_safe_embedding_lookup)
5968 if use_safe_embedding_lookup:
Dfeature_column_test.py4924 def test_get_dense_tensor(self, use_safe_embedding_lookup, argument
4982 use_safe_embedding_lookup=use_safe_embedding_lookup)
5003 if use_safe_embedding_lookup:
5779 def test_get_dense_tensor(self, use_safe_embedding_lookup, argument
5823 if use_safe_embedding_lookup:
5852 use_safe_embedding_lookup=use_safe_embedding_lookup))
5876 if use_safe_embedding_lookup:
/external/tensorflow/tensorflow/python/keras/feature_column/
Ddense_features_test.py724 def test_dense_features(self, use_safe_embedding_lookup, partition_variables): argument
780 use_safe_embedding_lookup=use_safe_embedding_lookup)
816 if use_safe_embedding_lookup: