Searched refs:dropout_seed (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| D | training_ops_test.py | 350 dropout_seed=123,
464 dropout_seed=123,
616 dropout_seed=123,
802 dropout_seed=123,
940 dropout_seed=123,
1012 dropout_seed=123,
1115 dropout_seed=123,
1198 dropout_seed=123,
1266 dropout_seed=123,
1347 dropout_seed=123,
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | training_ops.cc | 362 const uint64 dropout_seed = seed_t->scalar<int64>()(); in Compute() local
405 dropout_seed, max_tree_depth, in Compute()
564 const uint64 dropout_seed) { in UpdateTreeWeightsIfDropout() argument
588 dropout_seed, dropout_config_, trees_not_to_drop, weights, in UpdateTreeWeightsIfDropout()
615 const float learning_rate, const uint64 dropout_seed, in UpdateAndRetrieveGrowableTree() argument
641 UpdateTreeWeightsIfDropout(ensemble_resource, dropout_seed); in UpdateAndRetrieveGrowableTree()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 1005 dropout_seed = predictions_dict[NUM_TREES_ATTEMPTED]
1023 dropout_seed, class_id),
1032 dropout_seed, class_id): argument
1117 dropout_seed=dropout_seed,
1133 dropout_seed=dropout_seed,
|