Skip to content

Commit 9f2b667

Browse files
committed
Improve readability
1 parent 5a1ed16 commit 9f2b667

File tree

1 file changed

+8
-10
lines changed

1 file changed

+8
-10
lines changed

src/tensorflow.cc

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -598,17 +598,15 @@ SetStringInputTensor(
598598
}
599599
#endif // TRITON_ENABLE_GPU
600600

601-
auto callback = [](TRITONTF_Tensor* tensor, const size_t tensor_offset,
602-
const size_t element_idx, const char* content,
603-
const uint32_t len) {
604-
TRITONTF_TensorSetString(tensor, tensor_offset + element_idx, content, len);
605-
};
606-
auto fn = std::bind(
607-
callback, tensor, tensor_offset, std::placeholders::_1,
608-
std::placeholders::_2, std::placeholders::_3);
609-
601+
std::vector<std::pair<const char*, const uint32_t>> str_list;
610602
err = ValidateStringBuffer(
611-
content, content_byte_size, request_element_cnt, name, &element_idx, fn);
603+
content, content_byte_size, request_element_cnt, name, &str_list);
604+
// Set string values.
605+
for (; element_idx < str_list.size(); ++element_idx) {
606+
const auto& [addr, len] = str_list[element_idx];
607+
TRITONTF_TensorSetString(tensor, tensor_offset + element_idx, addr, len);
608+
}
609+
612610
if (err != nullptr) {
613611
RESPOND_AND_SET_NULL_IF_ERROR(response, err);
614612
FillStringTensor(

0 commit comments

Comments
 (0)