Home
last modified time | relevance | path

Searched refs:GetAllocator (Results 1 – 25 of 70) sorted by relevance

123

/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_subgraph.cc319 auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(tensor, tensor.DataSize()); in Prepare()
345 auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(tensor, tensor.DataSize()); in Prepare()
389 …auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(trt_in_tensor_name_[i], inputs_[i].Dat… in ReSize()
415 …auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(trt_out_tensor_name_[i], outputs_[i].D… in ReSize()
433 if (runtime_->GetAllocator()->GetMemIsValid(trt_in_tensor_name_[i])) { in Execute()
437 …int ret = runtime_->GetAllocator()->SyncMemInHostAndDevice(inputs_[i], trt_in_tensor_name_[i], tru… in Execute()
442 runtime_->GetAllocator()->MarkMemValid(trt_in_tensor_name_[i], true); in Execute()
469 runtime_->GetAllocator()->MarkMemValid(trt_out_tensor_name_[i], true); in Execute()
470 …int sync_ret = runtime_->GetAllocator()->SyncMemInHostAndDevice(outputs_[i], trt_out_tensor_name_[… in Execute()
475 runtime_->GetAllocator()->MarkMemValid(trt_out_tensor_name_[i], false); in Execute()
[all …]
Dtensorrt_runtime.h56 TensorRTAllocator *GetAllocator() { return this->allocator_; } in GetAllocator() function
/third_party/flutter/engine/flutter/shell/platform/common/cpp/client_wrapper/
Djson_method_codec.cc78 auto& allocator = message.GetAllocator(); in EncodeMethodCallInternal()
104 result_value.CopyFrom(*result, envelope.GetAllocator()); in EncodeSuccessEnvelopeInternal()
106 envelope.PushBack(result_value, envelope.GetAllocator()); in EncodeSuccessEnvelopeInternal()
122 auto& allocator = envelope.GetAllocator(); in EncodeErrorEnvelopeInternal()
/third_party/mindspore/mindspore/lite/src/runtime/gpu/opencl/
Dopencl_runtime_wrapper.cc92 return ocl_runtime->GetAllocator()->MapBuffer(host_ptr, flags, nullptr, sync); in MapBuffer()
98 if (ocl_runtime->GetAllocator()->UnmapBuffer(host_ptr, nullptr) == RET_OK) { in UnmapBuffer()
125 std::shared_ptr<Allocator> OpenCLRuntimeWrapper::GetAllocator() { in GetAllocator() function in mindspore::registry::opencl::OpenCLRuntimeWrapper
128 return ocl_runtime->GetAllocator(); in GetAllocator()
Dopencl_executor.h30 OpenCLExecutor() : Executor() { allocator_ = ocl_runtime_.GetInstance()->GetAllocator().get(); } in OpenCLExecutor()
/third_party/flutter/engine/flutter/runtime/
Dservice_protocol.cc90 document.AddMember("code", -32000, document.GetAllocator()); in WriteServerErrorResponse()
92 message_value.SetString(message, document.GetAllocator()); in WriteServerErrorResponse()
93 document.AddMember("message", message_value, document.GetAllocator()); in WriteServerErrorResponse()
260 auto& allocator = response.GetAllocator(); in HandleListViewsMethod()
/third_party/mindspore/tests/ut/cpp/dataset/
Darena_test.cc63 std::vector<int, Allocator<int>> w(v, SystemPool::GetAllocator<int>()); in TEST_F()
67 std::vector<int, Allocator<int>> s(std::move(v), SystemPool::GetAllocator<int>()); in TEST_F()
/third_party/mindspore/mindspore/lite/src/cxx_api/
Dconverters.cc95 if (cpu_context->GetAllocator() == nullptr) { in Convert()
98 …ret = AddCpuDevice(cpu_context->GetAllocator(), context->GetThreadAffinityMode(), cpu_context->Get… in Convert()
103 … gpu_context->GetProviderDevice(), gpu_context->GetAllocator(), inner_context.get()); in Convert()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dfill.cc31 auto allocator_ = ocl_runtime_->GetAllocator(); in RunFill()
56 auto allocator_ = ocl_runtime_->GetAllocator(); in RunShape()
Dbatchnorm.cc109 auto allocator = ocl_runtime_->GetAllocator(); in UnmapBuffer()
126 auto allocator = ocl_runtime_->GetAllocator(); in MapBuffer()
144 auto allocator = ocl_runtime_->GetAllocator(); in Initweight()
Dmatmul.cc108 auto allocator = ocl_runtime_->GetAllocator(); in PadWeight()
179 auto allocator = ocl_runtime_->GetAllocator(); in InitWeights()
210 auto allocator = ocl_runtime_->GetAllocator(); in InitBias()
Dgather.cc152 auto allocator = ocl_runtime_->GetAllocator(); in ConvertTensorToweight()
199 auto allocator = ocl_runtime_->GetAllocator(); in InitWeights()
Dsplit.cc33 auto allocator_ = ocl_runtime_->GetAllocator(); in RunAxis0()
106 auto allocator = ocl_runtime_->GetAllocator(); in AlignSplitSizes()
/third_party/flutter/engine/flutter/shell/common/
Dshell.cc1105 auto& allocator = response.GetAllocator(); in ServiceProtocolParameterError()
1119 auto& allocator = response.GetAllocator(); in ServiceProtocolFailureError()
1135 auto& allocator = response.GetAllocator(); in OnServiceProtocolScreenshot()
1156 auto& allocator = response.GetAllocator(); in OnServiceProtocolScreenshotSKP()
1216 auto& allocator = response.GetAllocator(); in OnServiceProtocolRunInView()
1248 response.AddMember("type", "Success", response.GetAllocator()); in OnServiceProtocolFlushUIThreadTasks()
1258 response.GetAllocator()); in OnServiceProtocolGetDisplayRefreshRate()
1274 auto& allocator = response.GetAllocator(); in OnServiceProtocolSetAssetBundlePath()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dfill_tests.cc39 auto allocator = runtime->GetAllocator(); in TEST_F()
110 auto allocator = runtime->GetAllocator(); in TEST_F()
Dcommon.cc183 ocl_runtime->GetAllocator()->MapBuffer(outputs[i]->data(), CL_MAP_READ, nullptr, true); in TestMain()
186 ocl_runtime->GetAllocator()->UnmapBuffer(outputs[i]->data()); in TestMain()
348 ocl_runtime->GetAllocator()->MapBuffer(output.data(), CL_MAP_READ, nullptr, true); in TestMain()
350 ocl_runtime->GetAllocator()->UnmapBuffer(output.data()); in TestMain()
Dcast_tests.cc44 auto allocator = ocl_runtime->GetAllocator(); in TEST_F()
154 auto allocator = ocl_runtime->GetAllocator(); in TEST_F()
/third_party/flutter/engine/flutter/shell/platform/glfw/
Dkey_event_handler.cc42 auto& allocator = event.GetAllocator(); in KeyboardHook()
Dplatform_handler.cc63 rapidjson::Document::AllocatorType& allocator = document.GetAllocator(); in HandleMethodCall()
/third_party/flutter/engine/flutter/shell/platform/windows/
Dkey_event_handler.cc45 auto& allocator = event.GetAllocator(); in KeyboardHook()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/util/
Dsystem_pool.h73 static Allocator<T> GetAllocator() { in GetAllocator() function
/third_party/abseil-cpp/absl/container/internal/
Dinlined_vector.h182 AllocatorTraits::deallocate(GetAllocator(), GetData(), GetCapacity()); in ~AllocationTransaction()
189 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function
195 GetData() = AllocatorTraits::allocate(GetAllocator(), capacity); in Allocate()
222 inlined_vector_internal::DestroyElements(std::addressof(GetAllocator()), in ~ConstructionTransaction()
230 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function
237 inlined_vector_internal::ConstructElements(std::addressof(GetAllocator()), in Construct()
/third_party/mindspore/mindspore/lite/include/registry/
Dopencl_runtime_wrapper.h108 std::shared_ptr<Allocator> GetAllocator();
/third_party/skia/third_party/externals/abseil-cpp/absl/container/internal/
Dinlined_vector.h206 AllocatorTraits::deallocate(GetAllocator(), GetData(), GetCapacity()); in ~AllocationTransaction()
213 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function
219 GetData() = AllocatorTraits::allocate(GetAllocator(), capacity); in Allocate()
246 inlined_vector_internal::DestroyElements(std::addressof(GetAllocator()), in ~ConstructionTransaction()
254 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function
261 inlined_vector_internal::ConstructElements(std::addressof(GetAllocator()), in Construct()
/third_party/mindspore/mindspore/lite/examples/runtime_gpu_extend/src/
Dcustom_add_kernel_gpu.cc72 auto allocator = opencl_runtime_.GetAllocator(); in Prepare()
219 auto allocator = opencl_runtime_.GetAllocator(); in FreeWeight()

123