server icon indicating copy to clipboard operation
server copied to clipboard

Setting tensor name in python bls

Open SunXuan90 opened this issue 1 year ago • 3 comments

This is really for Python backend, but I can't find issue page there. I noticed that in source code of python backend, PbTensor class has a method SetName, but not implemented. This method is pretty useful for bls. Sometimes we need to pass some tensors from one model's output to another one's input, but the tensor name is different. Currently we can only do so using as_numpy or to_dlpack to get the content out and create a new tensor. It would be really nice that the method gets implemented.

SunXuan90 avatar Jan 17 '24 05:01 SunXuan90

CC @Tabrizian

nv-kmcgill53 avatar Jan 18 '24 20:01 nv-kmcgill53

Pinging @Tabrizian

the-david-oy avatar Feb 20 '24 19:02 the-david-oy

Hi, Has this issue been resolved?

XiaoxueWang1 avatar Aug 19 '24 18:08 XiaoxueWang1