Searched refs:num_probes (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/contrib/nearest_neighbor/kernels/ |
| D | hyperplane_lsh_probes.cc | 87 int num_probes = num_probes_tensor.scalar<int32>()(); in Compute() local
88 OP_REQUIRES(context, num_probes >= 1, in Compute()
109 TensorShape output_shape({batch_size, num_probes}); in Compute()
120 if (num_probes > num_tables) { in Compute()
122 110 * num_hyperplanes_per_table * (num_probes - num_tables); in Compute()
131 num_probes); in Compute()
132 for (int ii = 0; ii < num_probes; ++ii) { in Compute()
|
| D | hyperplane_lsh_probes_test.cc | 115 int num_probes = (1 << dim) * num_tables; in TEST() local
116 multiprobe.SetupProbing(hash_vector, num_probes); in TEST()
119 for (int ii = 0; ii < num_probes; ++ii) { in TEST()
|
| D | hyperplane_lsh_probes.h | 69 void SetupProbing(const Vector& hash_vector, int_fast64_t num_probes) { in SetupProbing() argument
72 num_probes_ = num_probes; in SetupProbing()
|
| /external/tensorflow/tensorflow/core/lib/gtl/ |
| D | flatrep.h | 147 uint32 num_probes = 1; // Needed for quadratic probing in Find() local
157 index = NextIndex(index, num_probes); in Find()
158 num_probes++; in Find()
172 uint32 num_probes = 1; // Needed for quadratic probing in FindOrInsert() local
198 index = NextIndex(index, num_probes); in FindOrInsert()
199 num_probes++; in FindOrInsert()
324 uint32 num_probes = 1; // Needed for quadratic probing in FreshInsert() local
335 index = NextIndex(index, num_probes); in FreshInsert()
336 num_probes++; in FreshInsert()
340 inline size_t NextIndex(size_t i, uint32 num_probes) const { in NextIndex() argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/nearest_neighbor/python/ops/ |
| D | nearest_neighbor_ops.py | 32 num_probes, argument
61 num_probes,
|
| /external/tensorflow/tensorflow/contrib/nearest_neighbor/python/kernel_tests/ |
| D | hyperplane_lsh_probes_test.py | 40 num_probes = 4
44 num_probes)
|
| /external/bcc/tests/cc/ |
| D | test_usdt_probes.cc | 40 REQUIRE(ctx.num_probes() >= 1);
156 REQUIRE(ctx.num_probes() > 10);
157 mri_probe_count = ctx.num_probes();
230 REQUIRE(ctx.num_probes() >= mri_probe_count);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_op.cc | 423 int64 num_probes = 0; in Find() local
440 ++num_probes; in Find()
442 (bucket_index + num_probes) & bit_mask; // quadratic probing in Find()
443 if (num_probes >= num_buckets_) { in Find()
604 int64 num_probes = 0; in DoInsert() local
627 ++num_probes; in DoInsert()
629 (bucket_index + num_probes) & bit_mask; // quadratic probing in DoInsert()
630 if (num_probes >= num_buckets_) { in DoInsert()
667 int64 num_probes = 0; in DoRemove() local
680 ++num_probes; in DoRemove()
[all …]
|
| /external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
| D | dense_image_warp_test.py | 187 num_probes=5): argument
204 for _ in range(num_probes):
|
| /external/bcc/src/cc/ |
| D | usdt.h | 267 size_t num_probes() const { return probes_.size(); } in num_probes() function
|
| /external/bcc/src/cc/includes/ |
| D | usdt.h | 267 size_t num_probes() const { return probes_.size(); } in num_probes() function
|
| /external/bcc/src/cc/usdt/ |
| D | usdt.cc | 457 for (size_t j = 0; j < ctx->num_probes(); j++) { in bcc_usdt_genargs()
|