HomeSort by relevance Sort by last modified time
    Searched full:bucket_ids (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/
split_handler_ops_test.py 41 bucket_ids = array_ops.constant(
50 bucket_ids=bucket_ids,
113 bucket_ids = array_ops.constant(
123 bucket_ids=bucket_ids,
154 bucket_ids = array_ops.constant([[]], dtype=dtypes.int64)
162 bucket_ids=bucket_ids,
191 bucket_ids = array_ops.constant([-1, 0, 1, -1, 1], dtype=dtypes.int64
    [all...]
stats_accumulator_ops_test.py 47 num_updates, partition, bucket_ids, grads, hessians = accumulator.flush(
49 num_updates, partition, bucket_ids, grads, hessians = sess.run(
50 [num_updates, partition, bucket_ids, grads, hessians])
52 result = _AccumulatorResultToDict(partition, bucket_ids, grads, hessians)
75 num_updates, partition, bucket_ids, grads, hessians = accumulator.flush(
77 num_updates, partition, bucket_ids, grads, hessians = sess.run(
78 [num_updates, partition, bucket_ids, grads, hessians])
80 result = _AccumulatorResultToDict(partition, bucket_ids, grads, hessians)
  /external/tensorflow/tensorflow/contrib/boosted_trees/ops/
split_handler_ops.cc 34 .Input("bucket_ids: int64")
72 bucket_ids: A rank 2 tensor of buckets IDs and dimensions.
93 .Input("bucket_ids: int64")
132 bucket_ids: A rank 2 tensor of buckets IDs and dimensions.
  /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/
ordinal_split_handler.py 254 num_minibatches, partition_ids, bucket_ids, gradients, hessians = (
266 bucket_ids=bucket_ids,
394 num_minibatches, partition_ids, bucket_ids, gradients, hessians = (
405 bucket_ids=bucket_ids,
504 bucket_ids = array_ops.concat([bias_feature_ids, quantized_feature], 0)
506 return partition_ids, bucket_ids, filtered_gradients, filtered_hessians
  /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
split_handler_ops.cc 133 OP_REQUIRES_OK(context, context->input("bucket_ids", &bucket_ids_t));
134 const auto& bucket_ids = bucket_ids_t->matrix<int64>(); variable
225 bucket_boundaries(bucket_ids(best_bucket_idx, 0)));
267 OP_REQUIRES_OK(context, context->input("bucket_ids", &bucket_ids_t));

Completed in 231 milliseconds