Feat tensor stride property (#5543)
* feat(Stride): add Stride class
* feat(Tensor): support stride and storage_offset interface
* feat(Tensor): add is_contiguous interface
* remove test declaration
* feat(TensorMeta): add hash and compare for stride
* refine code
* refine IsContiguous function
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/api/python/framework/tensor.cpp 13 additions, 0 deletionsoneflow/api/python/framework/tensor.cpp
- oneflow/core/common/shape_vec.h 2 additions, 0 deletionsoneflow/core/common/shape_vec.h
- oneflow/core/framework/stride.cpp 51 additions, 0 deletionsoneflow/core/framework/stride.cpp
- oneflow/core/framework/stride.h 65 additions, 0 deletionsoneflow/core/framework/stride.h
- oneflow/core/framework/tensor.h 4 additions, 0 deletionsoneflow/core/framework/tensor.h
- oneflow/core/framework/tensor_impl.cpp 10 additions, 2 deletionsoneflow/core/framework/tensor_impl.cpp
- oneflow/core/framework/tensor_impl.h 4 additions, 0 deletionsoneflow/core/framework/tensor_impl.h
- oneflow/core/framework/tensor_meta.h 9 additions, 2 deletionsoneflow/core/framework/tensor_meta.h
- oneflow/core/framework/tensor_method.cpp 43 additions, 0 deletionsoneflow/core/framework/tensor_method.cpp
- oneflow/core/framework/tensor_method.h 32 additions, 0 deletionsoneflow/core/framework/tensor_method.h
- oneflow/python/framework/tensor.py 12 additions, 0 deletionsoneflow/python/framework/tensor.py
- oneflow/python/test/tensor/test_tensor.py 9 additions, 0 deletionsoneflow/python/test/tensor/test_tensor.py
Please register or sign in to comment