Searched defs:GPUDeviceAddress (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | gpu_device_address.h | 35 …explicit GPUDeviceAddress(const KernelTensorPtr &kernel_tensor) : LoadableDeviceAddress(kernel_ten… in GPUDeviceAddress() function
38 …GPUDeviceAddress(void *ptr, size_t size) : LoadableDeviceAddress(ptr, size) { SetDevicePtrDeleter(… in GPUDeviceAddress() function
39 GPUDeviceAddress(void *ptr, size_t size, const string &format, TypeId type_id) in GPUDeviceAddress() function
43 …GPUDeviceAddress(void *ptr, size_t size, const std::string &format, TypeId type_id, const KernelWi… in GPUDeviceAddress() function
47 …GPUDeviceAddress(void *ptr, size_t size, const std::string &format, TypeId type_id, const std::str… in GPUDeviceAddress() function
52 …GPUDeviceAddress(void *ptr, size_t size, const ShapeVector &shape_vector, const Format &format, Ty… in GPUDeviceAddress() function
|