Skip to content

Commit efa2208

Browse files
committed
Initial commit
1 parent 7f21b67 commit efa2208

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/python_be.cc

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -443,7 +443,9 @@ ModelInstanceState::GetInputTensor(
443443
if (input_dtype == TRITONSERVER_TYPE_BYTES) {
444444
const char* content = reinterpret_cast<char*>(input_tensor->DataPtr());
445445
size_t content_byte_size = input_tensor->ByteSize();
446-
const size_t request_element_cnt = GetElementCount(input_tensor->Dims());
446+
int64_t request_element_cnt = 0;
447+
RETURN_IF_ERROR(
448+
GetElementCount(input_tensor->Dims(), &request_element_cnt));
447449
RETURN_IF_ERROR(ValidateStringBuffer(
448450
content, content_byte_size, request_element_cnt, input_name,
449451
nullptr /* str_list */));

0 commit comments

Comments
 (0)