Home
last modified time | relevance | path

Searched refs:loss_updater (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dloss_test.cc37 void TestComputeUpdatedDual(const DualLossUpdater &loss_updater, in TestComputeUpdatedDual() argument
42 double new_dual = loss_updater.ComputeUpdatedDual( in TestComputeUpdatedDual()
48 EXPECT_NEAR(new_dual, -loss_updater.PrimalLossDerivative(new_wx, label, 1.0), in TestComputeUpdatedDual()
53 LogisticLossUpdater loss_updater; in TEST() local
55 loss_updater.ComputePrimalLoss(0 /* wx */, 1 /* label */, in TEST()
59 loss_updater.ComputePrimalLoss(70 /* wx */, 1 /* label */, in TEST()
63 loss_updater.ComputePrimalLoss(-70 /* wx */, -1 /* label */, in TEST()
69 LogisticLossUpdater loss_updater; in TEST() local
71 loss_updater.ComputeDualLoss(0 /* current dual */, 1 /* label */, in TEST()
75 loss_updater.ComputeDualLoss(1 /* current dual */, 1 /* label */, in TEST()
[all …]
Dsdca_ops.cc72 loss_updater.reset(new LogisticLossUpdater); in ComputeOptions()
74 loss_updater.reset(new SquaredLossUpdater); in ComputeOptions()
76 loss_updater.reset(new HingeLossUpdater); in ComputeOptions()
78 loss_updater.reset(new SmoothHingeLossUpdater); in ComputeOptions()
80 loss_updater.reset(new PoissonLossUpdater); in ComputeOptions()
117 std::unique_ptr<DualLossUpdater> loss_updater; member
161 model_weights, example_state_data, options.loss_updater, in DoCompute()
181 options.loss_updater->ConvertLabel(&example_label); in DoCompute()
198 const double new_dual = options.loss_updater->ComputeUpdatedDual( in DoCompute()
213 options.loss_updater->ComputePrimalLoss( in DoCompute()
[all …]
Dsdca_internal.cc239 const std::unique_ptr<DualLossUpdater>& loss_updater, in SampleAdaptiveProbabilities() argument
251 const Status conversion_status = loss_updater->ConvertLabel(&label); in SampleAdaptiveProbabilities()
257 loss_updater->PrimalLossDerivative( in SampleAdaptiveProbabilities()
262 loss_updater->SmoothnessConstant()) * in SampleAdaptiveProbabilities()
Dsdca_internal.h332 const std::unique_ptr<DualLossUpdater>& loss_updater,