| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | prediction_ops_test.py | 56 logits_dimension=1)
123 logits_dimension=1)
196 logits_dimension=1)
306 logits_dimension=1)
436 logits_dimension=1)
518 logits_dimension=1)
642 logits_dimension=1)
785 logits_dimension=1)
862 logits_dimension=1)
894 logits_dimension=1)
[all …]
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | head.py | 138 def logits_dimension(self): member in Head
509 self, problem_type, logits_dimension, label_name=None, argument
513 if logits_dimension is None or logits_dimension < 1:
514 raise ValueError("Invalid logits_dimension %s." % logits_dimension)
516 self._logits_dimension = logits_dimension
522 def logits_dimension(self): member in _SingleHead
592 def _logits(logits_input, logits, logits_dimension): argument
610 if (logits_dimension is None) or (logits_dimension < 1):
611 raise ValueError("Invalid logits_dimension %s." % logits_dimension)
617 return layers_lib.linear(logits_input, logits_dimension, scope="logits")
[all …]
|
| D | composable_model_test.py | 80 num_label_columns=head.logits_dimension),
94 num_label_columns=head.logits_dimension, _joint_weights=True),
108 num_label_columns=head.logits_dimension,
|
| D | dnn_linear_combined.py | 276 head.logits_dimension,
301 num_outputs=head.logits_dimension,
308 num_outputs=head.logits_dimension,
315 units=head.logits_dimension,
|
| D | linear.py | 167 num_outputs=head.logits_dimension,
174 units=head.logits_dimension,
238 assert head.logits_dimension == 1, ("SDCA only applies for "
|
| D | debug.py | 108 logits = array_ops.zeros([batch_size, head.logits_dimension])
|
| D | dnn.py | 187 head.logits_dimension,
|
| D | dnn_linear_combined_test.py | 74 def logits_dimension(self): member in _CheckCallsHead
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | estimator.py | 44 logits_dimension=None, argument
79 logits_dimension=logits_dimension,
100 logits_dimension=None, argument
174 logits_dimension=logits_dimension,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tensor_forest_ops.cc | 74 int logits_dimension; in __anoncf2fc2a90302() local
75 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anoncf2fc2a90302()
76 c->set_output(0, c->Matrix(batch_size, logits_dimension)); in __anoncf2fc2a90302()
|
| D | boosted_trees_ops.cc | 204 int logits_dimension; in __anone3ce78ae0702() local
205 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anone3ce78ae0702()
207 c->MakeShape({c->Dim(feature_shape, 0), logits_dimension}); in __anone3ce78ae0702()
275 int logits_dimension; in __anone3ce78ae0a02() local
276 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anone3ce78ae0a02()
278 c->MakeShape({c->Dim(feature_shape, 0), logits_dimension}); in __anone3ce78ae0a02()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | model.py | 124 logits_dimension=head.logits_dimension,
309 logits_dimension=head.logits_dimension,
331 logits_dimension=head.logits_dimension,
|
| D | dnn_tree_combined_estimator.py | 215 head.logits_dimension,
264 logits_dimension=head.logits_dimension,
288 n_classes = 2 if head.logits_dimension == 1 else head.logits_dimension
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesCenterBias.pbtxt | 13 A tensor with shape=[logits_dimension] with mean of gradients for a first node.
19 A tensor with shape=[logits_dimension] mean of hessians for a first node.
|
| D | api_def_TensorForestTreePredict.pbtxt | 5 name: "logits_dimension"
|
| D | api_def_BoostedTreesPredict.pbtxt | 24 name: "logits_dimension"
|
| D | api_def_BoostedTreesExampleDebugOutputs.pbtxt | 24 name: "logits_dimension"
|
| D | api_def_BoostedTreesTrainingPredict.pbtxt | 51 name: "logits_dimension"
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch_test.py | 236 logits_dimension=1,
344 logits_dimension=1,
500 logits_dimension=1,
604 logits_dimension=1,
708 logits_dimension=1,
780 logits_dimension=1,
844 logits_dimension=1,
929 logits_dimension=1,
1033 logits_dimension=1,
1081 logits_dimension=5,
[all …]
|
| D | gbdt_batch.py | 302 logits_dimension, argument
345 self._logits_dimension = logits_dimension
362 if logits_dimension == 1:
369 if logits_dimension == 1 or learner_config.multi_class_strategy == (
377 self._gradient_shape = tensor_shape.TensorShape([logits_dimension])
381 ([logits_dimension, logits_dimension]))
384 self._hessian_shape = tensor_shape.TensorShape(([logits_dimension]))
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| D | sdca_estimator.py | 131 assert head.logits_dimension == 1, (
265 assert head.logits_dimension == 1
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | training_ops.cc | 225 const int64 logits_dimension = delta_updates_t->dim_size(0); in Compute() local
229 RetrieveBias(ensemble_resource, logits_dimension); in Compute()
260 int64 logits_dimension) { in RetrieveBias() argument
268 for (size_t idx = 0; idx < logits_dimension; ++idx) { in RetrieveBias()
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| D | estimator_test.py | 75 logits_dimension=2)
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | head_impl.py | 147 def logits_dimension(self): member in GANHead
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | head.py | 91 def logits_dimension(self): member in TimeSeriesRegressionHead
|