Searched refs:example_label (Results 1 - 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dloss.h37 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;
H A Dhinge-loss.h75 double ComputeDualLoss(const double current_dual, const double example_label,
79 const double y_alpha = current_dual * example_label; // y \alpha
89 double ComputePrimalLoss(const double wx, const double example_label,
91 const double y_wx = example_label * wx;
109 Status ConvertLabel(float* const example_label) const final {
110 if (*example_label == 0.0) {
111 *example_label = -1;
114 if (*example_label == 1.0) {
120 *example_label);
H A Dsmooth-hinge-loss.h56 double ComputeDualLoss(const double current_dual, const double example_label,
60 const double y_alpha = current_dual * example_label; // y \alpha
68 double ComputePrimalLoss(const double wx, const double example_label,
70 const double y_wx = example_label * wx;
78 Status ConvertLabel(float* const example_label) const final {
79 if (*example_label == 0.0) {
80 *example_label = -1;
83 if (*example_label == 1.0) {
89 *example_label);
H A Dsquared-loss.h44 double ComputeDualLoss(const double current_dual, const double example_label,
48 return current_dual * (0.5 * current_dual - example_label) * example_weight;
52 double ComputePrimalLoss(const double wx, const double example_label,
54 const double error = wx - example_label;
66 Status ConvertLabel(float* const example_label) const final {
H A Dlogistic-loss.h48 double ComputeDualLoss(const double current_dual, const double example_label,
52 const double ay = current_dual * example_label;
61 double ComputePrimalLoss(const double wx, const double example_label,
68 const double y_wx = example_label * wx;
99 Status ConvertLabel(float* const example_label) const final {
100 if (*example_label == 0.0) {
101 *example_label = -1;
104 if (*example_label == 1.0) {
110 *example_label);
H A Dloss_test.cc190 float example_label = 1.0; local
194 TF_EXPECT_OK(loss_updater.ConvertLabel(&example_label));
195 EXPECT_EQ(1.0, example_label);
198 example_label = 0.0;
199 TF_EXPECT_OK(loss_updater.ConvertLabel(&example_label));
200 EXPECT_EQ(-1.0, example_label);
203 example_label = 0.5;
204 status = loss_updater.ConvertLabel(&example_label);
H A Dsdca_ops.cc170 float example_label = example.example_label();
172 options.loss_updater->ConvertLabel(&example_label);
190 options.num_loss_partitions, example_label, example_weight, dual,
205 example_statistics.prev_wx[0], example_label, example_weight);
207 options.loss_updater->ComputeDualLoss(dual, example_label,
H A Dsdca_internal.h140 float example_label() const { return example_label_; } function in class:tensorflow::sdca::Example
H A Dsdca_internal.cc244 float label = example.example_label();

Completed in 509 milliseconds