Home
last modified time | relevance | path

Searched refs:embedding_lookup (Results 1 – 25 of 64) sorted by relevance

123

/external/tensorflow/tensorflow/contrib/learn/python/learn/ops/
Dembeddings_ops.py39 def embedding_lookup(params, ids, name='embedding_lookup'): function
65 embeds_flat = nn.embedding_lookup(params, ids_flat, name)
92 return embedding_lookup(embeddings, tensor_in)
Dops_test.py57 embed_tf = ops.embedding_lookup(embeds, ids).eval()
/external/tensorflow/tensorflow/python/kernel_tests/
Dembedding_ops_test.py260 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/lite/kernels/
Dembedding_lookup.cc48 namespace embedding_lookup { namespace
154 static TfLiteRegistration r = {nullptr, nullptr, embedding_lookup::Prepare, in Register_EMBEDDING_LOOKUP()
155 embedding_lookup::Eval}; in Register_EMBEDDING_LOOKUP()
DAndroid.bp58 "embedding_lookup.cc",
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dencoders.py87 embedding_ops.embedding_lookup(embeddings, ids), axis=1)
141 return embedding_ops.embedding_lookup(embeddings, ids)
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Dadagrad_test.py77 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]))
Dadadelta_test.py152 pred = math_ops.matmul(embedding_ops.embedding_lookup([var0], [0]), x)
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dwals_test.py54 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(
Dfactorization_ops_test.py54 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(
Dfactorization_ops.py895 right = embedding_ops.embedding_lookup(
925 row_weights_slice = embedding_ops.embedding_lookup(
936 col_weights = embedding_ops.embedding_lookup(
Dgmm_ops.py37 from tensorflow.python.ops.embedding_ops import embedding_lookup
88 clusters_init = embedding_lookup(data, indices, partition_strategy='div')
/external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/ops/
Dseq2seq.py104 emb_prev = embedding_ops.embedding_lookup(embedding, prev_symbol)
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
784 embedding_ops.embedding_lookup(embedding, i) for i in decoder_inputs
/external/tensorflow/tensorflow/examples/saved_model/integration_tests/
Dexport_text_rnn_model.py108 sentence_embeddings = tf.nn.embedding_lookup(self._embeddings,
153 token_embeddings = tf.nn.embedding_lookup(self._embeddings, current_id)
/external/tensorflow/tensorflow/python/training/
Dadagrad_test.py106 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]))
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dreg_adagrad_optimizer_test.py80 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/
Dembedding_ops.py251 def embedding_lookup( function
366 return embedding_lookup(params, ids, "div", name,
485 embeddings = embedding_lookup(
/external/tensorflow/tensorflow/contrib/gan/python/features/python/
Dconditioning_utils_impl.py85 return embedding_ops.embedding_lookup(
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_test.py129 embedding_lookup = embedding_column._get_dense_tensor(
140 self.assertAllEqual(expected_lookups, embedding_lookup.eval())
/external/tensorflow/tensorflow/lite/g3doc/performance/
Dpost_training_quantization.md26 * [tf.nn.embedding_lookup](https://www.tensorflow.org/api_docs/python/tf/nn/embedding_lookup)
/external/tensorflow/tensorflow/examples/tutorials/word2vec/
Dword2vec_basic.py176 embed = tf.nn.embedding_lookup(embeddings, train_inputs)
212 valid_embeddings = tf.nn.embedding_lookup(normalized_embeddings,
/external/tensorflow/tensorflow/python/eager/
Dbackprop_test.py58 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)
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dadagrad_test.py219 …pred = math_ops.matmul(embedding_ops.embedding_lookup([var0], [0]), x) # pylint: disable=cell-var…
351 … embedding_ops.embedding_lookup(var_repeated, [0, 0])) # pylint: disable=cell-var-from-loop
355 … embedding_ops.embedding_lookup(var_aggregated, [0])) # pylint: disable=cell-var-from-loop
/external/tensorflow/tensorflow/contrib/nn/python/ops/
Dsampling_ops.py99 embedding_ops.embedding_lookup(biases, sampled, partition_strategy), [-1])
/external/tensorflow/tensorflow/python/keras/layers/
Dembeddings.py178 out = embedding_ops.embedding_lookup(self.embeddings, inputs)

123