Searched defs:embedding_size (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op_test.cc | 283 void BM_ScatterHelper(::testing::benchmark::State& state, int embedding_size, in BM_ScatterHelper()
317 const int embedding_size = state.range(0); in BM_ScatterUpdateInt32() local
322 const int embedding_size = state.range(0); in BM_ScatterUpdateInt64() local
328 const int embedding_size = state.range(0); in BM_ScatterAddInt32() local
334 const int embedding_size = state.range(0); in BM_ScatterAddInt32Large() local
339 const int embedding_size = state.range(0); in BM_ScatterAddInt64() local
345 const int embedding_size = state.range(0); in BM_ScatterMulInt32() local
350 const int embedding_size = state.range(0); in BM_ScatterMulInt64() local
356 const int embedding_size = state.range(0); in BM_ScatterDivInt32() local
361 const int embedding_size = state.range(0); in BM_ScatterDivInt64() local
[all …]
|
| D | scatter_nd_op_test.cc | 257 void BM_ScatterNdHelper(::testing::benchmark::State& state, int embedding_size, in BM_ScatterNdHelper()
291 const int embedding_size = state.range(0); in BM_ScatterNdUpdateInt32() local
296 const int embedding_size = state.range(0); in BM_ScatterNdUpdateInt64() local
302 const int embedding_size = state.range(0); in BM_ScatterNdAddInt32() local
307 const int embedding_size = state.range(0); in BM_ScatterNdAddInt64() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse.cc | 127 float current_squares_weight, int embedding_size, in FinalizeAggregation()
180 size_t embedding_size = 1; in Eval() local
|
| /external/libtextclassifier/native/actions/ |
| D | feature-processor.cc | 86 const int embedding_size = options_->embedding_size(); in AppendFeatures() local
|
| /external/libtextclassifier/native/annotator/ |
| D | model-executor.cc | 46 const flatbuffers::Vector<uint8_t>* model_spec_buffer, int embedding_size, in FromBuffer()
|
| D | feature-processor.cc | 867 const int embedding_size = GetOptions()->embedding_size(); in AppendTokenFeaturesWithCache() local
|
| /external/cronet/third_party/icu/source/common/ |
| D | lstmbe.cpp | 392 int32_t embedding_size = ures_getInt(embeddings_res.getAlias(), &status); in LSTMData() local
|
| /external/icu/icu4c/source/common/ |
| D | lstmbe.cpp | 392 int32_t embedding_size = ures_getInt(embeddings_res.getAlias(), &status); in LSTMData() local
|