| /external/tensorflow/tensorflow/core/kernels/ |
| D | hinge-loss.h | 75 double ComputeDualLoss(const double current_dual, const double example_label, in ComputeDualLoss() argument
79 const double y_alpha = current_dual * example_label; // y \alpha in ComputeDualLoss()
89 double ComputePrimalLoss(const double wx, const double example_label, in ComputePrimalLoss() argument
91 const double y_wx = example_label * wx; in ComputePrimalLoss()
109 Status ConvertLabel(float* const example_label) const final { in ConvertLabel() argument
110 if (*example_label == 0.0) { in ConvertLabel()
111 *example_label = -1; in ConvertLabel()
114 if (*example_label == 1.0) { in ConvertLabel()
120 *example_label); in ConvertLabel()
|
| D | smooth-hinge-loss.h | 56 double ComputeDualLoss(const double current_dual, const double example_label, in ComputeDualLoss() argument
60 const double y_alpha = current_dual * example_label; // y \alpha in ComputeDualLoss()
68 double ComputePrimalLoss(const double wx, const double example_label, in ComputePrimalLoss() argument
70 const double y_wx = example_label * wx; in ComputePrimalLoss()
78 Status ConvertLabel(float* const example_label) const final { in ConvertLabel() argument
79 if (*example_label == 0.0) { in ConvertLabel()
80 *example_label = -1; in ConvertLabel()
83 if (*example_label == 1.0) { in ConvertLabel()
89 *example_label); in ConvertLabel()
|
| D | logistic-loss.h | 48 double ComputeDualLoss(const double current_dual, const double example_label, in ComputeDualLoss() argument
52 const double ay = current_dual * example_label; in ComputeDualLoss()
61 double ComputePrimalLoss(const double wx, const double example_label, in ComputePrimalLoss() argument
68 const double y_wx = example_label * wx; in ComputePrimalLoss()
99 Status ConvertLabel(float* const example_label) const final { in ConvertLabel() argument
100 if (*example_label == 0.0) { in ConvertLabel()
101 *example_label = -1; in ConvertLabel()
104 if (*example_label == 1.0) { in ConvertLabel()
110 *example_label); in ConvertLabel()
|
| D | poisson-loss.h | 49 double ComputeDualLoss(const double current_dual, const double example_label, in ComputeDualLoss() argument
54 const double y_minus_a = example_label - current_dual; in ComputeDualLoss()
65 double ComputePrimalLoss(const double wx, const double example_label, in ComputePrimalLoss() argument
67 return (exp(wx) - wx * example_label) * example_weight; in ComputePrimalLoss()
81 Status ConvertLabel(float* const example_label) const final { in ConvertLabel() argument
82 if (*example_label < 0.0) { in ConvertLabel()
85 "Found example with label: ", *example_label); in ConvertLabel()
|
| D | loss_test.cc | 195 float example_label = 1.0; in TEST() local
199 TF_EXPECT_OK(loss_updater.ConvertLabel(&example_label)); in TEST()
200 EXPECT_EQ(1.0, example_label); in TEST()
203 example_label = 0.0; in TEST()
204 TF_EXPECT_OK(loss_updater.ConvertLabel(&example_label)); in TEST()
205 EXPECT_EQ(-1.0, example_label); in TEST()
208 example_label = 0.5; in TEST()
209 status = loss_updater.ConvertLabel(&example_label); in TEST()
339 float example_label = -1.0; in TEST() local
341 Status status = loss_updater.ConvertLabel(&example_label); in TEST()
|
| D | squared-loss.h | 44 double ComputeDualLoss(const double current_dual, const double example_label, in ComputeDualLoss() argument
48 return current_dual * (0.5 * current_dual - example_label) * example_weight; in ComputeDualLoss()
52 double ComputePrimalLoss(const double wx, const double example_label, in ComputePrimalLoss() argument
54 const double error = wx - example_label; in ComputePrimalLoss()
66 Status ConvertLabel(float* const example_label) const final { in ConvertLabel() argument
|
| D | loss.h | 37 const double example_label,
42 virtual double ComputePrimalLoss(const double wx, const double example_label,
47 const double example_label,
55 virtual Status ConvertLabel(float* const example_label) const = 0;
|
| D | sdca_ops.cc | 180 float example_label = example.example_label(); in DoCompute() local
182 options.loss_updater->ConvertLabel(&example_label); in DoCompute()
200 options.num_loss_partitions, example_label, example_weight, dual, in DoCompute()
215 example_statistics.prev_wx[0], example_label, example_weight); in DoCompute()
217 options.loss_updater->ComputeDualLoss(dual, example_label, in DoCompute()
|
| D | sdca_internal.h | 138 float example_label() const { return example_label_; } in example_label() function
|
| D | sdca_internal.cc | 250 float label = example.example_label(); in SampleAdaptiveProbabilities()
|