| /external/tensorflow/tensorflow/contrib/learn/python/learn/ops/ |
| embeddings_ops.py | 38 @deprecated('2016-12-01', 'Use `tf.embedding_lookup` instead.')
39 def embedding_lookup(params, ids, name='embedding_lookup'): function
40 """Provides a N dimensional version of tf.embedding_lookup.
42 Ids are flattened to a 1d tensor before being passed to embedding_lookup
59 with ops.name_scope(name, 'embedding_lookup', [params, ids]):
65 embeds_flat = nn.embedding_lookup(params, ids_flat, name)
92 return embedding_lookup(embeddings, tensor_in)
|
| ops_test.py | 57 embed_tf = ops.embedding_lookup(embeds, ids).eval()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| embedding_lookup.cc | 48 namespace embedding_lookup { namespace in namespace:tflite::ops::builtin
151 } // namespace embedding_lookup
154 static TfLiteRegistration r = {nullptr, nullptr, embedding_lookup::Prepare,
155 embedding_lookup::Eval};
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| encoders.py | 87 embedding_ops.embedding_lookup(embeddings, ids), axis=1)
141 return embedding_ops.embedding_lookup(embeddings, ids)
|
| /external/tensorflow/tensorflow/contrib/gan/python/features/python/ |
| conditioning_utils_impl.py | 85 return embedding_ops.embedding_lookup(
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| embedding_ops_test.py | 260 embedding = embedding_ops.embedding_lookup(p, ids)
273 embedding = embedding_ops.embedding_lookup(
284 embedding = embedding_ops.embedding_lookup(
303 embedding = embedding_ops.embedding_lookup(p_variable, ids)
326 embedding = embedding_ops.embedding_lookup(p_variable, ids)
356 embedding = embedding_ops.embedding_lookup(p, ids)
379 embedding = embedding_ops.embedding_lookup(p, ids)
402 embedding = embedding_ops.embedding_lookup(
428 embedding = embedding_ops.embedding_lookup(
453 embedding = embedding_ops.embedding_lookup(
[all...] |
| /external/tensorflow/tensorflow/examples/saved_model/integration_tests/ |
| export_text_rnn_model.py | 108 sentence_embeddings = tf.nn.embedding_lookup(self._embeddings,
153 token_embeddings = tf.nn.embedding_lookup(self._embeddings, current_id)
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| adagrad_test.py | 77 pred = math_ops.matmul(embedding_ops.embedding_lookup([var0], [0]), x)
181 embedding_ops.embedding_lookup(var_repeated, [0, 0]))
185 embedding_ops.embedding_lookup(var_aggregated, [0]))
|
| adadelta_test.py | 152 pred = math_ops.matmul(embedding_ops.embedding_lookup([var0], [0]), x)
|
| gradient_descent_test.py | 115 pred = math_ops.matmul(embedding_ops.embedding_lookup([var0], [0]), x)
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| wals_test.py | 54 current_rows = embedding_ops.embedding_lookup(
57 current_cols = embedding_ops.embedding_lookup(
60 row_wts = embedding_ops.embedding_lookup(
63 col_wts = embedding_ops.embedding_lookup(
|
| factorization_ops_test.py | 54 current_rows = embedding_ops.embedding_lookup(
58 current_cols = embedding_ops.embedding_lookup(
62 row_wts = embedding_ops.embedding_lookup(
66 col_wts = embedding_ops.embedding_lookup(
[all...] |
| gmm_ops.py | 37 from tensorflow.python.ops.embedding_ops import embedding_lookup
88 clusters_init = embedding_lookup(data, indices, partition_strategy='div')
|
| /external/tensorflow/tensorflow/contrib/nn/python/ops/ |
| sampling_ops.py | 93 # Calling this protected form of embedding_lookup allows co-locating
99 embedding_ops.embedding_lookup(biases, sampled, partition_strategy), [-1])
200 See `tf.nn.embedding_lookup` for more details.
314 Default is `"mod"`. See `tf.nn.embedding_lookup` for more details.
|
| /external/tensorflow/tensorflow/python/tpu/ |
| feature_column_test.py | 129 embedding_lookup = embedding_column._get_dense_tensor(
140 self.assertAllEqual(expected_lookups, embedding_lookup.eval())
|
| /external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/ops/ |
| seq2seq.py | 103 # embedding_lookup.
104 emb_prev = embedding_ops.embedding_lookup(embedding, prev_symbol)
256 next = embedding_lookup(embedding, argmax(previous_output)),
295 emb_inp = (embedding_ops.embedding_lookup(embedding, i)
479 embedding_ops.embedding_lookup(embedding, x) for x in encoder_inputs
482 embedding_ops.embedding_lookup(embedding, x) for x in decoder_inputs
[all...] |
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| reg_adagrad_optimizer_test.py | 80 pred = math_ops.matmul(embedding_ops.embedding_lookup([var0], [0]), x)
177 embedding_ops.embedding_lookup(var_repeated, [0, 0]))
181 embedding_ops.embedding_lookup(var_aggregated, [0]))
|
| /external/tensorflow/tensorflow/python/ops/ |
| embedding_ops.py | 90 """Helper function for embedding_lookup and _compute_sampled_logits.
92 This function is a generalization of embedding_lookup that optionally
103 params: See embedding_lookup.
104 ids: See embedding_lookup.
105 partition_strategy: See embedding_lookup.
106 name: See embedding_lookup.
107 max_norm: See embedding_lookup.
113 See embedding_lookup for details.
124 with ops.name_scope(name, "embedding_lookup", params + [ids]) as name:
250 @tf_export(v1=["nn.embedding_lookup"])
251 def embedding_lookup( function
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| adagrad_test.py | 106 pred = math_ops.matmul(embedding_ops.embedding_lookup([var0], [0]), x)
217 embedding_ops.embedding_lookup(var_repeated, [0, 0]))
221 embedding_ops.embedding_lookup(var_aggregated, [0]))
|
| adadelta_test.py | 175 pred = math_ops.matmul(embedding_ops.embedding_lookup([var0], [0]), x)
|
| adagrad_da_test.py | 93 pred = math_ops.matmul(embedding_ops.embedding_lookup([var0], [0]), x)
|
| /external/tensorflow/tensorflow/examples/tutorials/word2vec/ |
| word2vec_basic.py | 176 embed = tf.nn.embedding_lookup(embeddings, train_inputs)
212 valid_embeddings = tf.nn.embedding_lookup(normalized_embeddings,
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| embeddings.py | 178 out = embedding_ops.embedding_lookup(self.embeddings, inputs)
|
| /external/tensorflow/tensorflow/python/eager/ |
| backprop_test.py | 58 g1 = embedding_ops.embedding_lookup(x, ind1)
59 g2 = embedding_ops.embedding_lookup(x, ind2)
60 g3 = embedding_ops.embedding_lookup(x, ind3)
74 tf_g1 = embedding_ops.embedding_lookup(tf_var, tf_ind1)
75 tf_g2 = embedding_ops.embedding_lookup(tf_var, tf_ind2)
76 tf_g3 = embedding_ops.embedding_lookup(tf_var, tf_ind3)
211 embedded_x = embedding_ops.embedding_lookup(embedding, x)
222 tf_embedded_x = embedding_ops.embedding_lookup(tf_embedding, tf_x)
[all...] |
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| adadelta_test.py | 158 pred = math_ops.matmul(embedding_ops.embedding_lookup([var0], [0]), x) # pylint: disable=cell-var-from-loop
|
