aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/nearest_neighbor
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-01-30 10:43:03 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-01-30 12:33:54 -0800
commit4463d105a8a4a83642b9709ba79310e8f4ddf577 (patch)
tree240e9a0a9a6b9ad956c704776a33126ba00cbfe8 /tensorflow/contrib/nearest_neighbor
parent8f0e7207774279f4fe50f4d6c4fbd576e2941463 (diff)
Cleanup: Ran clang-format on all *.{cc,h} files in tensorflow/contrib/.../*.{hh,c}.
PiperOrigin-RevId: 183855242
Diffstat (limited to 'tensorflow/contrib/nearest_neighbor')
-rw-r--r--tensorflow/contrib/nearest_neighbor/kernels/hyperplane_lsh_probes.cc24
1 files changed, 12 insertions, 12 deletions
diff --git a/tensorflow/contrib/nearest_neighbor/kernels/hyperplane_lsh_probes.cc b/tensorflow/contrib/nearest_neighbor/kernels/hyperplane_lsh_probes.cc
index 2b412fac9a..13db6f62f5 100644
--- a/tensorflow/contrib/nearest_neighbor/kernels/hyperplane_lsh_probes.cc
+++ b/tensorflow/contrib/nearest_neighbor/kernels/hyperplane_lsh_probes.cc
@@ -75,7 +75,8 @@ class HyperplaneLSHProbesOp : public OpKernel {
num_hyperplanes_per_table, "."));
OP_REQUIRES(context, num_hyperplanes_per_table <= 30,
InvalidArgument("Need num_hyperplanes_per_table <= 30, got ",
- num_hyperplanes_per_table, ". "
+ num_hyperplanes_per_table,
+ ". "
"If you need more hyperplanes, change this Op"
" to work for larger integer types (int64)."));
@@ -88,12 +89,13 @@ class HyperplaneLSHProbesOp : public OpKernel {
InvalidArgument("num_probes must be at least 1."));
int expected_num_hyperplanes = num_tables * num_hyperplanes_per_table;
- OP_REQUIRES(
- context, products_tensor.dim_size(1) == expected_num_hyperplanes,
- InvalidArgument("Expected number of hyperplanes is ",
- expected_num_hyperplanes, " but received ",
- products_tensor.dim_size(1), " inner products per "
- "point."));
+ OP_REQUIRES(context,
+ products_tensor.dim_size(1) == expected_num_hyperplanes,
+ InvalidArgument("Expected number of hyperplanes is ",
+ expected_num_hyperplanes, " but received ",
+ products_tensor.dim_size(1),
+ " inner products per "
+ "point."));
auto products_eigen_tensor = products_tensor.matrix<CoordinateType>();
ConstMatrixMap products_matrix(products_eigen_tensor.data(),
@@ -116,13 +118,11 @@ class HyperplaneLSHProbesOp : public OpKernel {
// lschmidt's workstation.
int64 cost_per_unit = 21 * num_hyperplanes_per_table * num_tables;
if (num_probes > num_tables) {
- cost_per_unit += 110 * num_hyperplanes_per_table
- * (num_probes - num_tables);
+ cost_per_unit +=
+ 110 * num_hyperplanes_per_table * (num_probes - num_tables);
}
context->device()->tensorflow_cpu_worker_threads()->workers->ParallelFor(
- batch_size,
- cost_per_unit,
- [&](int64 start, int64 end) {
+ batch_size, cost_per_unit, [&](int64 start, int64 end) {
HyperplaneMultiprobe<CoordinateType, int32> multiprobe(
num_hyperplanes_per_table, num_tables);