Home
last modified time | relevance | path

Searched refs:num_loss_partitions (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/linear_optimizer/python/
Dsdca_estimator.py116 num_loss_partitions = params["num_loss_partitions"]
135 n_loss_partitions = num_loss_partitions or max(1, config.num_worker_replicas)
138 num_loss_partitions=n_loss_partitions,
217 num_loss_partitions=None, argument
271 "num_loss_partitions": num_loss_partitions,
344 num_loss_partitions=None, argument
387 num_loss_partitions=num_loss_partitions,
475 num_loss_partitions=None, argument
518 num_loss_partitions=num_loss_partitions,
Dsdca_optimizer.py73 num_loss_partitions=1, argument
80 self._num_loss_partitions = num_loss_partitions
95 def num_loss_partitions(self): member in SDCAOptimizer
274 num_loss_partitions=self._num_loss_partitions,
/external/tensorflow/tensorflow/core/kernels/
Dpoisson-loss.h29 double ComputeUpdatedDual(const int num_loss_partitions, const double label, in ComputeUpdatedDual() argument
41 x = NewtonStep(x, num_loss_partitions, label, wx, example_weight, in ComputeUpdatedDual()
92 double NewtonStep(const double x, const int num_loss_partitions, in NewtonStep() argument
99 x - wx - num_loss_partitions * weighted_example_norm * in NewtonStep()
102 1 + num_loss_partitions * weighted_example_norm * example_weight * expx; in NewtonStep()
Dlogistic-loss.h31 double ComputeUpdatedDual(const int num_loss_partitions, const double label, in ComputeUpdatedDual() argument
40 x = NewtonStep(x, num_loss_partitions, label, wx, example_weight, in ComputeUpdatedDual()
115 double NewtonStep(const double x, const int num_loss_partitions, in NewtonStep() argument
122 num_loss_partitions * weighted_example_norm * in NewtonStep()
126 -2 * label - num_loss_partitions * weighted_example_norm * in NewtonStep()
Dsdca_ops.cc111 context, context->GetAttr("num_loss_partitions", &num_loss_partitions)); in ComputeOptions()
122 int num_loss_partitions = 0; member
160 options.num_loss_partitions, options.regularizations, in DoCompute()
195 options.num_loss_partitions, model_weights, in DoCompute()
199 options.num_loss_partitions, example_label, example_weight, dual, in DoCompute()
Dsquared-loss.h32 double ComputeUpdatedDual(const int num_loss_partitions, const double label, in ComputeUpdatedDual() argument
38 1 + num_loss_partitions * weighted_example_norm * example_weight; in ComputeUpdatedDual()
Dhinge-loss.h42 double ComputeUpdatedDual(const int num_loss_partitions, const double label, in ComputeUpdatedDual() argument
53 current_dual + (label - wx) / (num_loss_partitions * example_weight * in ComputeUpdatedDual()
Dloss_test.cc38 const int num_loss_partitions, const double label, in TestComputeUpdatedDual() argument
43 num_loss_partitions, label, example_weight, current_dual, wx, in TestComputeUpdatedDual()
46 double new_wx = wx + (new_dual - current_dual) * num_loss_partitions * in TestComputeUpdatedDual()
Dsdca_internal.cc155 const int num_loss_partitions, const ModelWeights& model_weights, in ComputeWxAndWeightedExampleNorm() argument
178 sparse_weights.deltas(l, feature_index) * num_loss_partitions; in ComputeWxAndWeightedExampleNorm()
196 dense_weights.deltas().constant(num_loss_partitions); in ComputeWxAndWeightedExampleNorm()
236 const int num_loss_partitions, const Regularizations& regularization, in SampleAdaptiveProbabilities() argument
253 example.ComputeWxAndWeightedExampleNorm(num_loss_partitions, in SampleAdaptiveProbabilities()
Dloss.h30 const int num_loss_partitions, const double label,
Dsdca_internal.h134 const int num_loss_partitions, const ModelWeights& model_weights,
329 const int num_loss_partitions, const Regularizations& regularization,
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dsvm.py101 num_loss_partitions=1, argument
140 num_loss_partitions=num_loss_partitions,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.estimator.experimental.-linear-s-d-c-a.pbtxt7 …argspec: "args=[\'self\', \'example_id_column\', \'num_loss_partitions\', \'num_table_shards\', \'…
Dtensorflow.train.pbtxt429 …eights\', \'example_state_data\', \'loss_type\', \'l1\', \'l2\', \'num_loss_partitions\', \'num_in…
Dtensorflow.raw_ops.pbtxt3229 …eights\', \'example_state_data\', \'loss_type\', \'l1\', \'l2\', \'num_loss_partitions\', \'num_in…
3233 …eights\', \'example_state_data\', \'loss_type\', \'l1\', \'l2\', \'num_loss_partitions\', \'num_in…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.estimator.experimental.-linear-s-d-c-a.pbtxt7 …argspec: "args=[\'self\', \'example_id_column\', \'num_loss_partitions\', \'num_table_shards\', \'…
Dtensorflow.train.pbtxt101 …eights\', \'example_state_data\', \'loss_type\', \'l1\', \'l2\', \'num_loss_partitions\', \'num_in…
Dtensorflow.raw_ops.pbtxt3229 …eights\', \'example_state_data\', \'loss_type\', \'l1\', \'l2\', \'num_loss_partitions\', \'num_in…
3233 …eights\', \'example_state_data\', \'loss_type\', \'l1\', \'l2\', \'num_loss_partitions\', \'num_in…
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/kernel_tests/
Dsdca_ops_test.py469 for num_loss_partitions in _NUM_LOSS_PARTITIONS:
475 symmetric_l2_regularization=num_loss_partitions,
479 num_loss_partitions=num_loss_partitions)
500 for worker_id in range(num_loss_partitions):
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/
Dsdca_ops.py520 num_loss_partitions=self._num_loss_partitions(),
539 num_loss_partitions=self._num_loss_partitions(),
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SdcaOptimizerV2.pbtxt140 name: "num_loss_partitions"
Dapi_def_SdcaOptimizer.pbtxt139 name: "num_loss_partitions"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go8490 …le_state_data tf.Output, loss_type string, l1 float32, l2 float32, num_loss_partitions int64, num_…
8494 …"loss_type": loss_type, "l1": l1, "l2": l2, "num_loss_partitions": num_loss_partitions, "num_inner…
16274 …le_state_data tf.Output, loss_type string, l1 float32, l2 float32, num_loss_partitions int64, num_…
16278 …"loss_type": loss_type, "l1": l1, "l2": l2, "num_loss_partitions": num_loss_partitions, "num_inner…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt33578 name: "num_loss_partitions"
33697 name: "num_loss_partitions"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt67265 name: "num_loss_partitions"
67384 name: "num_loss_partitions"
67503 name: "num_loss_partitions"

12