Searched refs:pruning_index (Results 1 – 1 of 1) sorted by relevance
50 int32 pruning_index; in BoostedTreesUpdateEnsembleOp() local51 OP_REQUIRES_OK(context, context->GetAttr("pruning_mode", &pruning_index)); in BoostedTreesUpdateEnsembleOp()52 pruning_mode_ = static_cast<PruningMode>(pruning_index); in BoostedTreesUpdateEnsembleOp()