Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dmodel_ops.cc176 std::vector<int64> used_handlers = ensemble_resource->GetUsedHandlers(); in Compute() local
177 output_num_used_handlers_t->scalar<int64>()() = used_handlers.size(); in Compute()
179 if (handler_idx >= used_handlers.size() || in Compute()
180 used_handlers[handler_idx] > i) { in Compute()
183 OP_REQUIRES(context, used_handlers[handler_idx] == i, in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py88 used_handlers, argument
112 result[NUM_USED_HANDLERS] = used_handlers.num_used_handlers
113 result[USED_HANDLERS_MASK] = used_handlers.used_handlers_mask
481 used_handlers = model_ops.tree_ensemble_used_handlers(
539 ensemble_stats, used_handlers, leaf_index)
900 used_handlers = array_ops.reshape(predictions_dict[USED_HANDLERS_MASK],
902 used_handlers = array_ops.concat([used_handlers, used_handlers], axis=1)
903 return math_ops.logical_and(used_handlers, active_handlers)