Searched refs:SetMiniBatchSize (Results 1 - 3 of 3) sorted by relevance
/frameworks/ml/bordeaux/learning/stochastic_linear_ranker/native/ | ||
H A D | stochastic_linear_ranker.h | 50 learning_rate_controller_.SetMiniBatchSize(mini_batch_size_); 127 void SetMiniBatchSize(const uint64 msize) { function in class:learning_stochastic_linear::StochasticLinearRanker 129 learning_rate_controller_.SetMiniBatchSize(msize); |
H A D | learning_rate_controller-inl.h | 60 void SetMiniBatchSize(uint64 size) { function in class:learning_stochastic_linear::LearningRateController |
/frameworks/ml/bordeaux/learning/stochastic_linear_ranker/jni/ | ||
H A D | jni_stochastic_linear_ranker.cpp | 217 classifier->SetMiniBatchSize((uint64) v); |
Completed in 96 milliseconds