Searched refs:num_minibatches (Results 1 - 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/ |
| H A D | categorical_split_handler.py | 174 num_minibatches, partition_ids, feature_ids, gradients, hessians = (
178 num_minibatches=num_minibatches,
|
| H A D | ordinal_split_handler.py | 254 num_minibatches, partition_ids, bucket_ids, gradients, hessians = (
263 num_minibatches=num_minibatches,
394 num_minibatches, partition_ids, bucket_ids, gradients, hessians = (
402 num_minibatches=num_minibatches,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| H A D | split_handler_ops_test.py | 48 num_minibatches=2,
121 num_minibatches=2,
160 num_minibatches=0,
200 num_minibatches=2,
285 num_minibatches=2,
333 num_minibatches=2,
422 num_minibatches=2,
468 num_minibatches=2,
564 num_minibatches=2,
601 num_minibatches
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| H A D | split_handler_ops.cc | 119 context->input("num_minibatches", &num_minibatches_t));
120 const int64 num_minibatches = num_minibatches_t->scalar<int64>()(); variable
121 const float normalizer_ratio = (1.0f / num_minibatches);
253 context->input("num_minibatches", &num_minibatches_t));
254 const int64 num_minibatches = num_minibatches_t->scalar<int64>()(); variable
255 const float normalizer_ratio = (1.0f / num_minibatches);
539 context->input("num_minibatches", &num_minibatches_t));
540 const int64 num_minibatches = num_minibatches_t->scalar<int64>()(); variable
541 const float normalizer_ratio = (1.0f / num_minibatches);
|
| /external/tensorflow/tensorflow/core/util/ |
| H A D | example_proto_fast_parsing.cc | 924 const size_t num_minibatches = [&] { local
944 return (serialized.size() * minibatch) / num_minibatches;
948 // num_minibatches calculation to take into account actual amount of work
953 // Maybe accept outside parameter #num_minibatches?
956 std::vector<std::vector<SparseBuffer>> sparse_buffers(num_minibatches);
957 std::vector<std::vector<SparseBuffer>> varlen_dense_buffers(num_minibatches);
958 std::vector<Status> status_of_minibatch(num_minibatches);
974 ParallelFor(ProcessMiniBatch, num_minibatches, thread_pool);
|
Completed in 752 milliseconds