Home
last modified time | relevance | path

Searched refs:hessians_t (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dstats_accumulator_ops.cc155 Tensor* hessians_t = nullptr; in SerializeScalarAccumulatorToOutput() local
158 TensorShape({num_slots}), &hessians_t)); in SerializeScalarAccumulatorToOutput()
159 auto hessians = hessians_t->vec<float>(); in SerializeScalarAccumulatorToOutput()
201 Tensor* hessians_t = nullptr; in SerializeTensorAccumulatorToOutput() local
203 hessian_shape, &hessians_t)); in SerializeTensorAccumulatorToOutput()
204 auto hessians = hessians_t->flat_outer_dims<float>(); in SerializeTensorAccumulatorToOutput()
225 const Tensor& gradients_t, const Tensor& hessians_t) { in AddToScalarAccumulator() argument
232 const auto& hessians = hessians_t.vec<float>(); in AddToScalarAccumulator()
259 const Tensor* hessians_t; in AddToScalarAccumulator() local
260 OP_REQUIRES_OK(context, context->input("hessians", &hessians_t)); in AddToScalarAccumulator()
[all …]
Dsplit_handler_ops.cc161 const Tensor* hessians_t; in Compute() local
162 OP_REQUIRES_OK(context, context->input("hessians", &hessians_t)); in Compute()
228 hessians_t, &output_partition_ids, &gains, &output_splits); in Compute()
235 hessians_t, &output_partition_ids, &gains, &output_splits); in Compute()
248 const Tensor* gradients_t, const Tensor* hessians_t, in ComputeNormalDecisionTree() argument
260 GradientStats(*gradients_t, *hessians_t, bucket_idx); in ComputeNormalDecisionTree()
270 GradientStats g(*gradients_t, *hessians_t, bucket_idx); in ComputeNormalDecisionTree()
308 const Tensor* gradients_t, const Tensor* hessians_t, in ComputeObliviousDecisionTree() argument
322 GradientStats(*gradients_t, *hessians_t, bucket_idx); in ComputeObliviousDecisionTree()
359 GradientStats g(*gradients_t, *hessians_t, idx); in ComputeObliviousDecisionTree()
[all …]
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dstats_ops.cc227 const Tensor* hessians_t; in Compute() local
228 OP_REQUIRES_OK(context, context->input("hessians", &hessians_t)); in Compute()
229 const auto hessians = hessians_t->matrix<float>(); in Compute()