Searched defs:num_loss_partitions (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | poisson-loss.h | 29 double ComputeUpdatedDual(const int num_loss_partitions, const double label, in ComputeUpdatedDual()
92 double NewtonStep(const double x, const int num_loss_partitions, in NewtonStep()
|
| H A D | logistic-loss.h | 31 double ComputeUpdatedDual(const int num_loss_partitions, const double label, in ComputeUpdatedDual()
115 double NewtonStep(const double x, const int num_loss_partitions, in NewtonStep()
|
| H A D | sdca_internal.cc | 169 const int num_loss_partitions, const ModelWeights& model_weights, in ComputeWxAndWeightedExampleNorm()
250 const int num_loss_partitions, const Regularizations& regularization, in SampleAdaptiveProbabilities()
|
| H A D | squared-loss.h | 32 double ComputeUpdatedDual(const int num_loss_partitions, const double label, in ComputeUpdatedDual()
|
| H A D | hinge-loss.h | 42 double ComputeUpdatedDual(const int num_loss_partitions, const double label, in ComputeUpdatedDual()
|
| H A D | loss_test.cc | 38 const int num_loss_partitions, const double label, in TestComputeUpdatedDual()
|
| H A D | sdca_ops.cc | 123 int num_loss_partitions = 0; member
|