Searched defs:embedding_size (Results 1 – 6 of 6) sorted by relevance
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() local322 const int embedding_size = state.range(0); in BM_ScatterUpdateInt64() local328 const int embedding_size = state.range(0); in BM_ScatterAddInt32() local334 const int embedding_size = state.range(0); in BM_ScatterAddInt32Large() local339 const int embedding_size = state.range(0); in BM_ScatterAddInt64() local345 const int embedding_size = state.range(0); in BM_ScatterMulInt32() local350 const int embedding_size = state.range(0); in BM_ScatterMulInt64() local356 const int embedding_size = state.range(0); in BM_ScatterDivInt32() local361 const int embedding_size = state.range(0); in BM_ScatterDivInt64() local[all …]
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() local296 const int embedding_size = state.range(0); in BM_ScatterNdUpdateInt64() local302 const int embedding_size = state.range(0); in BM_ScatterNdAddInt32() local307 const int embedding_size = state.range(0); in BM_ScatterNdAddInt64() local
126 float current_squares_weight, int embedding_size, in FinalizeAggregation()177 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()
867 const int embedding_size = GetOptions()->embedding_size(); in AppendTokenFeaturesWithCache() local