Home
last modified time | relevance | path

Searched refs:min_node_weight_t (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dstats_ops.cc60 const Tensor* min_node_weight_t; in Compute() local
62 context->input("min_node_weight", &min_node_weight_t)); in Compute()
63 const auto min_node_weight = min_node_weight_t->scalar<float>()(); in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dsplit_handler_ops.cc61 const Tensor* min_node_weight_t; in SplitBuilderState() local
63 context->input("min_node_weight", &min_node_weight_t)); in SplitBuilderState()
91 min_node_weight_ = min_node_weight_t->scalar<float>()(); in SplitBuilderState()