Home
last modified time | relevance | path

Searched defs:embedding_size (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dscatter_op_test.cc283 static void BM_ScatterHelper(int iters, int embedding_size, const char* op, in BM_ScatterHelper()
319 static void BM_ScatterUpdateInt32(int iters, int embedding_size) { in BM_ScatterUpdateInt32()
322 static void BM_ScatterUpdateInt64(int iters, int embedding_size) { in BM_ScatterUpdateInt64()
326 static void BM_ScatterAddInt32(int iters, int embedding_size) { in BM_ScatterAddInt32()
330 static void BM_ScatterAddInt32Large(int iters, int embedding_size) { in BM_ScatterAddInt32Large()
333 static void BM_ScatterAddInt64(int iters, int embedding_size) { in BM_ScatterAddInt64()
337 static void BM_ScatterMulInt32(int iters, int embedding_size) { in BM_ScatterMulInt32()
340 static void BM_ScatterMulInt64(int iters, int embedding_size) { in BM_ScatterMulInt64()
344 static void BM_ScatterDivInt32(int iters, int embedding_size) { in BM_ScatterDivInt32()
347 static void BM_ScatterDivInt64(int iters, int embedding_size) { in BM_ScatterDivInt64()
[all …]
Dscatter_nd_op_test.cc254 static void BM_ScatterNdHelper(int iters, int embedding_size, const char* op) { in BM_ScatterNdHelper()
289 static void BM_ScatterNdUpdateInt32(int iters, int embedding_size) { in BM_ScatterNdUpdateInt32()
292 static void BM_ScatterNdUpdateInt64(int iters, int embedding_size) { in BM_ScatterNdUpdateInt64()
296 static void BM_ScatterNdAddInt32(int iters, int embedding_size) { in BM_ScatterNdAddInt32()
299 static void BM_ScatterNdAddInt64(int iters, int embedding_size) { in BM_ScatterNdAddInt64()
/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_sparse.cc119 float current_squares_weight, int embedding_size, in FinalizeAggregation()
164 int embedding_size = 1; in Eval() local
/external/libtextclassifier/native/actions/
Dfeature-processor.cc86 const int embedding_size = options_->embedding_size(); in AppendFeatures() local
/external/libtextclassifier/native/annotator/
Dmodel-executor.cc46 const flatbuffers::Vector<uint8_t>* model_spec_buffer, int embedding_size, in FromBuffer()
Dfeature-processor.cc903 const int embedding_size = GetOptions()->embedding_size(); in AppendTokenFeaturesWithCache() local