Dev optimize tensor setitem (#5501)
* Refactor expand and tensor setitem functional apis.
* Raise IndexInception in tensor getitem.
* Bugfix
* Bugfix
* Fix conflicit with xrt
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/api/python/framework/throw.h 1 addition, 1 deletiononeflow/api/python/framework/throw.h
- oneflow/core/common/error.cpp 6 additions, 0 deletionsoneflow/core/common/error.cpp
- oneflow/core/common/error.h 1 addition, 0 deletionsoneflow/core/common/error.h
- oneflow/core/common/error.proto 3 additions, 0 deletionsoneflow/core/common/error.proto
- oneflow/core/common/exception.h 1 addition, 0 deletionsoneflow/core/common/exception.h
- oneflow/core/common/shape.cpp 10 additions, 0 deletionsoneflow/core/common/shape.cpp
- oneflow/core/common/shape.h 3 additions, 0 deletionsoneflow/core/common/shape.h
- oneflow/core/functional/functional_api.yaml 4 additions, 1 deletiononeflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/array_functor.cpp 109 additions, 38 deletionsoneflow/core/functional/impl/array_functor.cpp
- oneflow/core/functional/tensor_index.cpp 24 additions, 20 deletionsoneflow/core/functional/tensor_index.cpp
- oneflow/core/functional/tensor_index.h 8 additions, 1 deletiononeflow/core/functional/tensor_index.h
- oneflow/python/framework/tensor.py 5 additions, 32 deletionsoneflow/python/framework/tensor.py
- oneflow/python/nn/modules/expand.py 1 addition, 33 deletionsoneflow/python/nn/modules/expand.py
Please register or sign in to comment