Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc237 OP_REQUIRES_OK(context, context->GetAttr("logits_dimension", &logits_dim_)); in BoostedTreesUpdateEnsembleV2Op()
334 logits_dim_, &left_node_id, in Compute()
339 logits_dim_, &left_node_id, in Compute()
355 ensemble_resource->PostPruneTree(current_tree, logits_dim_); in Compute()
359 ensemble_resource->AddNewTree(kLayerByLayerTreeWeight, logits_dim_); in Compute()
382 current_tree = resource->AddNewTree(kLayerByLayerTreeWeight, logits_dim_); in UpdateGlobalAttemptsAndRetrieveGrowableTree()
425 for (int i = 0; i < logits_dim_; ++i) { in FindBestSplitsPerNode()
450 int32 logits_dim_; member in tensorflow::BoostedTreesUpdateEnsembleV2Op
Dstats_ops.cc239 OP_REQUIRES_OK(context, context->GetAttr("logits_dimension", &logits_dim_)); in BoostedTreesCalculateBestFeatureSplitOp()
258 const int32 logits_dim = logits_dim_; in Compute()
267 if (logits_dim_ > 1) { in Compute()
547 int logits_dim_; member in tensorflow::BoostedTreesCalculateBestFeatureSplitOp
562 OP_REQUIRES_OK(context, context->GetAttr("logits_dimension", &logits_dim_)); in BoostedTreesCalculateBestFeatureSplitV2()
584 const int32 logits_dim = logits_dim_; in Compute()
625 if (logits_dim_ > 1) { in Compute()
931 int logits_dim_; member in tensorflow::BoostedTreesCalculateBestFeatureSplitV2
952 OP_REQUIRES_OK(context, context->GetAttr("logits_dimension", &logits_dim_)); in BoostedTreesSparseCalculateBestFeatureSplitOp()
1120 Eigen::VectorXf unused(logits_dim_); in process_node()
[all …]