Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dstats_accumulator_ops.cc149 Tensor* gradients_t = nullptr; in SerializeScalarAccumulatorToOutput() local
152 "output_gradients", TensorShape({num_slots}), &gradients_t)); in SerializeScalarAccumulatorToOutput()
153 auto gradients = gradients_t->vec<float>(); in SerializeScalarAccumulatorToOutput()
192 Tensor* gradients_t = nullptr; in SerializeTensorAccumulatorToOutput() local
195 &gradients_t)); in SerializeTensorAccumulatorToOutput()
196 auto gradients = gradients_t->flat_outer_dims<float>(); in SerializeTensorAccumulatorToOutput()
225 const Tensor& gradients_t, const Tensor& hessians_t) { in AddToScalarAccumulator() argument
231 const auto& gradients = gradients_t.vec<float>(); in AddToScalarAccumulator()
257 const Tensor* gradients_t; in AddToScalarAccumulator() local
258 OP_REQUIRES_OK(context, context->input("gradients", &gradients_t)); in AddToScalarAccumulator()
[all …]
Dsplit_handler_ops.cc158 const Tensor* gradients_t; in Compute() local
159 OP_REQUIRES_OK(context, context->input("gradients", &gradients_t)); in Compute()
227 bucket_boundaries, partition_ids, bucket_ids, gradients_t, in Compute()
234 bucket_boundaries, partition_ids, bucket_ids, gradients_t, 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.cc223 const Tensor* gradients_t; in Compute() local
224 OP_REQUIRES_OK(context, context->input("gradients", &gradients_t)); in Compute()
225 const auto gradients = gradients_t->matrix<float>(); in Compute()