| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op_test.cc | 244 static void BM_ScatterHelper(int iters, int embedding_size, const char* op) { in BM_ScatterHelper() argument
246 const int kRows = 10000000 / embedding_size; in BM_ScatterHelper()
249 for (int i = 0; i < kRows * embedding_size; i++) { in BM_ScatterHelper()
259 for (int j = 0; j < embedding_size; j++) { in BM_ScatterHelper()
266 bm.AddInputFromArray<float>(TensorShape({kRows, embedding_size}), values); in BM_ScatterHelper()
268 bm.AddInputFromArray<float>(TensorShape({kNumUpdates, embedding_size}), in BM_ScatterHelper()
270 testing::ItemsProcessed((static_cast<int64>(kNumUpdates) * embedding_size) * in BM_ScatterHelper()
279 static void BM_ScatterUpdateInt32(int iters, int embedding_size) { in BM_ScatterUpdateInt32() argument
280 BM_ScatterHelper<int32>(iters, embedding_size, "ScatterUpdate"); in BM_ScatterUpdateInt32()
282 static void BM_ScatterUpdateInt64(int iters, int embedding_size) { in BM_ScatterUpdateInt64() argument
[all …]
|
| D | scatter_nd_op_test.cc | 254 static void BM_ScatterNdHelper(int iters, int embedding_size, const char* op) { in BM_ScatterNdHelper() argument
256 const int kRows = 10000000 / embedding_size; in BM_ScatterNdHelper()
259 for (int i = 0; i < kRows * embedding_size; i++) { in BM_ScatterNdHelper()
269 for (int j = 0; j < embedding_size; j++) { in BM_ScatterNdHelper()
276 bm.AddInputFromArray<float>(TensorShape({kRows, embedding_size}), values); in BM_ScatterNdHelper()
278 bm.AddInputFromArray<float>(TensorShape({kNumUpdates, embedding_size}), in BM_ScatterNdHelper()
280 testing::ItemsProcessed((static_cast<int64>(kNumUpdates) * embedding_size) * in BM_ScatterNdHelper()
289 static void BM_ScatterNdUpdateInt32(int iters, int embedding_size) { in BM_ScatterNdUpdateInt32() argument
290 BM_ScatterNdHelper<int32>(iters, embedding_size, "ScatterNdUpdate"); in BM_ScatterNdUpdateInt32()
292 static void BM_ScatterNdUpdateInt64(int iters, int embedding_size) { in BM_ScatterNdUpdateInt64() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse.cc | 118 float current_squares_weight, int embedding_size, in FinalizeAggregation() argument
132 for (int k = 0; k < embedding_size; k++) { in FinalizeAggregation()
163 int embedding_size = 1; in Eval() local
172 embedding_size *= dim; in Eval()
176 const int output_size = lookup_size * embedding_size; in Eval()
205 const int output_offset = output_bucket * embedding_size; in Eval()
211 current_squares_weight, embedding_size, in Eval()
223 const int example_embedding_offset = idx * embedding_size; in Eval()
227 for (int k = 0; k < embedding_size; k++) { in Eval()
235 current_squares_weight, embedding_size, in Eval()
|
| /external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/ops/ |
| D | seq2seq.py | 237 embedding_size, argument
291 [num_symbols, embedding_size])
306 embedding_size, argument
362 embedding_size=embedding_size)
375 embedding_size,
389 embedding_size,
414 embedding_size, argument
476 "embedding", [num_symbols, embedding_size], dtype=dtype)
716 embedding_size, argument
779 [num_symbols, embedding_size])
[all …]
|
| /external/libtextclassifier/actions/ |
| D | feature-processor.cc | 76 return options_->embedding_size() + in GetTokenEmbeddingSize()
86 const int embedding_size = options_->embedding_size(); in AppendFeatures() local
87 output_features->resize(output_features->size() + embedding_size); in AppendFeatures()
93 /*dest=*/output_features_end - embedding_size, in AppendFeatures()
94 /*dest_size=*/embedding_size)) { in AppendFeatures()
|
| D | feature-processor_test.cc | 67 options.embedding_size = 4; in TEST_F()
86 options.embedding_size = 4; in TEST_F()
107 options.embedding_size = 4; in TEST_F()
|
| D | actions_model.fbs | 84 // A (max tokens, embedding_size) float tensor specifying the embeddings of
120 embedding_size:int = -1;
|
| /external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/kernel_tests/ |
| D | seq2seq_test.py | 115 dec_inp, enc_state, cell_fn(), num_symbols=4, embedding_size=2)
146 embedding_size=2)
165 embedding_size=2)
184 embedding_size=2,
203 embedding_size=2,
212 embedding_size=2,
221 embedding_size=2,
244 enc_inp, dec_inp, cell(), num_symbols=5, embedding_size=2)
263 embedding_size=2)
278 embedding_size=2,
[all …]
|
| /external/tensorflow/tensorflow/contrib/gan/python/features/python/ |
| D | conditioning_utils_impl.py | 79 def _one_hot_to_embedding(one_hot, embedding_size): argument
84 'embedding', [num_tokens, embedding_size])
94 def condition_tensor_from_onehot(tensor, one_hot_labels, embedding_size=256): argument
114 conditioning = _one_hot_to_embedding(one_hot_labels, embedding_size)
|
| /external/libtextclassifier/annotator/ |
| D | model-executor.cc | 46 const flatbuffers::Vector<uint8_t>* model_spec_buffer, int embedding_size, in FromBuffer() argument
78 embedding_size)) { in FromBuffer()
85 embedding_size, scales, embeddings, std::move(interpreter), in FromBuffer()
|
| D | feature-processor.cc | 837 const int embedding_size = GetOptions()->embedding_size(); in AppendTokenFeaturesWithCache() local
838 output_features->resize(output_features->size() + embedding_size); in AppendTokenFeaturesWithCache()
844 /*dest=*/output_features_end - embedding_size, in AppendTokenFeaturesWithCache()
845 /*dest_size=*/embedding_size)) { in AppendTokenFeaturesWithCache()
854 output_features_end - embedding_size, output_features_end); in AppendTokenFeaturesWithCache()
|
| D | model-executor.h | 80 const flatbuffers::Vector<uint8_t>* model_spec_buffer, int embedding_size,
|
| D | feature-processor.h | 168 int EmbeddingSize() const { return options_->embedding_size(); } in EmbeddingSize()
|
| D | feature-processor_test.cc | 513 options.embedding_size = 4; in TEST_F()
621 options.embedding_size = 4; in TEST_F()
657 options.embedding_size = 4; in TEST_F()
|
| D | annotator.cc | 343 (model_->selection_feature_options()->embedding_size() != in ValidateAndInitialize()
344 model_->classification_feature_options()->embedding_size() || in ValidateAndInitialize()
354 model_->classification_feature_options()->embedding_size(), in ValidateAndInitialize()
|
| D | model.fbs | 509 embedding_size:int = -1;
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ops/ |
| D | embeddings_ops.py | 73 def categorical_variable(tensor_in, n_classes, embedding_size, name): argument
91 [n_classes, embedding_size])
|
| D | ops_test.py | 66 cat_var_idx, n_classes=5, embedding_size=10, name="my_cat_var")
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | core_rnn_cell.py | 203 embedding_size, argument
224 if embedding_classes <= 0 or embedding_size <= 0:
226 "%d, %d." % (embedding_classes, embedding_size))
229 self._embedding_size = embedding_size
|
| /external/tensorflow/tensorflow/examples/tutorials/word2vec/ |
| D | word2vec_basic.py | 147 embedding_size = 128 # Dimension of the embedding vector.
175 tf.random_uniform([vocabulary_size, embedding_size], -1.0, 1.0))
181 tf.truncated_normal([vocabulary_size, embedding_size],
182 stddev=1.0 / math.sqrt(embedding_size)))
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | callbacks_v1.py | 269 embedding_size = np.prod(embedding_input.shape[1:])
271 (step, int(embedding_size)))
272 shape = (self.embeddings_data[0].shape[0], int(embedding_size))
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | hierarchical_controller.py | 177 self.embedding_size = self.group_emb_size
195 self.embedding_size + self.hparams.hidden_size / 2,
199 self.embedding_size + self.hparams.hidden_size / 2,
231 self.embedding_size + self.hparams.hidden_size,
749 embedding_size = array_ops.shape(x)[2]
754 signals[i], [self.hparams.num_children, embedding_size])
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | gru_v2_test.py | 511 embedding_size = 11
520 x = random_ops.random_uniform([1, time_steps, embedding_size])
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop_test.py | 201 embedding_size = 512
204 random_init = random_ops.random_uniform([vocab_size, embedding_size])
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | rnn_cell_test.py | 277 rnn_cell_impl.GRUCell(2), embedding_classes=3, embedding_size=2)
297 embedding_size=2)
|