InfiniTensor/include
PanZezhong1725 2412c25e67
Issue 107: Add copyin Numpy and covertion to Numpy (#126)
* Add copyin_numpy and to_numpy for pybind TensorObj

* fix copyin size assertion

* fix size calculation for scalar (rank = 0) tensor

* Use pybind buffer instead of returning array

* fix format
2023-09-01 11:20:26 +08:00
..
bang Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
core Issue 107: Add copyin Numpy and covertion to Numpy (#126) 2023-09-01 11:20:26 +08:00
cuda 框架支持bert/gpt2模型构图 (#94) 2023-08-29 16:06:52 +08:00
ffi Add TVM codegen for MemboundOp (#35) 2022-09-22 18:06:45 +08:00
intelcpu Cpu backend2 (#77) 2023-04-17 12:15:23 +08:00
nnet Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
operators 框架支持bert/gpt2模型构图 (#94) 2023-08-29 16:06:52 +08:00
utils Add cuda transpose kernel (#115) 2023-08-22 14:22:15 +08:00
test.h Add python interface for CUDA operator evaluation (#42) 2022-09-27 10:41:12 +08:00