| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| H A D | estimator.py | 35 logits_dimension=None,
48 logits_dimension: Number of logits per example. This is the size of the last
50 `[batch_size, logits_dimension]`).
70 logits_dimension=logits_dimension,
83 logits_dimension=None,
96 logits_dimension: Number of logits per example. This is the size of the
98 `[batch_size, logits_dimension]`).
157 logits_dimension=logits_dimension,
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| H A D | head.py | 76 last_hidden_layer_out, head.logits_dimension, activation_fn=None)
128 def logits_dimension(self): member in class:Head
131 Typically, logits is of shape `[batch_size, logits_dimension]`.
492 self, problem_type, logits_dimension, label_name=None,
496 if logits_dimension is None or logits_dimension < 1:
497 raise ValueError("Invalid logits_dimension %s." % logits_dimension)
499 self._logits_dimension = logits_dimension
505 def logits_dimension(sel member in class:_SingleHead
1451 def logits_dimension(self): member in class:_LossOnlyHead
1571 def logits_dimension(self): member in class:_MultiHead
[all...] |
| H A 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,
|
| H A D | debug.py | 104 logits = array_ops.zeros([batch_size, head.logits_dimension])
|
| H A D | dnn_linear_combined.py | 271 head.logits_dimension,
296 num_outputs=head.logits_dimension,
303 num_outputs=head.logits_dimension,
310 units=head.logits_dimension,
|
| H A D | linear.py | 162 num_outputs=head.logits_dimension,
169 units=head.logits_dimension,
233 assert head.logits_dimension == 1, ("SDCA only applies for "
234 "logits_dimension=1.")
|
| H A D | dnn.py | 182 head.logits_dimension,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| H A D | model.py | 95 logits_dimension=head.logits_dimension,
|
| H A D | dnn_tree_combined_estimator.py | 163 head.logits_dimension,
203 logits_dimension=head.logits_dimension,
|
| /external/tensorflow/tensorflow/python/estimator/canned/ |
| H A D | dnn_testing_utils.py | 127 def mock_head(testcase, hidden_units, logits_dimension, expected_logits):
158 head.logits_dimension = logits_dimension
227 def _test_logits(self, mode, hidden_units, logits_dimension, inputs,
235 logits_dimension=logits_dimension,
280 logits_dimension=1,
307 logits_dimension=3,
338 logits_dimension=3,
365 logits_dimension
[all...] |
| H A D | dnn_linear_combined.py | 162 units=head.logits_dimension,
184 units=head.logits_dimension,
|
| H A D | baseline.py | 161 logit_fn = _baseline_logit_fn_builder(head.logits_dimension, weight_column)
|
| H A D | head_test.py | 152 self.assertEqual(n_classes, head.logits_dimension)
179 self.assertEqual(n_classes, head.logits_dimension)
215 self.assertEqual(n_classes, head.logits_dimension)
244 self.assertEqual(n_classes, head.logits_dimension)
274 self.assertEqual(n_classes, head.logits_dimension)
293 self.assertEqual(n_classes, head.logits_dimension)
335 self.assertEqual(n_classes, head.logits_dimension)
1329 self.assertEqual(1, head.logits_dimension)
1355 self.assertEqual(1, head.logits_dimension)
1388 self.assertEqual(1, head.logits_dimension)
[all...] |
| H A D | head.py | 93 last_hidden_layer_out, head.logits_dimension, activation_fn=None)
138 def logits_dimension(self): member in class:_Head
141 Typically, logits is of shape `[batch_size, logits_dimension]`.
207 Consider logits of shape [D0, D1, ... DN, logits_dimension]. Then labels
278 Consider logits of shape [D0, D1, ... DN, logits_dimension]. Weights shape
280 * [D0, D1, ... DN, logits_dimension] if `allow_per_logit_weights=True`.
290 dimension, namely shape `[D0, D1, ... DN, logits_dimension]`.
299 '[D0, D1, ... DN, logits_dimension]')
353 """Checks that logits shape is [D0, D1, ... DN, logits_dimension]."""
359 message='logits shape must be [D0, D1, ... DN, logits_dimension]')
622 def logits_dimension(self): member in class:_MultiClassHeadWithSoftmaxCrossEntropyLoss
919 def logits_dimension(self): member in class:_BinaryLogisticHeadWithSigmoidCrossEntropyLoss
1238 def logits_dimension(self): member in class:_RegressionHeadWithMeanSquaredErrorLoss
[all...] |
| H A D | dnn.py | 178 units=head.logits_dimension,
|
| H A D | linear.py | 120 labels: `Tensor` of shape `[batch_size, logits_dimension]`.
157 units=head.logits_dimension, feature_columns=feature_columns)
|
| /external/tensorflow/tensorflow/contrib/estimator/python/estimator/ |
| H A D | multi_head.py | 80 `[D0, D1, ... DN, logits_dimension]`. It will split the `Tensor` along the
91 logits = logit_fn(logits_dimension=head.logits_dimension)
170 self._logits_dimension += head.logits_dimension
180 def logits_dimension(self): member in class:_MultiHead
276 [minus_ones_like_batch_shape, [head.logits_dimension]], axis=0)
279 begin_idx += head.logits_dimension
|
| H A D | head.py | 350 def logits_dimension(self): member in class:_MultiLabelHead
398 expected_labels_dimension=self.logits_dimension)
448 logits = head_lib._check_logits_final_dim(logits, self.logits_dimension) # pylint:disable=protected-access
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| H A D | gbdt_batch_test.py | 167 logits_dimension=1, features=features)
271 logits_dimension=1, features=features)
374 logits_dimension=1, features=features)
445 logits_dimension=1, features=features)
508 logits_dimension=1, features=features)
591 logits_dimension=1, features=features)
630 logits_dimension=5, features=features)
733 logits_dimension=5, features=features)
836 logits_dimension=5, features=features)
|
| H A D | gbdt_batch.py | 260 logits_dimension,
274 logits_dimension: An int, the dimension of logits.
289 self._logits_dimension = logits_dimension
299 if logits_dimension == 1:
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| H A D | sdca_estimator.py | 130 assert head.logits_dimension == 1, (
131 "SDCA only applies to logits_dimension=1.")
259 assert head.logits_dimension == 1
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| H A D | estimator_test.py | 75 logits_dimension=2)
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| H A D | head_impl.py | 113 def logits_dimension(self): member in class:GANHead
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| H A D | training_ops.cc | 218 const int64 logits_dimension = delta_updates_t->dim_size(0); variable
222 RetrieveBias(ensemble_resource, logits_dimension);
253 int64 logits_dimension) {
261 for (size_t idx = 0; idx < logits_dimension; ++idx) {
251 RetrieveBias( boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource, int64 logits_dimension) argument
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| H A D | head.py | 105 def logits_dimension(self): member in class:_TimeSeriesRegressionHead
|