Searched refs:pruning_mode_ (Results 1 – 1 of 1) sorted by relevance
52 pruning_mode_ = static_cast<PruningMode>(pruning_index); in BoostedTreesUpdateEnsembleOp()126 if (pruning_mode_ == kPrePruning) { in Compute()161 if (pruning_mode_ == kPostPruning) { in Compute()238 PruningMode pruning_mode_; member in tensorflow::BoostedTreesUpdateEnsembleOp