|
17 | 17 | #include "openvino/op/parameter.hpp" |
18 | 18 | #include "openvino/reference/utils/coordinate_transform.hpp" |
19 | 19 | #include "openvino/runtime/allocator.hpp" |
| 20 | +#include "openvino/runtime/make_tensor.hpp" |
20 | 21 | #include "openvino/runtime/remote_tensor.hpp" |
21 | 22 | #include "openvino/runtime/tensor.hpp" |
22 | 23 |
|
@@ -953,6 +954,22 @@ TEST_F(OVTensorTest, createAllocationOverflow) { |
953 | 954 | HasSubstr("Cannot allocate memory")); |
954 | 955 | } |
955 | 956 |
|
| 957 | +TEST_F(OVTensorTest, getTensorDataOffsetFromRoiTensor) { |
| 958 | + const ov::Tensor tensor(ov::element::f32, ov::Shape{3, 5, 32, 128}); |
| 959 | + ov::Tensor roi_tensor; |
| 960 | + OV_ASSERT_NO_THROW(roi_tensor = ov::Tensor(tensor, {1, 2, 5, 64}, {2, 3, 10, 128})); |
| 961 | + size_t tensor_data_offset = ov::get_tensor_data_offset(ov::get_tensor_impl(roi_tensor)); |
| 962 | + EXPECT_EQ(static_cast<const uint8_t*>(roi_tensor.data()), |
| 963 | + static_cast<const uint8_t*>(tensor.data()) + tensor_data_offset); |
| 964 | +} |
| 965 | + |
| 966 | +TEST_F(OVTensorTest, getTensorDataOffsetFromGeneralTensor) { |
| 967 | + const ov::Tensor tensor(ov::element::f32, ov::Shape{3, 5, 32, 128}); |
| 968 | + size_t tensor_data_offset = ov::get_tensor_data_offset(ov::get_tensor_impl(tensor)); |
| 969 | + EXPECT_EQ(static_cast<const uint8_t*>(tensor.data()), |
| 970 | + static_cast<const uint8_t*>(tensor.data()) + tensor_data_offset); |
| 971 | +} |
| 972 | + |
956 | 973 | struct TestParams { |
957 | 974 | ov::Shape src_shape; |
958 | 975 | ov::Strides src_strides; |
|
0 commit comments