Searched refs:max_elements_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/resources/ |
| D | quantile_stream_resource.h | 41 max_elements_(max_elements), in QuantileStreamResource()
71 stream_ = QuantileStream(epsilon_, max_elements_); in Reset()
104 int64 max_elements_; variable
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/ |
| D | quantile_stream_resource.h | 39 max_elements_(max_elements) { in BoostedTreesQuantileStreamResource()
89 int64 max_elements_; variable
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | quantile_ops.cc | 100 OP_REQUIRES_OK(context, context->GetAttr(kMaxElementsName, &max_elements_)); in BoostedTreesCreateQuantileStreamResourceOp()
120 new QuantileStreamResource(epsilon, max_elements_, num_streams); in Compute()
130 int64 max_elements_; member in tensorflow::BoostedTreesCreateQuantileStreamResourceOp
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 236 OP_REQUIRES_OK(context, context->GetAttr(kMaxElementsName, &max_elements_)); in CreateQuantileAccumulatorOp()
252 max_elements_, generate_quantiles_, in Compute()
265 int64 max_elements_; member in tensorflow::CreateQuantileAccumulatorOp
|