| /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
36 empty_hess_shape = [1] + hessian_shape.as_list()
64 hessian_shape = tensor_shape.scalar()
75 hessian_shape=hessian_shape,
81 gradient_shape, hessian_shape)
187 hessian_shape = tensor_shape.scalar()
198 hessian_shape=hessian_shape,
205 gradient_shape, hessian_shape)
298 hessian_shape = tensor_shape.scalar()
309 hessian_shape=hessian_shape,
[all …]
|
| D | ordinal_split_handler_test.py | 38 def get_empty_tensors(gradient_shape, hessian_shape): argument
39 empty_hess_shape = [1] + hessian_shape.as_list()
67 hessian_shape = tensor_shape.scalar()
79 hessian_shape=hessian_shape,
84 gradient_shape, hessian_shape)
201 hessian_shape = tensor_shape.scalar()
213 hessian_shape=hessian_shape,
219 gradient_shape, hessian_shape)
337 hessian_shape = tensor_shape.scalar()
349 hessian_shape=hessian_shape,
[all …]
|
| D | ordinal_split_handler.py | 101 hessian_shape, argument
135 hessian_shape=hessian_shape,
141 hessian_shape,
171 hessian_shape, argument
211 hessian_shape=hessian_shape,
354 hessian_shape, argument
390 hessian_shape=hessian_shape,
|
| D | categorical_split_handler.py | 46 hessian_shape, argument
81 hessian_shape=hessian_shape,
88 hessian_shape,
|
| D | base_split_handler.py | 40 hessian_shape, argument
68 self._hessian_shape = hessian_shape
|
| /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
125 hessian_shape); in TEST()
153 const auto& hessian_shape = TensorShape({1, kNumClasses, kNumClasses}); in TEST() local
156 hessian_shape); in TEST()
162 const auto& hessian_shape = TensorShape({1, kNumClasses}); in TEST() local
164 h = AsTensor<float>({3, 4, 8, 9}, hessian_shape); in TEST()
189 const auto& hessian_shape = TensorShape({1, kNumClasses, kNumClasses}); in TEST() local
192 hessian_shape); in TEST()
198 const auto& hessian_shape = TensorShape({1, kNumClasses}); in TEST() local
200 h = AsTensor<float>({3, 0, 8, 0}, hessian_shape); in TEST()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | stats_accumulator_ops.cc | 87 const TensorShape& hessian_shape) in StatsAccumulatorResource() argument
89 hessian_shape_(hessian_shape), in StatsAccumulatorResource()
96 TensorShapeUtils::IsScalar(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
199 int64 num_hessian_elements = hessian_shape.num_elements(); in SerializeTensorAccumulatorToOutput()
200 hessian_shape.InsertDim(0, num_slots); in SerializeTensorAccumulatorToOutput()
203 hessian_shape, &hessians_t)); in SerializeTensorAccumulatorToOutput()
292 context, hessians_shape == accumulator_resource->hessian_shape(), in AddToTensorAccumulator()
295 accumulator_resource->hessian_shape().DebugString()))); in AddToTensorAccumulator()
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| D | stats_accumulator_ops_test.py | 36 hessian_shape=tensor_shape.scalar())
64 hessian_shape=tensor_shape.scalar())
93 hessian_shape=tensor_shape.scalar())
125 hessian_shape=tensor_shape.scalar())
166 hessian_shape=tensor_shape.scalar())
203 hessian_shape=tensor_shape.scalar())
225 hessian_shape=tensor_shape.TensorShape([2, 2]))
263 hessian_shape=tensor_shape.TensorShape([2, 2]))
301 hessian_shape=tensor_shape.TensorShape([2, 2]))
338 hessian_shape=tensor_shape.TensorShape([2, 2]))
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | stats_accumulator_ops.py | 117 hessian_shape, argument
134 self._hessian_shape = hessian_shape
138 hessian_shape == tensor_shape.scalar()):
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 745 hessian_shape=self._hessian_shape,
772 hessian_shape=self._hessian_shape,
795 hessian_shape=self._hessian_shape,
832 hessian_shape=self._hessian_shape,
838 hessian_shape=tensor_shape.scalar(),
1209 def _get_weights(self, hessian_shape, hessians): argument
1211 if hessian_shape == tensor_shape.scalar():
1214 elif len(hessian_shape.dims) == 1:
|