| /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 | 44 hessians = array_ops.constant([0.4, 0.38, 0.26])
52 hessians=hessians,
116 hessians = array_ops.constant([[[0.4, 1], [1, 1]], [[0.38, 1], [1, 1]],
125 hessians=hessians,
156 hessians = array_ops.constant([])
164 hessians=hessians,
196 hessians = array_ops.constant([0.78, 0.4, 0.24, 0.26, 0.26])
204 hessians=hessians,
281 hessians = array_ops.constant([0.78, 0.26])
289 hessians=hessians,
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | stats_accumulator_ops.py | 93 hessians) = self.serialize()
105 saver.BaseSaverBuilder.SaveSpec(hessians, slice_spec,
114 def add(self, stamp_token, partition_ids, feature_ids, gradients, hessians): argument
116 partition_ids, feature_ids, gradients, hessians = (self._make_summary(
117 partition_ids, feature_ids, gradients, hessians))
121 [gradients], [hessians])
125 [gradients], [hessians])
127 def schedule_add(self, partition_ids, feature_ids, gradients, hessians): argument
129 partition_ids, feature_ids, gradients, hessians = (self._make_summary(
130 partition_ids, feature_ids, gradients, hessians))
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/ |
| D | ordinal_split_handler.py | 199 hessians, empty_gradients, empty_hessians, weights, argument
225 feature_ids, gradients, hessians) = dense_make_stats_update(
227 example_partition_ids, gradients, hessians, weights, empty_gradients,
234 example_partition_ids, feature_ids, gradients, hessians)
254 num_minibatches, partition_ids, bucket_ids, gradients, hessians = (
268 hessians=hessians,
339 hessians, empty_gradients, empty_hessians, weights, argument
365 feature_ids, gradients, hessians) = sparse_make_stats_update(
369 example_partition_ids, gradients, hessians, weights, empty_gradients,
377 example_partition_ids, feature_ids, gradients, hessians)
[all …]
|
| D | ordinal_split_handler_test.py | 60 hessians = array_ops.constant([0.12, 0.07, 0.2, 0.13])
89 hessians,
101 hessians,
193 hessians = array_ops.constant(
224 hessians,
236 hessians,
279 hessians = array_ops.constant(
310 hessians,
322 hessians,
363 hessians = array_ops.constant([0.12, 0.07, 0.2, 0.13])
[all …]
|
| D | categorical_split_handler.py | 85 hessians, empty_gradients, empty_hessians, weights, argument
125 filtered_hessians = array_ops.gather(hessians, example_indices)
138 hessians, mapped_partitions, array_ops.size(unique_partitions))
174 num_minibatches, partition_ids, feature_ids, gradients, hessians = (
182 hessians=hessians,
|
| D | categorical_split_handler_test.py | 58 hessians = array_ops.constant([0.12, 0.07, 0.2, 0.13])
88 hessians,
172 hessians = array_ops.constant(
179 hessians = array_ops.constant(
208 hessians,
243 hessians = array_ops.constant([0.12, 0.07, 0.2, 0.13])
273 hessians,
291 hessians = array_ops.constant([0.12, 0.07, 0.2, 0.13])
321 hessians,
|
| D | base_split_handler.py | 72 hessians, empty_gradients, empty_hessians, weights, argument
97 hessians, empty_gradients, empty_hessians, weights, argument
122 stamp_token, example_partition_ids, gradients, hessians,
|
| /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/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 510 hessians = gradients_impl.gradients(
519 squeezed_hessians = array_ops.squeeze(hessians, axis=[1])
523 hessians = array_ops.stack(hessian_list, axis=1)
534 _get_column_by_index(hessians, class_id))
549 hessians = array_ops.stack(hessian_list, axis=1)
550 squeezed_hessians = hessians
790 def _get_weights(self, hessian_shape, hessians): argument
794 weights = hessians
797 weights = math_ops.reduce_sum(hessians, axis=1)
800 weights = math_ops.trace(hessians)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients.py | 24 from tensorflow.python.ops.gradients_impl import hessians
|
| D | gradients_test.py | 592 hess = gradients.hessians(x_mat_x, x)[0]
611 hessians = gradients.hessians(xs_mats_xs, xs)
612 hessians_actual = [hess.eval() for hess in hessians]
622 gradients.hessians(x, x)
636 hess = gradients.hessians(x_square, x)[0]
655 hess = gradients.hessians(x_square, x)[0]
|
| D | gradients_impl.py | 979 def hessians(ys, function
1017 hessians = []
1041 hessians.append(_reshaped_hessian)
1042 return hessians
|
| /external/tensorflow/tensorflow/docs_src/api_guides/python/ |
| D | train.md | 40 * @{tf.hessians}
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_xent_op_test.py | 218 _ = gradients_impl.hessians(loss, [weights])
|
| /external/tensorflow/tensorflow/tools/api/golden/ |
| D | tensorflow.pbtxt | 1160 name: "hessians"
1161 …ts\', \'aggregation_method\'], varargs=None, keywords=None, defaults=[\'hessians\', \'False\', \'F…
|