Searched defs:embedding_size (Results 1 – 10 of 10) sorted by relevance
244 static void BM_ScatterHelper(int iters, int embedding_size, const char* op) { in BM_ScatterHelper()279 static void BM_ScatterUpdateInt32(int iters, int embedding_size) { in BM_ScatterUpdateInt32()282 static void BM_ScatterUpdateInt64(int iters, int embedding_size) { in BM_ScatterUpdateInt64()286 static void BM_ScatterAddInt32(int iters, int embedding_size) { in BM_ScatterAddInt32()289 static void BM_ScatterAddInt64(int iters, int embedding_size) { in BM_ScatterAddInt64()293 static void BM_ScatterMulInt32(int iters, int embedding_size) { in BM_ScatterMulInt32()296 static void BM_ScatterMulInt64(int iters, int embedding_size) { in BM_ScatterMulInt64()300 static void BM_ScatterDivInt32(int iters, int embedding_size) { in BM_ScatterDivInt32()303 static void BM_ScatterDivInt64(int iters, int embedding_size) { in BM_ScatterDivInt64()307 static void BM_ScatterMinInt32(int iters, int embedding_size) { in BM_ScatterMinInt32()[all …]
254 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()
79 def _one_hot_to_embedding(one_hot, embedding_size): argument94 def condition_tensor_from_onehot(tensor, one_hot_labels, embedding_size=256): argument
237 embedding_size, argument306 embedding_size, argument414 embedding_size, argument716 embedding_size, argument802 embedding_size, argument932 embedding_size, argument
73 def categorical_variable(tensor_in, n_classes, embedding_size, name): argument
118 float current_squares_weight, int embedding_size, in FinalizeAggregation()163 int embedding_size = 1; in Eval() local
86 const int embedding_size = options_->embedding_size(); in AppendFeatures() local
46 const flatbuffers::Vector<uint8_t>* model_spec_buffer, int embedding_size, in FromBuffer()
837 const int embedding_size = GetOptions()->embedding_size(); in AppendTokenFeaturesWithCache() local
203 embedding_size, argument