Home
last modified time | relevance | path

Searched refs:embedding_weights (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dembedding_ops_test.py805 embedding_weights = list(variable_scope.get_variable(
810 for w in embedding_weights:
812 embedding_weights = [self.evaluate(w) for w in embedding_weights]
813 return embedding_weights
868 embedding_weights = self._random_weights()
872 embedding_ops.safe_embedding_lookup_sparse_v2(embedding_weights,
878 [(1.0 * embedding_weights[0][0] + 2.0 * embedding_weights[0][1]) /
879 3.0, [0] * 4, [0] * 4, embedding_weights[0][2], [0] * 4])
884 embedding_weights = self._random_weights()
889 embedding_weights, sparse_ids, sparse_weights, default_id=3))
[all …]
/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py673 def safe_embedding_lookup_sparse_v2(embedding_weights, argument
763 embedding_weights,
775 def safe_embedding_lookup_sparse(embedding_weights, argument
860 if embedding_weights is None:
861 raise ValueError("Missing embedding_weights %s." % embedding_weights)
862 if isinstance(embedding_weights, variables.PartitionedVariable):
863 embedding_weights = list(embedding_weights) # get underlying Variables.
864 if not isinstance(embedding_weights, list):
865 embedding_weights = [embedding_weights]
866 if len(embedding_weights) < 1:
[all …]
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_v2.py973 embedding_weights = state_manager.get_variable(
975 return sparse_embedding_aggregate_slice(embedding_weights, (values, mask),
1013 embedding_weights = variable_scope.get_variable(
1020 return sparse_embedding_aggregate_slice(embedding_weights, (values, mask),
1090 embedding_weights = self.shared_embedding_column_creator.embedding_weights
1091 return sparse_embedding_aggregate_slice(embedding_weights, (values, mask),
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py2537 embedding_weights = self.layer_creator(
2542 to_restore = embedding_weights
2557 embedding_weights,
2670 embedding_weights = shared_embedding_collection[0]
2671 if embedding_weights.get_shape() != embedding_shape:
2679 embedding_weights.name,
2680 embedding_weights.get_shape(), embedding_shape))
2682 embedding_weights = variable_scope.get_variable(
2690 embedding_weights)
2692 to_restore = embedding_weights
[all …]
Dfeature_column_v2.py2910 embedding_weights): argument
2915 to_restore = embedding_weights
2930 embedding_weights,
2939 embedding_weights = state_manager.get_variable(
2942 embedding_weights)
2951 embedding_weights = variable_scope.get_variable(
2959 embedding_weights)
3122 def embedding_weights(self): member in SharedEmbeddingColumnCreator
3219 embedding_weights = self.shared_embedding_column_creator.embedding_weights
3229 embedding_weights,
Dfeature_column_test.py5279 embedding_weights = trainable_vars[
5287 self.evaluate(embedding_weights))
5294 embedding_weights.assign((
5359 embedding_weights = trainable_vars[
5366 self.evaluate(embedding_weights))
5373 embedding_weights.assign((
6052 embedding_weights = trainable_vars[
6062 self.evaluate(embedding_weights))
6071 embedding_weights.assign((
6145 embedding_weights = trainable_vars[
[all …]
Dfeature_column_v2_test.py5238 embedding_weights = trainable_vars[
5248 self.evaluate(embedding_weights))
5255 embedding_weights.assign((
5320 embedding_weights = trainable_vars[
5330 self.evaluate(embedding_weights))
5337 embedding_weights.assign((
/external/tensorflow/tensorflow/python/distribute/
Dcustom_training_loop_input_test.py756 embedding_weights = array_ops.zeros((1, 128))
758 embeddings = array_ops.gather(embedding_weights, flat_inputs)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt261 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt329 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…