| /external/tensorflow/tensorflow/core/kernels/ |
| D | logistic-loss.h | 32 const double example_weight, in ComputeUpdatedDual() argument
40 x = NewtonStep(x, num_loss_partitions, label, wx, example_weight, in ComputeUpdatedDual()
49 const double example_weight) const final { in ComputeDualLoss() argument
56 return ((ay * log_ay) + (one_minus_ay * log_one_minus_ay)) * example_weight; in ComputeDualLoss()
62 const double example_weight) const final { in ComputePrimalLoss() argument
72 return log1p(exp(-y_wx)) * example_weight; in ComputePrimalLoss()
77 return (log1p(exp(y_wx)) - y_wx) * example_weight; in ComputePrimalLoss()
82 const double example_weight) const final { in PrimalLossDerivative() argument
89 return -inverse_exp_term * label * example_weight; in PrimalLossDerivative()
117 const double example_weight, in NewtonStep() argument
[all …]
|
| D | poisson-loss.h | 30 const double example_weight, in ComputeUpdatedDual() argument
41 x = NewtonStep(x, num_loss_partitions, label, wx, example_weight, in ComputeUpdatedDual()
50 const double example_weight) const final { in ComputeDualLoss() argument
62 return y_minus_a * (log(y_minus_a) - 1) * example_weight; in ComputeDualLoss()
66 const double example_weight) const final { in ComputePrimalLoss() argument
67 return (exp(wx) - wx * example_label) * example_weight; in ComputePrimalLoss()
71 const double example_weight) const final { in PrimalLossDerivative() argument
72 return (exp(wx) - label) * example_weight; in PrimalLossDerivative()
94 const double example_weight, in NewtonStep() argument
100 example_weight * (label - current_dual - expx); in NewtonStep()
[all …]
|
| D | squared-loss.h | 33 const double example_weight, in ComputeUpdatedDual() argument
38 1 + num_loss_partitions * weighted_example_norm * example_weight; in ComputeUpdatedDual()
45 const double example_weight) const final { in ComputeDualLoss() argument
48 return current_dual * (0.5 * current_dual - example_label) * example_weight; in ComputeDualLoss()
53 const double example_weight) const final { in ComputePrimalLoss() argument
55 return error * error * example_weight * 0.5; in ComputePrimalLoss()
59 const double example_weight) const final { in PrimalLossDerivative() argument
60 return (wx - label) * example_weight; in PrimalLossDerivative()
|
| D | hinge-loss.h | 43 const double example_weight, in ComputeUpdatedDual() argument
53 current_dual + (label - wx) / (num_loss_partitions * example_weight * in ComputeUpdatedDual()
76 const double example_weight) const final { in ComputeDualLoss() argument
83 return -y_alpha * example_weight; in ComputeDualLoss()
90 const double example_weight) const final { in ComputePrimalLoss() argument
92 return std::max(0.0, 1 - y_wx) * example_weight; in ComputePrimalLoss()
96 const double example_weight) const final { in PrimalLossDerivative() argument
98 return -label * example_weight; in PrimalLossDerivative()
|
| D | smooth-hinge-loss.h | 34 const double example_weight, in ComputeUpdatedDual() argument
46 (num_partitions * example_weight * weighted_example_norm + gamma); in ComputeUpdatedDual()
57 const double example_weight) const final { in ComputeDualLoss() argument
65 example_weight; in ComputeDualLoss()
69 const double example_weight) const final { in ComputePrimalLoss() argument
72 if (y_wx <= 1 - gamma) return (1 - y_wx - gamma / 2) * example_weight; in ComputePrimalLoss()
73 return (1 - y_wx) * (1 - y_wx) * example_weight * 0.5 / gamma; in ComputePrimalLoss()
93 const double example_weight) const final { in PrimalLossDerivative() argument
|
| D | loss.h | 31 const double example_weight, const double current_dual, const double wx,
38 const double example_weight) const = 0;
43 const double example_weight) const = 0;
48 const double example_weight) const = 0;
|
| D | sdca_ops.cc | 178 const float example_weight = example.example_weight(); in DoCompute() local
199 options.num_loss_partitions, example_label, example_weight, dual, in DoCompute()
204 (new_dual - dual) * example_weight / in DoCompute()
214 example_statistics.prev_wx[0], example_label, example_weight); in DoCompute()
217 example_weight); in DoCompute()
218 example_state_data(example_index, 3) = example_weight; in DoCompute()
|
| D | loss_test.cc | 39 const double example_weight, in TestComputeUpdatedDual() argument
43 num_loss_partitions, label, example_weight, current_dual, wx, in TestComputeUpdatedDual()
47 weighted_example_norm * example_weight; in TestComputeUpdatedDual()
|
| D | sdca_internal.cc | 249 const double example_weight = example.example_weight(); in SampleAdaptiveProbabilities() local
259 probabilities_[example_id] = example_weight * in SampleAdaptiveProbabilities()
|
| D | sdca_internal.h | 140 float example_weight() const { return example_weight_; } in example_weight() function
|