Searched refs:embedding_matrix (Results 1 – 3 of 3) sorted by relevance
| /external/libtextclassifier/native/lang_id/common/ |
| D | embedding-network.cc | 157 const EmbeddingNetworkParams::Matrix &embedding_matrix = in ConcatEmbeddings() local
159 const int embedding_dim = embedding_matrix.cols; in ConcatEmbeddings()
190 SAFTM_CHECK_LT(feature_id, embedding_matrix.rows); in ConcatEmbeddings()
194 (reinterpret_cast<const char *>(embedding_matrix.elements) + in ConcatEmbeddings()
197 switch (embedding_matrix.quant_type) { in ConcatEmbeddings()
207 multiplier *= Float16To32(embedding_matrix.quant_scales[feature_id]); in ConcatEmbeddings()
219 multiplier *= Float16To32(embedding_matrix.quant_scales[feature_id]); in ConcatEmbeddings()
237 << static_cast<int>(embedding_matrix.quant_type); in ConcatEmbeddings()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | eager_test.py | 277 embedding_matrix = resource_variable_ops.ResourceVariable(
282 embedding = embedding_ops.embedding_lookup(embedding_matrix, [1])
284 dy_dx = tape.gradient(y, embedding_matrix)
289 optimizer.apply_gradients([(dy_dx, embedding_matrix)])
294 embedding_matrix.assign_add(array_ops.ones([3, 2]))
298 [2.0, 2.0]], embedding_matrix.numpy())
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops_test.py | 209 embedding_matrix = variable_scope.get_variable(
217 embedding = embedding_ops.embedding_lookup(embedding_matrix + 0.0, [0])
232 embedding_matrix = variable_scope.get_variable(
239 embedding = embedding_ops.embedding_lookup(embedding_matrix, [0])
251 embedding_matrix = variable_scope.get_variable(
260 embedding = embedding_ops.embedding_lookup(embedding_matrix, [0])
270 dynamic_grads = gradients_impl.gradients(cost, [embedding_matrix])[0]
274 embedding = embedding_ops.embedding_lookup(embedding_matrix, [0])
278 static_grads = gradients_impl.gradients(static, [embedding_matrix])[0]
|