Home
last modified time | relevance | path

Searched refs:max_output_size (Results 1 – 25 of 52) sorted by relevance

123

/external/tensorflow/tensorflow/lite/kernels/internal/
Dnon_max_suppression_test.cc72 const int max_output_size = 4; in TEST() local
80 boxes.data(), /**num_boxes=**/ 0, scores.data(), max_output_size, in TEST()
99 const int max_output_size = kNumBoxes; in TEST() local
107 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
115 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
127 int max_output_size; in TEST() local
135 max_output_size = 100; in TEST()
137 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
145 max_output_size = 2; in TEST()
147 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
[all …]
/external/rust/crates/grpcio-sys/grpc/src/core/lib/compression/
Dstream_compression_identity.cc36 size_t max_output_size) { in grpc_stream_compression_pass_through() argument
37 if (max_output_size >= in->length) { in grpc_stream_compression_pass_through()
44 *output_size = max_output_size; in grpc_stream_compression_pass_through()
46 grpc_slice_buffer_move_first(in, max_output_size, out); in grpc_stream_compression_pass_through()
52 grpc_slice_buffer* out, size_t* output_size, size_t max_output_size, in grpc_stream_compress_identity() argument
57 grpc_stream_compression_pass_through(in, out, output_size, max_output_size); in grpc_stream_compress_identity()
63 grpc_slice_buffer* out, size_t* output_size, size_t max_output_size, in grpc_stream_decompress_identity() argument
68 grpc_stream_compression_pass_through(in, out, output_size, max_output_size); in grpc_stream_decompress_identity()
Dstream_compression_gzip.cc39 size_t* output_size, size_t max_output_size, int flush, in gzip_flate() argument
48 size_t original_max_output_size = max_output_size; in gzip_flate()
49 while (max_output_size > 0 && (in->length > 0 || flush) && !eoc) { in gzip_flate()
50 size_t slice_size = max_output_size < OUTPUT_BLOCK_SIZE ? max_output_size in gzip_flate()
125 max_output_size -= (slice_size - ctx->zs.avail_out); in gzip_flate()
132 *output_size = original_max_output_size - max_output_size; in gzip_flate()
141 size_t max_output_size, in grpc_stream_compress_gzip() argument
163 return gzip_flate(gzip_ctx, in, out, output_size, max_output_size, gzip_flush, in grpc_stream_compress_gzip()
171 size_t max_output_size, in grpc_stream_decompress_gzip() argument
179 return gzip_flate(gzip_ctx, in, out, output_size, max_output_size, in grpc_stream_decompress_gzip()
Dstream_compression.h56 size_t max_output_size, grpc_stream_compression_flush flush);
59 size_t* output_size, size_t max_output_size,
81 size_t* output_size, size_t max_output_size,
93 size_t* output_size, size_t max_output_size,
Dstream_compression.cc32 size_t* output_size, size_t max_output_size, in grpc_stream_compress() argument
34 return ctx->vtable->compress(ctx, in, out, output_size, max_output_size, in grpc_stream_compress()
40 size_t* output_size, size_t max_output_size, in grpc_stream_decompress() argument
42 return ctx->vtable->decompress(ctx, in, out, output_size, max_output_size, in grpc_stream_decompress()
/external/grpc-grpc/src/core/lib/compression/
Dstream_compression_identity.cc36 size_t max_output_size) { in grpc_stream_compression_pass_through() argument
37 if (max_output_size >= in->length) { in grpc_stream_compression_pass_through()
44 *output_size = max_output_size; in grpc_stream_compression_pass_through()
46 grpc_slice_buffer_move_first(in, max_output_size, out); in grpc_stream_compression_pass_through()
54 size_t max_output_size, in grpc_stream_compress_identity() argument
59 grpc_stream_compression_pass_through(in, out, output_size, max_output_size); in grpc_stream_compress_identity()
65 grpc_slice_buffer* out, size_t* output_size, size_t max_output_size, in grpc_stream_decompress_identity() argument
70 grpc_stream_compression_pass_through(in, out, output_size, max_output_size); in grpc_stream_decompress_identity()
Dstream_compression_gzip.cc39 size_t* output_size, size_t max_output_size, int flush, in gzip_flate() argument
48 size_t original_max_output_size = max_output_size; in gzip_flate()
49 while (max_output_size > 0 && (in->length > 0 || flush) && !eoc) { in gzip_flate()
50 size_t slice_size = max_output_size < OUTPUT_BLOCK_SIZE ? max_output_size in gzip_flate()
125 max_output_size -= (slice_size - ctx->zs.avail_out); in gzip_flate()
132 *output_size = original_max_output_size - max_output_size; in gzip_flate()
141 size_t max_output_size, in grpc_stream_compress_gzip() argument
163 return gzip_flate(gzip_ctx, in, out, output_size, max_output_size, gzip_flush, in grpc_stream_compress_gzip()
171 size_t max_output_size, in grpc_stream_decompress_gzip() argument
179 return gzip_flate(gzip_ctx, in, out, output_size, max_output_size, in grpc_stream_decompress_gzip()
Dstream_compression.h56 size_t max_output_size, grpc_stream_compression_flush flush);
59 size_t* output_size, size_t max_output_size,
81 size_t* output_size, size_t max_output_size,
93 size_t* output_size, size_t max_output_size,
Dstream_compression.cc31 size_t* output_size, size_t max_output_size, in grpc_stream_compress() argument
33 return ctx->vtable->compress(ctx, in, out, output_size, max_output_size, in grpc_stream_compress()
39 size_t* output_size, size_t max_output_size, in grpc_stream_decompress() argument
41 return ctx->vtable->decompress(ctx, in, out, output_size, max_output_size, in grpc_stream_decompress()
/external/tensorflow/tensorflow/compiler/tests/
Dimage_ops_test.py781 max_output_size = 128
796 max_output_size=max_output_size,
808 self.assertEqual(indices_tf.size, max_output_size)
819 max_output_size = 3
834 max_output_size=max_output_size,
847 self.assertEqual(indices_tf.size, max_output_size)
861 max_output_size = 3
876 max_output_size=max_output_size,
889 self.assertEqual(indices_tf.size, max_output_size)
904 max_output_size = 1
[all …]
/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cc163 int num_boxes, const Tensor& max_output_size, in DoNonMaxSuppressionOp() argument
170 const int output_size = max_output_size.scalar<int>()(); in DoNonMaxSuppressionOp()
590 const Tensor& max_output_size = context->input(2); in Compute() local
592 context, TensorShapeUtils::IsScalar(max_output_size.shape()), in Compute()
594 max_output_size.shape().DebugString())); in Compute()
608 DoNonMaxSuppressionOp<float>(context, scores, num_boxes, max_output_size, in Compute()
629 const Tensor& max_output_size = context->input(2); in Compute() local
631 context, TensorShapeUtils::IsScalar(max_output_size.shape()), in Compute()
633 max_output_size.shape().DebugString())); in Compute()
655 DoNonMaxSuppressionOp<T>(context, scores, num_boxes, max_output_size, in Compute()
[all …]
Dnon_max_suppression_op.cu.cc409 const Tensor& scores, const int64_t max_output_size, in DoNMS() argument
497 int len_output = pad_to_max_output ? max_output_size : 0; in DoNMS()
513 &num_to_keep, context, max_output_size, flip_boxes); in DoNMS()
520 int num_outputs = std::min(num_to_keep, (int)max_output_size); // no padding! in DoNMS()
521 if (pad_to_max_output && num_outputs != max_output_size) { in DoNMS()
523 0, TensorShape({max_output_size}), &output_indices)); in DoNMS()
524 config = GetGpuLaunchConfig(max_output_size, device); in DoNMS()
550 const Tensor& max_output_size, in CheckValidInputs() argument
552 if (!TensorShapeUtils::IsScalar(max_output_size.shape())) { in CheckValidInputs()
554 max_output_size.shape().DebugString(), in CheckValidInputs()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression_test.cc35 void SetMaxOutputSize(int max_output_size) { in SetMaxOutputSize() argument
36 PopulateTensor(input_max_output_size_, {max_output_size}); in SetMaxOutputSize()
72 const int max_output_size = -1) { in NonMaxSuppressionV4OpModel() argument
78 AddConstInput(TensorType_INT32, {max_output_size}); in NonMaxSuppressionV4OpModel()
163 const int max_output_size = -1) { in NonMaxSuppressionV5OpModel() argument
169 AddConstInput(TensorType_INT32, {max_output_size}); in NonMaxSuppressionV5OpModel()
/external/grpc-grpc/test/core/compression/
Dstream_compression_test.cc111 size_t max_output_size = 2; in test_stream_compression_simple_compress_decompress_with_output_size_constraint() local
113 max_output_size, &end_of_context)); in test_stream_compression_simple_compress_decompress_with_output_size_constraint()
114 GPR_ASSERT(output_size == max_output_size); in test_stream_compression_simple_compress_decompress_with_output_size_constraint()
118 GPR_ASSERT(GRPC_SLICE_LENGTH(slice_recv) == max_output_size); in test_stream_compression_simple_compress_decompress_with_output_size_constraint()
119 GPR_ASSERT(0 == strncmp(test_str, str_recv, max_output_size)); in test_stream_compression_simple_compress_decompress_with_output_size_constraint()
122 size_t remaining_size = sizeof(test_str) - 1 - max_output_size; in test_stream_compression_simple_compress_decompress_with_output_size_constraint()
128 GPR_ASSERT(slice_buffer_equals_string(&sink, test_str + max_output_size)); in test_stream_compression_simple_compress_decompress_with_output_size_constraint()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DNonMaxSuppressionV2.pbtxt12 name: "max_output_size"
35 name: "max_output_size"
71 name: "max_output_size"
DNonMaxSuppressionV3.pbtxt12 name: "max_output_size"
39 name: "max_output_size"
79 name: "max_output_size"
DNonMaxSuppressionV4.pbtxt12 name: "max_output_size"
50 name: "max_output_size"
101 name: "max_output_size"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DNonMaxSuppressionV2.pbtxt12 name: "max_output_size"
35 name: "max_output_size"
71 name: "max_output_size"
DNonMaxSuppressionV3.pbtxt12 name: "max_output_size"
39 name: "max_output_size"
79 name: "max_output_size"
DNonMaxSuppressionV4.pbtxt12 name: "max_output_size"
50 name: "max_output_size"
101 name: "max_output_size"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_NonMaxSuppressionV5.pbtxt17 name: "max_output_size"
49 `max_output_size`. Defaults to false.
56 indices from the boxes tensor, where `M <= max_output_size`.
63 scores for each selected box, where `M <= max_output_size`. Scores only differ
92 boxes, scores, max_output_size, iou_threshold, score_threshold)
Dapi_def_NonMaxSuppressionV4.pbtxt17 name: "max_output_size"
41 `max_output_size`. Defaults to false.
48 indices from the boxes tensor, where `M <= max_output_size`.
75 boxes, scores, max_output_size, iou_threshold, score_threshold)
Dapi_def_NonMaxSuppressionWithOverlaps.pbtxt18 name: "max_output_size"
42 indices from the boxes tensor, where `M <= max_output_size`.
59 overlaps, scores, max_output_size, overlap_threshold, score_threshold)
Dapi_def_NonMaxSuppression.pbtxt17 name: "max_output_size"
27 indices from the boxes tensor, where `M <= max_output_size`.
53 boxes, scores, max_output_size, iou_threshold)
Dapi_def_NonMaxSuppressionV2.pbtxt17 name: "max_output_size"
34 indices from the boxes tensor, where `M <= max_output_size`.
55 boxes, scores, max_output_size, iou_threshold)

123