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 270 milliseconds