Home
last modified time | relevance | path

Searched refs:num_minibatches (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/
Dordinal_split_handler.py307 num_minibatches, partition_ids, bucket_ids, gradients, hessians = (
311 num_minibatches, partition_ids, bucket_ids, gradients, hessians = (
315 num_minibatches = control_flow_ops.cond(
318 lambda: num_minibatches)
324 num_minibatches=num_minibatches,
485 num_minibatches, partition_ids, bucket_ids, gradients, hessians = (
489 num_minibatches, partition_ids, bucket_ids, gradients, hessians = (
492 num_minibatches = control_flow_ops.cond(
495 lambda: num_minibatches)
501 num_minibatches=num_minibatches,
Dcategorical_split_handler.py190 num_minibatches, partition_ids, feature_ids, gradients, hessians = (
194 num_minibatches = control_flow_ops.cond(
197 lambda: num_minibatches)
200 num_minibatches=num_minibatches,
/external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/
Dsplit_handler_ops_test.py50 num_minibatches=2,
124 num_minibatches=2,
164 num_minibatches=0,
205 num_minibatches=2,
290 num_minibatches=2,
338 num_minibatches=2,
442 num_minibatches=2,
486 num_minibatches=2,
532 num_minibatches=2,
629 num_minibatches=2,
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dsplit_handler_ops.cc142 const int64 num_minibatches = num_minibatches_t->scalar<int64>()(); in Compute() local
143 const float normalizer_ratio = (1.0f / num_minibatches); in Compute()
429 const int64 num_minibatches = num_minibatches_t->scalar<int64>()(); in Compute() local
430 const float normalizer_ratio = (1.0f / num_minibatches); in Compute()
709 const int64 num_minibatches = num_minibatches_t->scalar<int64>()(); in Compute() local
710 const float normalizer_ratio = (1.0f / num_minibatches); in Compute()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc1015 const size_t num_minibatches = [&] { in FastParseExample() local
1035 return (serialized.size() * minibatch) / num_minibatches; in FastParseExample()
1047 std::vector<std::vector<SparseBuffer>> sparse_buffers(num_minibatches); in FastParseExample()
1048 std::vector<std::vector<SparseBuffer>> varlen_dense_buffers(num_minibatches); in FastParseExample()
1049 std::vector<Status> status_of_minibatch(num_minibatches); in FastParseExample()
1069 ParallelFor(ProcessMiniBatch, num_minibatches, thread_pool); in FastParseExample()