| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| embedding_lookup_sparse.cc | 118 float current_squares_weight, int embedding_size,
132 for (int k = 0; k < embedding_size; k++) {
163 int embedding_size = 1; local
172 embedding_size *= dim;
176 const int output_size = lookup_size * embedding_size;
203 const int output_offset = output_bucket * embedding_size;
209 current_squares_weight, embedding_size,
221 const int example_embedding_offset = idx * embedding_size;
225 for (int k = 0; k < embedding_size; k++) {
233 current_squares_weight, embedding_size,
[all...] |
| /external/tensorflow/tensorflow/examples/tutorials/word2vec/ |
| word2vec_basic.py | 164 embedding_size = 128 # Dimension of the embedding vector. variable
192 tf.random_uniform([vocabulary_size, embedding_size], -1.0, 1.0))
199 [vocabulary_size, embedding_size],
200 stddev=1.0 / math.sqrt(embedding_size)))
|
| /external/libtextclassifier/ |
| feature-processor.cc | 972 const int embedding_size = GetOptions()->embedding_size(); local
[all...] |
| model_generated.h | 2430 int32_t embedding_size; member in struct:libtextclassifier2::FeatureProcessorOptionsT
2524 int32_t embedding_size() const { function in struct:libtextclassifier2::FLATBUFFERS_FINAL_CLASS
[all...] |
