class ov::intel_gpu::ocl::USMTensor¶
Overview¶
This class represents an abstraction for GPU plugin remote tensor which can be shared with user-supplied USM device pointer. The plugin object derived from this class can be obtained with ClContext::create_tensor() call. More…
#include <ocl.hpp>
class USMTensor: public ov::RemoteTensor
{
public:
// methods
static void type_check(const Tensor& tensor);
void \* get();
};
Inherited Members¶
public:
// methods
Tensor& operator = (const Tensor& other);
Tensor& operator = (Tensor&& other);
void set_shape(const ov::Shape& shape);
element::Type get_element_type() const;
Shape get_shape() const;
size_t get_size() const;
size_t get_byte_size() const;
Strides get_strides() const;
void \* data(const element::Type type = {}) const;
template <typename T, typename datatype = typename std::decay<T>::type>
T \* data() const;
bool operator ! () const;
operator bool () const;
template <typename T>
std::enable_if<std::is_base_of<Tensor, T>::value, bool>::type is() const;
template <typename T>
const std::enable_if<std::is_base_of<Tensor, T>::value, T>::type as() const;
static void type_check(const Tensor& tensor);
static void type_check(
const Tensor& tensor,
const std::map<std::string, std::vector<std::string>>& type_info = {}
);
void \* data(const element::Type);
template <typename T>
T \* data();
ov::AnyMap get_params() const;
Detailed Documentation¶
This class represents an abstraction for GPU plugin remote tensor which can be shared with user-supplied USM device pointer. The plugin object derived from this class can be obtained with ClContext::create_tensor() call.
User can obtain USM pointer from this class.
Methods¶
static void type_check(const Tensor& tensor)
Checks that type defined runtime parameters are presented in remote object.
Parameters:
tensor |
a tensor to check |
void \* get()
Returns the underlying USM pointer.
Returns:
underlying USM pointer