HomeSort by relevance Sort by last modified time
    Searched refs:embedding_matrix (Results 1 - 3 of 3) sorted by null

  /external/libtextclassifier/lang_id/common/
embedding-network.cc 157 const EmbeddingNetworkParams::Matrix &embedding_matrix = local
159 const int embedding_dim = embedding_matrix.cols;
190 SAFTM_CHECK_LT(feature_id, embedding_matrix.rows);
194 (reinterpret_cast<const char *>(embedding_matrix.elements) +
197 switch (embedding_matrix.quant_type) {
207 multiplier *= Float16To32(embedding_matrix.quant_scales[feature_id]);
219 multiplier *= Float16To32(embedding_matrix.quant_scales[feature_id]);
237 << static_cast<int>(embedding_matrix.quant_type);
  /external/tensorflow/tensorflow/compiler/tests/
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/
control_flow_ops_test.py 195 embedding_matrix = variable_scope.get_variable(
196 "embedding_matrix", [5, 5],
203 embedding = embedding_ops.embedding_lookup(embedding_matrix + 0.0, [0])
218 embedding_matrix = variable_scope.get_variable(
219 "embedding_matrix", initializer=[[2.0], [3.0]], use_resource=True)
225 embedding = embedding_ops.embedding_lookup(embedding_matrix, [0])
237 embedding_matrix = variable_scope.get_variable(
238 "embedding_matrix", [5, 5],
246 embedding = embedding_ops.embedding_lookup(embedding_matrix, [0])
256 dynamic_grads = gradients_impl.gradients(cost, [embedding_matrix])[0
    [all...]

Completed in 792 milliseconds