Home
last modified time | relevance | path

Searched refs:AllocateMemory (Results 1 – 25 of 64) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dmemory_manager_actor.cc25 void MemoryManagerActor::AllocateMemory(const std::vector<DeviceTensor *> *alloc_list, in AllocateMemory() function in mindspore::runtime::MemoryManagerActor
38 if (!device_context->AllocateMemory(device_tensor, device_tensor->GetSize())) { in AllocateMemory()
103 if (!device_context->AllocateMemory(device_tensor, device_tensor->GetSize())) { in AllocateBatchMemory()
Dkernel_actor.cc152 void AllocateMemory(const std::vector<DeviceTensor *> &alloc_list, const DeviceContext *device_cont… in AllocateMemory() function
163 if (!device_context->AllocateMemory(device_tensor, device_tensor->GetSize())) { in AllocateMemory()
193 …Async(memory_manager_aid_, &MemoryManagerActor::AllocateMemory, &memory_alloc_list_, device_contex… in SendMemoryAllocReq()
196 AllocateMemory(memory_alloc_list_, device_contexts_[0], context, GetAID().Name()); in SendMemoryAllocReq()
285 if (!device_contexts_[0]->AllocateMemory(copy_input_device_tensors_[input_data->index_].get(), in CopyInputDeviceTensor()
Dmemory_manager_actor.h40 …void AllocateMemory(const std::vector<DeviceTensor *> *alloc_list, const DeviceContext *device_con…
Ddata_source_actor.cc139 …Async(memory_manager_aid_, &MemoryManagerActor::AllocateMemory, &device_tensors, device_contexts_[… in SendMemoryAllocReq()
236 …Async(memory_manager_aid_, &MemoryManagerActor::AllocateMemory, &device_tensors, device_contexts_[… in SendMemoryAllocReq()
Ddata_prepare_actor.cc41 (!device_context->AllocateMemory(device_tensor.get(), device_tensor->GetSize()))) { in SyncTensorData()
384 if (!device_context->AllocateMemory(device_tensor.get(), device_tensor->GetSize())) { in PrepareDataForValueNode()
453 …(!another_device_context->AllocateMemory(another_device_tensor.get(), another_device_tensor->GetSi… in PrepareDataForWeightNode()
Dcopy_actor.cc71 Async(memory_manager_aid_, &MemoryManagerActor::AllocateMemory, &output_device_tensor_, in SendMemoryAllocReq()
/third_party/skia/third_party/externals/dawn/src/dawn_native/vulkan/
DResourceMemoryAllocatorVk.cpp65 ResultOrError<ResourceMemoryAllocation> AllocateMemory(uint64_t size, uint64_t alignment) { in AllocateMemory() function in dawn_native::vulkan::ResourceMemoryAllocator::SingleTypeAllocator
91 mDevice->fn.AllocateMemory(mDevice->GetVkDevice(), &allocateInfo, nullptr, in AllocateResourceHeap()
154 DAWN_TRY_ASSIGN(subAllocation, mAllocatorsPerType[memoryType]->AllocateMemory( in Allocate()
/third_party/skia/tools/gpu/vk/
DVkTestHelper.h49 DECLARE_VK_PROC(AllocateMemory);
DVkTestHelper.cpp72 ACQUIRE_DEVICE_VK_PROC(AllocateMemory) in init()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dwinograd.h46 int AllocateMemory();
Dwinograd.cc182 int WinogradOpenCLKernel::AllocateMemory() { in AllocateMemory() function in mindspore::kernel::WinogradOpenCLKernel
205 int ret = AllocateMemory(); in SetConstArgs()
/third_party/skia/third_party/externals/dawn/src/dawn_native/d3d12/
DStagingBufferD3D12.cpp43 mDevice->AllocateMemory(D3D12_HEAP_TYPE_UPLOAD, resourceDescriptor, in Initialize()
DResourceAllocatorManagerD3D12.h63 ResultOrError<ResourceHeapAllocation> AllocateMemory(
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.h40 bool AllocateMemory(DeviceAddress *const &address, size_t size) const override;
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/
Ddevice_context.h59 virtual bool AllocateMemory(DeviceAddress *const &address, size_t size) const = 0;
/third_party/flutter/skia/third_party/externals/dawn/src/dawn_native/vulkan/
DMemoryAllocator.cpp109 if (mDevice->fn.AllocateMemory(mDevice->GetVkDevice(), &allocateInfo, nullptr, in Allocate()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.h42 bool AllocateMemory(DeviceAddress *const &address, size_t size) const override;
/third_party/skia/third_party/externals/swiftshader/tests/VulkanUnitTests/
DDevice.hpp108 VkResult AllocateMemory(size_t size, VkMemoryPropertyFlags flags, VkDeviceMemory *out) const;
/third_party/flutter/engine/flutter/vulkan/
Dvulkan_proc_table.h68 DEFINE_PROC(AllocateMemory);
/third_party/mesa3d/src/vulkan/wsi/
Dwsi_common_drm.c349 result = wsi->AllocateMemory(chain->device, &memory_info, in wsi_create_native_image()
520 result = wsi->AllocateMemory(chain->device, &prime_memory_info, in wsi_create_prime_image()
574 result = wsi->AllocateMemory(chain->device, &memory_info, in wsi_create_prime_image()
Dwsi_common.h179 WSI_CB(AllocateMemory);
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.h83 void AllocateMemory(const KernelGraph *kernel_graph) const;
/third_party/skia/third_party/externals/dawn/src/dawn_native/vulkan/external_memory/
DMemoryServiceZirconHandle.cpp127 DAWN_TRY(CheckVkSuccess(mDevice->fn.AllocateMemory(mDevice->GetVkDevice(), &allocateInfo, in ImportMemory()
DMemoryServiceOpaqueFD.cpp126 DAWN_TRY(CheckVkSuccess(mDevice->fn.AllocateMemory(mDevice->GetVkDevice(), &allocateInfo, in ImportMemory()
/third_party/flutter/skia/tests/
DVkYcbcrSamplerTest.cpp72 DECLARE_VK_PROC(AllocateMemory);
163 ACQUIRE_DEVICE_VK_PROC(AllocateMemory) in init()

123