Searched refs:num_handlers_ (Results 1 – 2 of 2) sorted by relevance
145 context->GetAttr("num_all_handlers", &num_handlers_)); in TreeEnsembleUsedHandlersOp()167 context, context->allocate_output("used_handlers_mask", {num_handlers_}, in Compute()178 for (int64 i = 0; i < num_handlers_; ++i) { in Compute()192 int64 num_handlers_; member in tensorflow::boosted_trees::TreeEnsembleUsedHandlersOp
297 OP_REQUIRES_OK(context, context->GetAttr("num_handlers", &num_handlers_)); in GrowTreeEnsembleOp()447 for (int64 handler_id = 0; handler_id < num_handlers_; ++handler_id) { in FindBestSplitsPerPartitionNormal()482 for (int64 handler_id = 0; handler_id < num_handlers_; ++handler_id) { in FindBestSplitOblivious()890 int64 num_handlers_; member in tensorflow::boosted_trees::GrowTreeEnsembleOp