Home
last modified time | relevance | path

Searched defs:hessians (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dstats_accumulator_ops.py73 gradients, hessians): argument
196 def add(self, stamp_token, partition_ids, feature_ids, gradients, hessians): argument
209 def schedule_add(self, partition_ids, feature_ids, gradients, hessians): argument
230 def _make_summary(self, partition_ids, feature_ids, gradients, hessians): argument
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/
Dbase_split_handler.py78 hessians, empty_gradients, empty_hessians, weights, argument
103 hessians, empty_gradients, empty_hessians, weights, argument
Dordinal_split_handler.py224 hessians, empty_gradients, empty_hessians, weights, argument
401 hessians, empty_gradients, empty_hessians, weights, argument
608 hessians, weights, empty_gradients, empty_hessians): argument
657 hessians, weights, empty_gradients, empty_hessians): argument
Dcategorical_split_handler.py94 hessians, empty_gradients, empty_hessians, weights, argument
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py619 def update_stats(self, loss, predictions_dict, gradients=None, hessians=None): argument
1180 hessians=None): argument
1209 def _get_weights(self, hessian_shape, hessians): argument
1301 hessians=None): argument
/external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/
Dstats_accumulator_ops_test.py453 def _AccumulatorResultToDict(partition, feature, grads, hessians): argument
/external/tensorflow/tensorflow/python/ops/
Dgradients_impl.py333 def hessians(ys, function
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dstats_accumulator_ops.cc159 auto hessians = hessians_t->vec<float>(); in SerializeScalarAccumulatorToOutput() local
204 auto hessians = hessians_t->flat_outer_dims<float>(); in SerializeTensorAccumulatorToOutput() local
232 const auto& hessians = hessians_t.vec<float>(); in AddToScalarAccumulator() local
279 const auto& hessians = hessians_t.flat_outer_dims<float>(); in AddToTensorAccumulator() local
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dstats_ops.cc229 const auto hessians = hessians_t->matrix<float>(); in Compute() local