| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op_test.cc | 283 void BM_ScatterHelper(::testing::benchmark::State& state, int embedding_size, in BM_ScatterHelper() argument
285 const int kRows = 10000000 / embedding_size; in BM_ScatterHelper()
288 for (int i = 0; i < kRows * embedding_size; i++) { in BM_ScatterHelper()
298 for (int j = 0; j < embedding_size; j++) { in BM_ScatterHelper()
305 bm.AddInputFromArray<float>(TensorShape({kRows, embedding_size}), values); in BM_ScatterHelper()
307 bm.AddInputFromArray<float>(TensorShape({kNumUpdates, embedding_size}), in BM_ScatterHelper()
312 state.SetItemsProcessed((static_cast<int64>(kNumUpdates) * embedding_size) * in BM_ScatterHelper()
317 const int embedding_size = state.range(0); in BM_ScatterUpdateInt32() local
319 BM_ScatterHelper<int32>(state, embedding_size, "ScatterUpdate"); in BM_ScatterUpdateInt32()
322 const int embedding_size = state.range(0); in BM_ScatterUpdateInt64() local
[all …]
|
| D | scatter_nd_op_test.cc | 257 void BM_ScatterNdHelper(::testing::benchmark::State& state, int embedding_size, in BM_ScatterNdHelper() argument
259 const int kRows = 10000000 / embedding_size; in BM_ScatterNdHelper()
262 for (int i = 0; i < kRows * embedding_size; i++) { in BM_ScatterNdHelper()
272 for (int j = 0; j < embedding_size; j++) { in BM_ScatterNdHelper()
279 bm.AddInputFromArray<float>(TensorShape({kRows, embedding_size}), values); in BM_ScatterNdHelper()
281 bm.AddInputFromArray<float>(TensorShape({kNumUpdates, embedding_size}), in BM_ScatterNdHelper()
286 state.SetItemsProcessed((static_cast<int64>(kNumUpdates) * embedding_size) * in BM_ScatterNdHelper()
291 const int embedding_size = state.range(0); in BM_ScatterNdUpdateInt32() local
293 BM_ScatterNdHelper<int32>(state, embedding_size, "ScatterNdUpdate"); in BM_ScatterNdUpdateInt32()
296 const int embedding_size = state.range(0); in BM_ScatterNdUpdateInt64() local
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse.cc | 126 float current_squares_weight, int embedding_size, in FinalizeAggregation() argument
140 for (int k = 0; k < embedding_size; k++) { in FinalizeAggregation()
177 int embedding_size = 1; in Eval() local
186 embedding_size *= dim; in Eval()
190 const int output_size = lookup_size * embedding_size; in Eval()
223 const int output_offset = output_bucket * embedding_size; in Eval()
229 current_squares_weight, embedding_size, in Eval()
241 const int example_embedding_offset = idx * embedding_size; in Eval()
245 for (int k = 0; k < embedding_size; k++) { in Eval()
253 current_squares_weight, embedding_size, in Eval()
|
| /external/libtextclassifier/native/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 | 109 // A (max tokens, embedding_size) float tensor specifying the embeddings of
156 embedding_size:int = -1;
|
| D | actions-suggestions.cc | 373 options->embedding_model(), options->embedding_size(), in ValidateAndInitialize()
|
| D | actions-suggestions_test.cc | 1421 options_->embedding_size = 1; in EmbeddingTest()
|
| /external/libtextclassifier/native/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 | 867 const int embedding_size = GetOptions()->embedding_size(); in AppendTokenFeaturesWithCache() local
868 output_features->resize(output_features->size() + embedding_size); in AppendTokenFeaturesWithCache()
874 /*dest=*/output_features_end - embedding_size, in AppendTokenFeaturesWithCache()
875 /*dest_size=*/embedding_size)) { in AppendTokenFeaturesWithCache()
884 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 | 192 int EmbeddingSize() const { return options_->embedding_size(); } in EmbeddingSize()
|
| D | feature-processor_test.cc | 585 options.embedding_size = 4; in TEST_F()
696 options.embedding_size = 4; in TEST_F()
732 options.embedding_size = 4; in TEST_F()
|
| D | annotator.cc | 393 (model_->selection_feature_options()->embedding_size() != in ValidateAndInitialize()
394 model_->classification_feature_options()->embedding_size() || in ValidateAndInitialize()
404 model_->classification_feature_options()->embedding_size(), in ValidateAndInitialize()
|
| D | model.fbs | 662 embedding_size:int = -1;
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | callbacks_v1.py | 271 embedding_size = np.prod(embedding_input.shape[1:])
273 (step, int(embedding_size)))
274 shape = (self.embeddings_data[0].shape[0], int(embedding_size))
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | gru_v2_test.py | 678 embedding_size = 11
687 x = random_ops.random_uniform([1, time_steps, embedding_size])
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop_test.py | 266 embedding_size = 512
269 random_init = random_ops.random_uniform([vocab_size, embedding_size])
|