Searched refs:PruningMode (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | training_ops_test.py | 69 pruning_mode=boosted_trees_ops.PruningMode.NO_PRUNING,
260 pruning_mode=boosted_trees_ops.PruningMode.NO_PRUNING,
440 pruning_mode=boosted_trees_ops.PruningMode.NO_PRUNING,
596 pruning_mode=boosted_trees_ops.PruningMode.PRE_PRUNING,
735 pruning_mode=boosted_trees_ops.PruningMode.NO_PRUNING,
865 pruning_mode=boosted_trees_ops.PruningMode.PRE_PRUNING,
959 pruning_mode=boosted_trees_ops.PruningMode.POST_PRUNING,
1035 pruning_mode=boosted_trees_ops.PruningMode.POST_PRUNING,
1143 pruning_mode=boosted_trees_ops.PruningMode.POST_PRUNING,
1294 pruning_mode=boosted_trees_ops.PruningMode.POST_PRUNING,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | training_ops.cc | 40 enum PruningMode { kNoPruning = 0, kPrePruning = 1, kPostPruning = 2 }; enum
52 pruning_mode_ = static_cast<PruningMode>(pruning_index); in BoostedTreesUpdateEnsembleOp()
238 PruningMode pruning_mode_;
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/proto/ |
| D | learner.proto | 92 enum PruningMode { enum
133 PruningMode pruning_mode = 8;
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | boosted_trees_ops.py | 49 class PruningMode(object): class
|