Searched refs:num_weight_vectors (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dsdca_internal.h72 ExampleStatistics(const int num_weight_vectors) argument
73 : wx(num_weight_vectors, 0.0), prev_wx(num_weight_vectors, 0.0) {}
133 // For multiclass case, num_weight_vectors equals to the number of classes;
138 const int num_weight_vectors) const;
331 // binary classification, where the input argument for num_weight_vectors
338 const int num_weight_vectors);
H A Dsdca_internal.cc37 const size_t num_weight_vectors = normalized_bounded_dual_delta.size(); local
38 if (num_weight_vectors == 1) {
45 dual_matrix(normalized_bounded_dual_delta.data(), num_weight_vectors,
150 const Regularizations& regularization, const int num_weight_vectors) const {
151 ExampleStatistics result(num_weight_vectors);
168 for (int l = 0; l < num_weight_vectors; ++l) {
191 if (num_weight_vectors == 1) {
217 // above is of dimension num_weight_vectors * 1.
218 for (int l = 0; l < num_weight_vectors; ++l) {
234 const int num_weight_vectors) {
229 SampleAdaptativeProbabilities( const int num_loss_partitions, const Regularizations& regularization, const ModelWeights& model_weights, const TTypes<float>::Matrix example_state_data, const std::unique_ptr<DualLossUpdater>& loss_updater, const int num_weight_vectors) argument
[all...]

Completed in 98 milliseconds