Searched refs:CPUDeviceContext (Results 1 – 3 of 3) sorted by relevance
50 void CPUDeviceContext::Initialize() { in Initialize()70 void CPUDeviceContext::Destroy() { in Destroy()78 bool CPUDeviceContext::AllocateMemory(DeviceAddress *const &address, size_t size) const { in AllocateMemory()91 void CPUDeviceContext::FreeMemory(DeviceAddress *const &address) const { in FreeMemory()102 DeviceAddressPtr CPUDeviceContext::CreateDeviceAddress(void *const device_ptr, size_t device_size, … in CreateDeviceAddress()107 void CPUDeviceContext::OptimizeGraph(const KernelGraphPtr &graph) const { in OptimizeGraph()118 void CPUDeviceContext::OptimizeSingleOpGraph(const KernelGraphPtr &graph) const { in OptimizeSingleOpGraph()124 void CPUDeviceContext::OptimizeGraphImpl(const KernelGraphPtr &graph) const { in OptimizeGraphImpl()134 void CPUDeviceContext::UpdateGraphDynamicShapeAttr(const NotNull<KernelGraphPtr> &graph) const { in UpdateGraphDynamicShapeAttr()172 void CPUDeviceContext::SetOperatorInfo(const std::vector<CNodePtr> &nodes) const { in SetOperatorInfo()[all …]
30 class CPUDeviceContext : public DeviceContext {32 explicit CPUDeviceContext(const DeviceContextKey &device_context_key) in CPUDeviceContext() function34 ~CPUDeviceContext() override = default;61 DISABLE_COPY_AND_ASSIGN(CPUDeviceContext);
36 class CPUDeviceContext; variable121 friend class mindspore::device::cpu::CPUDeviceContext;