| @@ -17,6 +17,7 @@ | |||||
| #include "minddata/dataset/include/execute.h" | #include "minddata/dataset/include/execute.h" | ||||
| #include "minddata/dataset/core/de_tensor.h" | #include "minddata/dataset/core/de_tensor.h" | ||||
| #include "minddata/dataset/core/device_tensor.h" | #include "minddata/dataset/core/device_tensor.h" | ||||
| #include "minddata/dataset/core/device_resource.h" | |||||
| #include "minddata/dataset/core/tensor_row.h" | #include "minddata/dataset/core/tensor_row.h" | ||||
| #include "minddata/dataset/include/tensor.h" | #include "minddata/dataset/include/tensor.h" | ||||
| #include "minddata/dataset/include/type_id.h" | #include "minddata/dataset/include/type_id.h" | ||||
| @@ -22,13 +22,12 @@ | |||||
| #include <memory> | #include <memory> | ||||
| #include "include/api/context.h" | #include "include/api/context.h" | ||||
| #include "include/api/types.h" | #include "include/api/types.h" | ||||
| #include "minddata/dataset/core/device_resource.h" | |||||
| #include "minddata/dataset/include/constants.h" | #include "minddata/dataset/include/constants.h" | ||||
| #include "minddata/dataset/include/transforms.h" | #include "minddata/dataset/include/transforms.h" | ||||
| namespace mindspore { | namespace mindspore { | ||||
| namespace dataset { | namespace dataset { | ||||
| class DeviceResource; | |||||
| // class to run tensor operations in eager mode | // class to run tensor operations in eager mode | ||||
| class Execute { | class Execute { | ||||
| public: | public: | ||||