| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| D | stats_accumulator_ops_test.py | 43 hessians=[0.2, 0.4])
47 num_updates, partition, bucket_ids, grads, hessians = accumulator.flush(
49 num_updates, partition, bucket_ids, grads, hessians = sess.run(
50 [num_updates, partition, bucket_ids, grads, hessians])
52 result = _AccumulatorResultToDict(partition, bucket_ids, grads, hessians)
71 hessians=[0.2, 0.4, -9])
75 num_updates, partition, bucket_ids, grads, hessians = accumulator.flush(
77 num_updates, partition, bucket_ids, grads, hessians = sess.run(
78 [num_updates, partition, bucket_ids, grads, hessians])
80 result = _AccumulatorResultToDict(partition, bucket_ids, grads, hessians)
[all …]
|
| D | split_handler_ops_test.py | 46 hessians = array_ops.constant([0.4, 0.38, 0.26])
54 hessians=hessians,
119 hessians = array_ops.constant([[[0.4, 1], [1, 1]], [[0.38, 1], [1, 1]],
128 hessians=hessians,
160 hessians = array_ops.constant([])
168 hessians=hessians,
201 hessians = array_ops.constant([0.78, 0.4, 0.24, 0.26, 0.26])
209 hessians=hessians,
286 hessians = array_ops.constant([0.78, 0.26])
294 hessians=hessians,
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | stats_accumulator_ops.py | 45 hessians) = self.serialize()
57 saver.BaseSaverBuilder.SaveSpec(hessians, slice_spec,
73 gradients, hessians): argument
78 feature_ids, gradients, hessians)
82 feature_ids, gradients, hessians)
102 hessians=restored_tensors[5])
196 def add(self, stamp_token, partition_ids, feature_ids, gradients, hessians): argument
198 partition_ids, feature_ids, gradients, hessians = (self._make_summary(
199 partition_ids, feature_ids, gradients, hessians))
203 [gradients], [hessians])
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/ |
| D | ordinal_split_handler.py | 224 hessians, empty_gradients, empty_hessians, weights, argument
250 feature_ids, gradients, hessians) = dense_make_stats_update(
252 example_partition_ids, gradients, hessians, weights, empty_gradients,
259 example_partition_ids, feature_ids, gradients, hessians)
307 num_minibatches, partition_ids, bucket_ids, gradients, hessians = (
311 num_minibatches, partition_ids, bucket_ids, gradients, hessians = (
329 hessians=hessians,
401 hessians, empty_gradients, empty_hessians, weights, argument
427 hessians) = sparse_make_stats_update(
431 example_partition_ids, gradients, hessians, weights, empty_gradients,
[all …]
|
| D | ordinal_split_handler_test.py | 62 hessians = array_ops.constant([0.12, 0.07, 0.2, 0.13])
91 hessians,
105 hessians,
196 hessians = array_ops.constant([0.12, 0.07, 0.2, 0.13])
226 hessians,
243 hessians,
332 hessians = array_ops.constant([0.12, 0.07, 0.2, 0.13])
362 hessians,
376 hessians,
385 hessians,
[all …]
|
| D | categorical_split_handler_test.py | 58 hessians = array_ops.constant([0.12, 0.07, 0.2, 0.13])
88 hessians,
97 hessians,
181 hessians = array_ops.constant([0.12, 0.07, 0.2, 0.13])
212 hessians,
221 hessians,
292 hessians = array_ops.constant([0.12, 0.07, 0.2, 0.13])
323 hessians,
332 hessians,
416 hessians = array_ops.constant(
[all …]
|
| D | categorical_split_handler.py | 94 hessians, empty_gradients, empty_hessians, weights, argument
134 filtered_hessians = array_ops.gather(hessians, example_indices)
147 hessians64 = math_ops.cast(hessians, dtypes.float64)
190 num_minibatches, partition_ids, feature_ids, gradients, hessians = (
204 hessians=hessians,
|
| D | base_split_handler.py | 78 hessians, empty_gradients, empty_hessians, weights, argument
103 hessians, empty_gradients, empty_hessians, weights, argument
128 stamp_token, example_partition_ids, gradients, hessians,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_impl.py | 333 def hessians(ys, function
371 hessians = []
395 hessians.append(_reshaped_hessian)
396 return hessians
405 return hessians(ys, xs, name=name, gate_gradients=gate_gradients,
409 HessiansV2.__doc__ = hessians.__doc__
|
| D | gradients.py | 27 from tensorflow.python.ops.gradients_impl import hessians
|
| D | gradients_test.py | 663 hess = gradients.hessians(x_mat_x, x)[0]
683 hessians = gradients.hessians(xs_mats_xs, xs)
684 hessians_actual = [hess.eval() for hess in hessians]
695 gradients.hessians(x, x)
710 hess = gradients.hessians(x_square, x)[0]
730 hess = gradients.hessians(x_square, x)[0]
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 338 hessians=[[5.], [6.], [7.], [8.]],
350 hessians = [[.2], [.3], [.4], [.5], [.06], [.07], [.08], [.09]]
355 node_ids, gradients, hessians, bucketized_features, max_splits,
372 hessians = [[.2], [.3], [.4], [.5], [.06], [.07], [.08], [.09]]
378 node_ids, gradients, hessians, bucketized_features, max_splits,
403 hessians = constant_op.constant(
409 node_ids, gradients, hessians, [bucketized_features], max_splits,
|
| D | training_ops_test.py | 153 hessians = np.array([[24.]], dtype=np.float32)
159 mean_hessians=hessians,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesMakeStatsSummary.pbtxt | 17 name: "hessians"
19 float32; Rank 2 Tensor (shape=[#examples, 1]) for hessians.
31 … node and bucket. The first index of 4th dimension refers to gradients, and the second to hessians.
54 The summary stats contains gradients and hessians accumulated into the corresponding node and bucke…
|
| D | api_def_BoostedTreesCenterBias.pbtxt | 19 A tensor with shape=[logits_dimension] mean of hessians for a first node.
|
| D | api_def_BoostedTreesCalculateBestGainsPerFeature.pbtxt | 37 mininum avg of hessians in a node before required for the node to be considered for splitting.
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 619 def update_stats(self, loss, predictions_dict, gradients=None, hessians=None): argument
669 if hessians is None:
670 hessians = gradients_impl.gradients(
679 squeezed_hessians = array_ops.squeeze(hessians, axis=[1])
681 if hessians is not None:
685 hessians = array_ops.stack(hessian_list, axis=1)
691 _get_column_by_index(hessians, class_id))
693 if hessians is not None:
703 hessians = array_ops.stack(hessian_list, axis=1)
704 squeezed_hessians = hessians
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | stats_accumulator_ops.cc | 159 auto hessians = hessians_t->vec<float>(); in SerializeScalarAccumulatorToOutput() local
168 hessians(i) = iter.second.second; in SerializeScalarAccumulatorToOutput()
204 auto hessians = hessians_t->flat_outer_dims<float>(); in SerializeTensorAccumulatorToOutput() local
216 hessians(i, j) = iter.second.second[j]; in SerializeTensorAccumulatorToOutput()
232 const auto& hessians = hessians_t.vec<float>(); in AddToScalarAccumulator() local
243 itr->second.second += hessians(i); in AddToScalarAccumulator()
245 (*stats_map)[key] = {gradients(i), hessians(i)}; in AddToScalarAccumulator()
279 const auto& hessians = hessians_t.flat_outer_dims<float>(); in AddToTensorAccumulator() local
311 new_hessians[j] = hessians(i, j); in AddToTensorAccumulator()
321 stored_hessians[j] += hessians(i, j); in AddToTensorAccumulator()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 229 const auto hessians = hessians_t->matrix<float>(); in Compute() local
253 temp_stats_double(feature_idx, node, bucket, 1) += hessians(i, 0); in Compute()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_xent_op_test.py | 224 _ = gradients_impl.hessians(loss, [weights])
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 664 name: "hessians"
665 …n_method\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'None\', \'hessians\'], "
|
| D | tensorflow.raw_ops.pbtxt | 469 …argspec: "args=[\'node_ids\', \'gradients\', \'hessians\', \'bucketized_features_list\', \'max_spl…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 1376 name: "hessians"
1377 …ts\', \'aggregation_method\'], varargs=None, keywords=None, defaults=[\'hessians\', \'False\', \'F…
|
| D | tensorflow.raw_ops.pbtxt | 469 …argspec: "args=[\'node_ids\', \'gradients\', \'hessians\', \'bucketized_features_list\', \'max_spl…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 3553 func BoostedTreesMakeStatsSummary(scope *Scope, node_ids tf.Output, gradients tf.Output, hessians t…
3561 node_ids, gradients, hessians, tf.OutputList(bucketized_features_list),
|