Searched refs:GPUDeviceContext (Results 1 – 3 of 3) sorted by relevance
56 void GPUDeviceContext::Initialize() { in Initialize()111 bool GPUDeviceContext::InitDevice() { in InitDevice()139 void GPUDeviceContext::Destroy() { in Destroy()166 bool GPUDeviceContext::AllocateMemory(DeviceAddress *const &address, size_t size) const { in AllocateMemory()181 void GPUDeviceContext::FreeMemory(DeviceAddress *const &address) const { in FreeMemory()191 bool GPUDeviceContext::AllocateContinuousMemory(const std::vector<DeviceAddressPtr> &addr_list, siz… in AllocateContinuousMemory()199 DeviceAddressPtr GPUDeviceContext::CreateDeviceAddress(void *const device_ptr, size_t device_size, … in CreateDeviceAddress()204 void GPUDeviceContext::OptimizeGraph(const KernelGraphPtr &graph) const { in OptimizeGraph()228 void GPUDeviceContext::OptimizeGraphWithoutDeviceInfo(const KernelGraphPtr &graph) const { in OptimizeGraphWithoutDeviceInfo()237 void GPUDeviceContext::OptimizeGraphWithDeviceInfo(const KernelGraphPtr &graph) const { in OptimizeGraphWithDeviceInfo()[all …]
30 class GPUDeviceContext : public DeviceContext {32 explicit GPUDeviceContext(const DeviceContextKey &device_context_key) in GPUDeviceContext() function34 ~GPUDeviceContext() override = default;75 DISABLE_COPY_AND_ASSIGN(GPUDeviceContext);
51 class GPUDeviceContext; variable124 friend class mindspore::device::gpu::GPUDeviceContext;