Skip to content

Commit 138f0a1

Browse files
shanshanptshanshanpt
authored
[Serving] Fix parse input request bug. (#736)
1 parent 59a8fa3 commit 138f0a1

1 file changed

Lines changed: 19 additions & 0 deletions

File tree

‎serving/processor/serving/util.cc‎

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -400,6 +400,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
400400
input.float_val_size() * sizeof(float));
401401

402402
ret.tensor = std::move(tensor);
403+
break;
403404
}
404405
case tensorflow::eas::DT_DOUBLE: {
405406
if (total_size != input.double_val_size()) {
@@ -413,6 +414,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
413414
memcpy(flat.data(), input.double_val().data(),
414415
input.double_val_size() * sizeof(double));
415416
ret.tensor = std::move(tensor);
417+
break;
416418
}
417419
case tensorflow::eas::DT_INT32: {
418420
if (total_size != input.int_val_size()) {
@@ -426,6 +428,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
426428
memcpy(flat.data(), input.int_val().data(),
427429
input.int_val_size() * sizeof(int));
428430
ret.tensor = std::move(tensor);
431+
break;
429432
}
430433
case tensorflow::eas::DT_UINT8: {
431434
if (total_size != input.int_val_size()) {
@@ -440,6 +443,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
440443
flat(i) = (uint8)input.int_val(i);
441444
}
442445
ret.tensor = std::move(tensor);
446+
break;
443447
}
444448
case tensorflow::eas::DT_INT16: {
445449
if (total_size != input.int_val_size()) {
@@ -454,6 +458,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
454458
flat(i) = (int16)input.int_val(i);
455459
}
456460
ret.tensor = std::move(tensor);
461+
break;
457462
}
458463
case tensorflow::eas::DT_UINT16: {
459464
if (total_size != input.int_val_size()) {
@@ -468,6 +473,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
468473
flat(i) = (uint16)input.int_val(i);
469474
}
470475
ret.tensor = std::move(tensor);
476+
break;
471477
}
472478
case tensorflow::eas::DT_INT8: {
473479
if (total_size != input.int_val_size()) {
@@ -482,6 +488,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
482488
flat(i) = (int8)input.int_val(i);
483489
}
484490
ret.tensor = std::move(tensor);
491+
break;
485492
}
486493
case tensorflow::eas::DT_STRING: {
487494
if (total_size != input.string_val_size()) {
@@ -496,6 +503,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
496503
flat(i) = input.string_val(i);
497504
}
498505
ret.tensor = std::move(tensor);
506+
break;
499507
}
500508
case tensorflow::eas::DT_COMPLEX64: {
501509
if (total_size != input.float_val_size() / 2) {
@@ -510,6 +518,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
510518
flat(i) = complex64(input.float_val(i), input.float_val(i + 1));
511519
}
512520
ret.tensor = std::move(tensor);
521+
break;
513522
}
514523
case tensorflow::eas::DT_COMPLEX128: {
515524
if (total_size != input.double_val_size() / 2) {
@@ -524,6 +533,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
524533
flat(i) = complex64(input.double_val(i), input.double_val(i + 1));
525534
}
526535
ret.tensor = std::move(tensor);
536+
break;
527537
}
528538
case tensorflow::eas::DT_INT64: {
529539
if (total_size != input.int64_val_size()) {
@@ -537,6 +547,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
537547
memcpy(flat.data(), input.int64_val().data(),
538548
input.int64_val_size() * sizeof(int64));
539549
ret.tensor = std::move(tensor);
550+
break;
540551
}
541552
case tensorflow::eas::DT_BOOL: {
542553
if (total_size != input.bool_val_size()) {
@@ -551,6 +562,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
551562
flat(i) = input.bool_val(i);
552563
}
553564
ret.tensor = std::move(tensor);
565+
break;
554566
}
555567
case tensorflow::eas::DT_QINT8: {
556568
if (total_size != input.int_val_size()) {
@@ -565,6 +577,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
565577
flat(i) = qint8(input.int_val(i));
566578
}
567579
ret.tensor = std::move(tensor);
580+
break;
568581
}
569582
case tensorflow::eas::DT_QUINT8: {
570583
if (total_size != input.int_val_size()) {
@@ -579,6 +592,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
579592
flat(i) = quint8(input.int_val(i));
580593
}
581594
ret.tensor = std::move(tensor);
595+
break;
582596
}
583597
case tensorflow::eas::DT_QINT32: {
584598
if (total_size != input.int_val_size()) {
@@ -593,6 +607,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
593607
flat(i) = qint32(input.int_val(i));
594608
}
595609
ret.tensor = std::move(tensor);
610+
break;
596611
}
597612
case tensorflow::eas::DT_QINT16: {
598613
if (total_size != input.int_val_size()) {
@@ -607,6 +622,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
607622
flat(i) = qint16(input.int_val(i));
608623
}
609624
ret.tensor = std::move(tensor);
625+
break;
610626
}
611627
case tensorflow::eas::DT_QUINT16: {
612628
if (total_size != input.int_val_size()) {
@@ -621,6 +637,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
621637
flat(i) = quint16(input.int_val(i));
622638
}
623639
ret.tensor = std::move(tensor);
640+
break;
624641
}
625642
case tensorflow::eas::DT_BFLOAT16: {
626643
if (total_size != input.float_val_size()) {
@@ -635,6 +652,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
635652
flat.data(),
636653
input.float_val_size());
637654
ret.tensor = std::move(tensor);
655+
break;
638656
}
639657
case tensorflow::eas::DT_HALF: {
640658
if (total_size != input.float_val_size()) {
@@ -649,6 +667,7 @@ TensorWithStatus Proto2Tensor(const std::string& key,
649667
flat(i) = Eigen::half(input.float_val(i));
650668
}
651669
ret.tensor = std::move(tensor);
670+
break;
652671
}
653672
case tensorflow::eas::DT_RESOURCE: {
654673
ret.status = errors::InvalidArgument(

0 commit comments

Comments
 (0)