- 11 4月, 2023 1 次提交
-
-
由 HongyuJia 提交于
* [CustomOP Unittest] Polish unit test of custom operator, kCPU->CPU * AllocationType::CPU -> is_cpu()
-
- 28 3月, 2023 1 次提交
-
-
由 Zheng-Bicheng 提交于
-
- 27 3月, 2023 1 次提交
-
-
由 HongyuJia 提交于
* [CustomOP Inplace] Automap inplace dtype and shape, prepare for vector<Tensor> output * delete dtype,shape func of multi_inplace op * [CustomOP Inplace] Automap inplace dtype and shape, support vector<Tensor> output
-
- 20 3月, 2023 1 次提交
-
-
由 HongyuJia 提交于
-
- 16 3月, 2023 1 次提交
-
-
由 HongyuJia 提交于
* init unit test commit, contains register thinking * support inplace * get inplaced x.grad * Try support inplace and hook at the same time * Support inplace, need debug * Support inplace successfully * Inplace use Tensor&, consistent with Tensor* * fix MapPlainOutputs bug * fix double grad inplace error
-