Searched defs:hessian_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/stats/ |
| D | node-stats_test.cc | 123 const auto& hessian_shape = TensorShape({1, kNumClasses, kNumClasses}); in TEST() local
153 const auto& hessian_shape = TensorShape({1, kNumClasses, kNumClasses}); in TEST() local
162 const auto& hessian_shape = TensorShape({1, kNumClasses}); in TEST() local
189 const auto& hessian_shape = TensorShape({1, kNumClasses, kNumClasses}); in TEST() local
198 const auto& hessian_shape = TensorShape({1, kNumClasses}); in TEST() local
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/ |
| D | categorical_split_handler_test.py | 35 def get_empty_tensors(gradient_shape, hessian_shape): argument
|
| D | ordinal_split_handler_test.py | 38 def get_empty_tensors(gradient_shape, hessian_shape): argument
|
| D | categorical_split_handler.py | 46 hessian_shape, argument
|
| D | base_split_handler.py | 40 hessian_shape, argument
|
| D | ordinal_split_handler.py | 101 hessian_shape, argument
171 hessian_shape, argument
354 hessian_shape, argument
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | stats_accumulator_ops.cc | 87 const TensorShape& hessian_shape) in StatsAccumulatorResource()
115 const TensorShape& hessian_shape() const { return hessian_shape_; } in hessian_shape() function in tensorflow::boosted_trees::__anon2081c9750111::StatsAccumulatorResource
198 TensorShape hessian_shape = accumulator_resource.hessian_shape(); in SerializeTensorAccumulatorToOutput() local
365 TensorShape hessian_shape = TensorShape({}); in Compute() local
399 TensorShape hessian_shape = TensorShape(hessian_shape_t->vec<int64>()); in Compute() local
753 TensorShape hessian_shape = TensorShape({}); in Compute() local
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | stats_accumulator_ops.py | 117 hessian_shape, argument
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 1209 def _get_weights(self, hessian_shape, hessians): argument
|