Home
last modified time | relevance | path

Searched refs:num_valid (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dimage_ops_test.py984 (indices_tf, num_valid) = sess.run(
988 self.assertEqual(num_valid, 3)
989 self.assertAllClose(indices_tf[:num_valid], [3, 0, 5])
1026 (indices_tf, num_valid) = sess.run(
1030 self.assertEqual(num_valid, 2)
1031 self.assertAllClose(indices_tf[:num_valid], [3, 0])
1069 (indices_tf, num_valid) = sess.run(
1073 self.assertEqual(num_valid, 1)
1074 self.assertAllClose(indices_tf[:num_valid], [3])
1110 (indices_tf, num_valid) = sess.run(
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhere_op.cc307 XlaOp num_valid = in CompileWhereWithPrefixSum() local
311 &ctx->value_inference(), scattered, num_valid, 0); in CompileWhereWithPrefixSum()
317 return xla::SetDimensionSize(scattered, num_valid, 0); in CompileWhereWithPrefixSum()
Dimage_ops.cc420 xla::XlaOp selected_indices, num_valid; in Compile() local
585 xla::XlaOp num_valid = in ComputeResult() local
602 &context->value_inference(), selected_indices, num_valid, 0); in ComputeResult()
609 xla::SetDimensionSize(selected_indices, num_valid, 0); in ComputeResult()
613 if (pad_to_max_output_size) context->SetOutput(1, num_valid); in ComputeResult()
630 xla::XlaOp selected_indices, num_valid; in Compile() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_fingerprint_lookup.h69 size_t num_valid() { in num_valid() function
/external/grpc-grpc/src/core/ext/filters/client_channel/lb_policy/grpclb/
Dgrpclb.cc443 size_t num_valid = 0; in ProcessServerlist() local
447 if (IsServerValid(serverlist->servers[i], i, true)) ++num_valid; in ProcessServerlist()
450 grpc_lb_addresses_create(num_valid, &lb_token_vtable); in ProcessServerlist()
460 GPR_ASSERT(addr_idx < num_valid); in ProcessServerlist()
490 GPR_ASSERT(addr_idx == num_valid); in ProcessServerlist()
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_impl.py5416 idx, num_valid = non_max_suppression_padded_v2(
5421 idx = idx[0, :num_valid]
5428 return idx, num_valid
5644 num_valid = math_ops.minimum(output_size, max_output_size)
5665 num_valid_expanded = array_ops.expand_dims(num_valid, 1)
5669 num_valid = array_ops.reshape(num_valid, batch_dims)
5670 return idx, num_valid
Dimage_ops_test.py5372 selected_indices, num_valid = image_ops.non_max_suppression_padded(
5385 self.assertEqual(num_valid.eval(), 3)
5430 selected_indices_padded, num_valid_padded, selected_indices, num_valid = \
5438 self.assertEqual(self.evaluate(num_valid), 3)
5455 selected_indices, num_valid = image_ops.non_max_suppression_padded(
5465 self.assertEqual(num_valid.eval(), 3)
5498 selected_indices, num_valid = func(boxes_np, scores_np,
5504 self.assertEqual(self.evaluate(num_valid), 3)