| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops_test.py | 55 embedding_weights = list(variable_scope.get_variable(
60 for w in embedding_weights:
62 embedding_weights = [w.eval() for w in embedding_weights]
63 return embedding_weights
117 embedding_weights = self._random_weights()
121 embedding_weights, sparse_ids, sparse_weights).eval())
125 [(1.0 * embedding_weights[0][0] + 2.0 * embedding_weights[0][1]) /
126 3.0, [0] * 4, [0] * 4, embedding_weights[0][2], [0] * 4])
130 embedding_weights = self._random_weights()
134 embedding_weights, sparse_ids, sparse_weights, default_id=3).eval())
[all …]
|
| D | embedding_ops.py | 48 def safe_embedding_lookup_sparse(embedding_weights, argument
103 if embedding_weights is None:
104 raise ValueError("Missing embedding_weights %s." % embedding_weights)
105 if isinstance(embedding_weights, variables.PartitionedVariable):
106 embedding_weights = list(embedding_weights) # get underlying Variables.
107 if not isinstance(embedding_weights, list):
108 embedding_weights = [embedding_weights]
109 if len(embedding_weights) < 1:
110 raise ValueError("Missing embedding_weights %s." % embedding_weights)
113 if isinstance(embedding_weights, variables.PartitionedVariable):
[all …]
|
| D | feature_column_ops_test.py | 1518 embedding_weights = ops.get_collection("my_collection")
1520 embedding_weights)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 800 embedding_weights = list(variable_scope.get_variable(
805 for w in embedding_weights:
807 embedding_weights = [w.eval() for w in embedding_weights]
808 return embedding_weights
863 embedding_weights = self._random_weights()
868 embedding_weights, sparse_ids, sparse_weights).eval())
872 [(1.0 * embedding_weights[0][0] + 2.0 * embedding_weights[0][1]) /
873 3.0, [0] * 4, [0] * 4, embedding_weights[0][2], [0] * 4])
878 embedding_weights = self._random_weights()
883 embedding_weights, sparse_ids, sparse_weights,
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 627 def safe_embedding_lookup_sparse_v2(embedding_weights, argument
679 embedding_weights,
690 def safe_embedding_lookup_sparse(embedding_weights, argument
741 if embedding_weights is None:
742 raise ValueError('Missing embedding_weights %s.' % embedding_weights)
743 if isinstance(embedding_weights, variables.PartitionedVariable):
744 embedding_weights = list(embedding_weights) # get underlying Variables.
745 if not isinstance(embedding_weights, list):
746 embedding_weights = [embedding_weights]
747 if len(embedding_weights) < 1:
[all …]
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column.py | 2431 embedding_weights = self.layer_creator(
2436 to_restore = embedding_weights
2445 embedding_weights=embedding_weights,
2557 embedding_weights = shared_embedding_collection[0]
2558 if embedding_weights.get_shape() != embedding_shape:
2566 embedding_weights.name,
2567 embedding_weights.get_shape(), embedding_shape))
2569 embedding_weights = variable_scope.get_variable(
2577 embedding_weights)
2579 to_restore = embedding_weights
[all …]
|
| D | feature_column_v2.py | 3050 embedding_weights): argument
3055 to_restore = embedding_weights
3064 embedding_weights=embedding_weights,
3073 embedding_weights = state_manager.get_variable(
3076 embedding_weights)
3085 embedding_weights = variable_scope.get_variable(
3093 embedding_weights)
3245 def embedding_weights(self): member in SharedEmbeddingColumnCreator
3327 embedding_weights = self.shared_embedding_column_creator.embedding_weights
3331 embedding_weights=embedding_weights,
|
| D | feature_column_test.py | 5250 embedding_weights = trainable_vars[
5258 self.evaluate(embedding_weights))
5265 embedding_weights.assign((
5331 embedding_weights = trainable_vars[
5338 self.evaluate(embedding_weights))
5345 embedding_weights.assign((
5967 embedding_weights = trainable_vars[
5977 self.evaluate(embedding_weights))
5986 embedding_weights.assign((
6061 embedding_weights = trainable_vars[
[all …]
|
| D | feature_column_v2_test.py | 6771 embedding_weights = trainable_vars[
6781 self.evaluate(embedding_weights))
6788 embedding_weights.assign((
7052 embedding_weights = trainable_vars[
7062 self.evaluate(embedding_weights))
7069 embedding_weights.assign((
7134 embedding_weights = trainable_vars[
7144 self.evaluate(embedding_weights))
7151 embedding_weights.assign((
7669 embedding_weights = trainable_vars['aaa_bbb_shared_embedding:0']
[all …]
|
| /external/tensorflow/tensorflow/examples/saved_model/integration_tests/ |
| D | export_simple_text_embedding.py | 90 embedding_weights=self.embeddings,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 253 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 329 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…
|