Home
last modified time | relevance | path

Searched refs:max_elements (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/
Dweighted_quantiles_stream_test.cc57 void GenerateFixedUniformSummary(int32 worker_id, int64 max_elements, in GenerateFixedUniformSummary() argument
59 for (int64 i = 0; i < max_elements; ++i) { in GenerateFixedUniformSummary()
60 const double x = static_cast<double>(i) / max_elements; in GenerateFixedUniformSummary()
67 void GenerateFixedNonUniformSummary(int32 worker_id, int64 max_elements, in GenerateFixedNonUniformSummary() argument
69 for (int64 i = 0; i < max_elements; ++i) { in GenerateFixedNonUniformSummary()
70 const double x = static_cast<double>(i) / max_elements; in GenerateFixedNonUniformSummary()
77 void GenerateRandUniformFixedWeightsSummary(int32 worker_id, int64 max_elements, in GenerateRandUniformFixedWeightsSummary() argument
83 for (int64 i = 0; i < max_elements; ++i) { in GenerateRandUniformFixedWeightsSummary()
91 void GenerateRandUniformRandWeightsSummary(int32 worker_id, int64 max_elements, in GenerateRandUniformRandWeightsSummary() argument
97 for (int64 i = 0; i < max_elements; ++i) { in GenerateRandUniformRandWeightsSummary()
[all …]
Dweighted_quantiles_stream.h72 explicit WeightedQuantilesStream(double eps, int64 max_elements) in WeightedQuantilesStream() argument
77 std::tie(max_levels_, block_size_) = GetQuantileSpecs(eps, max_elements); in WeightedQuantilesStream()
78 buffer_ = Buffer(block_size_, max_elements); in WeightedQuantilesStream()
210 int64 max_elements);
296 double eps, int64 max_elements) { in GetQuantileSpecs() argument
300 QCHECK_GT(max_elements, 0); in GetQuantileSpecs()
305 block_size = std::max(max_elements, int64{2}); in GetQuantileSpecs()
319 (1LL << max_level) * block_size < max_elements; ++max_level) { in GetQuantileSpecs()
Dquantile_stream_resource.h34 const int64 max_elements, in BoostedTreesQuantileStreamResource() argument
39 max_elements_(max_elements) { in BoostedTreesQuantileStreamResource()
43 streams_.push_back(QuantileStream(epsilon, max_elements)); in BoostedTreesQuantileStreamResource()
Dweighted_quantiles_buffer.h54 explicit WeightedQuantilesBuffer(int64 block_size, int64 max_elements) in WeightedQuantilesBuffer() argument
55 : max_size_(std::min(block_size << 1, max_elements)) { in WeightedQuantilesBuffer()
57 << ", " << max_elements << ")"; in WeightedQuantilesBuffer()
/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/
Dquantile_ops_test.py42 def create_resource(self, name, eps, max_elements, num_streams=1): argument
48 max_elements=max_elements,
84 self.max_elements = 1 << 16
90 self.max_elements, 2)
114 self.max_elements, 2)
149 self.max_elements)
151 self.max_elements)
/external/webrtc/rtc_base/
Dbuffer.h245 size_t SetData(size_t max_elements, F&& setter) { in SetData() argument
249 const size_t written = AppendData<U>(max_elements, std::forward<F>(setter)); in SetData()
306 size_t AppendData(size_t max_elements, F&& setter) { in AppendData() argument
309 SetSize(old_size + max_elements); in AppendData()
311 size_t written_elements = setter(rtc::ArrayView<U>(base_ptr, max_elements)); in AppendData()
313 RTC_CHECK_LE(written_elements, max_elements); in AppendData()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DBoostedTreesCreateQuantileStreamResource.pbtxt16 name: "max_elements"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DBoostedTreesCreateQuantileStreamResource.pbtxt16 name: "max_elements"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BoostedTreesCreateQuantileStreamResource.pbtxt23 name: "max_elements"
/external/mesa3d/src/compiler/glsl/
Dopt_dead_builtin_varyings.cpp356 int max_elements, unsigned start_location, in prepare_array() argument
362 for (int i = max_elements-1; i >= 0; i--) { in prepare_array()
/external/tensorflow/tensorflow/python/ops/
Dboosted_trees_ops.py110 max_elements=None): argument
/external/XNNPACK/test/
Daverage-pooling-nhwc.cc17 static std::pair<size_t, size_t> SmallPoolSize(size_t max_elements) { in SmallPoolSize() argument
18 const size_t small_side = size_t(std::floor(std::sqrt(double(max_elements)))); in SmallPoolSize()
20 if (small_side * large_side < max_elements) { in SmallPoolSize()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc1598 auto max_elements = ops::Const<int32>(scope, {10}, {}); in TEST_F() local
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt565 …[\'quantile_stream_resource_handle\', \'epsilon\', \'num_streams\', \'max_elements\', \'name\'], v…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt565 …[\'quantile_stream_resource_handle\', \'epsilon\', \'num_streams\', \'max_elements\', \'name\'], v…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt5772 name: "max_elements"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt11728 name: "max_elements"