search

VeriSilicon / TIM-VX

VeriSilicon Tensor Interface Module
Other
216 stars 85 forks source link

希望在Tensor中添加配置SetScalar接口，来配置vsi_nn_tensor_t的scalar属性 #641

Closed xie-oritek closed 8 months ago

xie-oritek commented 10 months ago

因为在Gather算子的op_setup接口中，会根据是否为Scalar值，来计算输出的维度信息，如下：

https://github.com/VeriSilicon/TIM-VX/blob/265e74ff1668d21214c67c360c0bef91c44ca7fb/src/tim/vx/internal/src/ops/vsi_nn_op_gather.c#L196-L197

sunshinemyson commented 10 months ago

@xie-oritek ，

可以的。我们放在10月把他enable起来。

sunshinemyson commented 9 months ago

Postpone to end of Nov.

sunshinemyson commented 8 months ago

Done by https://github.com/VeriSilicon/TIM-VX/pull/655