Home
last modified time | relevance | path

Searched refs:memory_manager (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h200 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
203 literal, memory_manager.get(), device_ref.backend(), in Compute()
207 output.scalar<int64>()() = memory_manager->Register(allocation); in Compute()
240 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
244 xla_shape_, memory_manager.get(), device_ref.backend(), in Compute()
248 output.scalar<int64>()() = memory_manager->Register(allocation); in Compute()
344 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
347 literal, memory_manager.get(), device_ref.backend(), in Compute()
351 output.scalar<int64>()() = memory_manager->Register(allocation); in Compute()
393 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
[all …]
Dtpu_execute_op.cc171 OpKernelContext* context, XRTMemoryManager* memory_manager, in AllocateOutputTensors() argument
183 return CreateExecuteOutput(context, memory_manager, std::move(output_tuple), in AllocateOutputTensors()
231 XRTMemoryManager* memory_manager, const TPUExecutableInfoProto& executable, in ExecuteTPUProgram() argument
243 return memory_manager->Run<xla::ExecutionOutput>( in ExecuteTPUProgram()
340 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
341 XRTMemoryManager::WorkingSet working_set(memory_manager); in DoWork()
357 context, node_context.get(), memory_manager.get(), executable, in DoWork()
365 context, memory_manager.get(), node_context.get(), stream, config_proto, in DoWork()
427 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
457 ExecuteTPUProgram(context, node_context.get(), memory_manager.get(), in DoWork()
[all …]
Dxrt_execute_op.cc375 OpKernelContext* context, XRTMemoryManager* memory_manager, in ExecuteComputation() argument
389 return memory_manager->Run<RefPtr<XRTTupleAllocation>>( in ExecuteComputation()
395 OpKernelContext* context, const RefPtr<XRTMemoryManager>& memory_manager, in ExecuteComputation() argument
401 XRTMemoryManager::WorkingSet working_set(memory_manager); in ExecuteComputation()
406 return ExecuteComputation(context, memory_manager.get(), device_ref, in ExecuteComputation()
474 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
489 ExecuteComputation(context, memory_manager, &device_ref, executable, in DoWork()
493 return CreateExecuteOutput(context, memory_manager.get(), in DoWork()
557 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
569 context, memory_manager.get(), &device_ref, executable, input_tuples, in DoWork()
[all …]
/external/ImageMagick/coders/
Djxl.c109 static inline void JXLSetMemoryManager(JxlMemoryManager *memory_manager, in JXLSetMemoryManager() argument
114 memory_manager->opaque=memory_manager_info; in JXLSetMemoryManager()
115 memory_manager->alloc=JXLAcquireMemory; in JXLSetMemoryManager()
116 memory_manager->free=JXLRelinquishMemory; in JXLSetMemoryManager()
195 memory_manager; in ReadJXLImage() local
230 JXLSetMemoryManager(&memory_manager,&memory_manager_info,image,exception); in ReadJXLImage()
231 decoder=JxlDecoderCreate(&memory_manager); in ReadJXLImage()
510 memory_manager; in WriteJXLImage() local
544 JXLSetMemoryManager(&memory_manager,&memory_manager_info,image,exception); in WriteJXLImage()
545 encoder=JxlEncoderCreate(&memory_manager); in WriteJXLImage()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.h84 XRTMemoryManager* memory_manager,
92 XRTMemoryManager* memory_manager,
139 static Status MakeTuple(XRTMemoryManager* memory_manager,
160 xla::StatusOr<bool> SwapIn(XRTMemoryManager* memory_manager,
167 xla::StatusOr<bool> PinAndSwapIn(XRTMemoryManager* memory_manager,
Dxrt_memory_manager.cc101 Status CompactAllocations(XRTMemoryManager* memory_manager, in CompactAllocations() argument
134 auto swap_result_or = it->tuple->SwapIn(memory_manager, backend); in CompactAllocations()
188 RefPtr<XRTMemoryManager> memory_manager) in WorkingSet() argument
189 : memory_manager_(std::move(memory_manager)) {} in WorkingSet()
209 XRTMemoryManager* memory_manager = nullptr; in Get() local
211 *container, *name, &memory_manager, [](XRTMemoryManager** ret) { in Get()
215 return memory_manager; in Get()
Dxrt_state.cc73 XRTMemoryManager* memory_manager, xla::Backend* backend, int device_ordinal, in AllocateScopedShapedBuffer() argument
103 memory_manager->Allocate(backend, device_ordinal, size)); in AllocateScopedShapedBuffer()
170 const xla::LiteralBase& literal, XRTMemoryManager* memory_manager, in CreateAndTransfer() argument
175 TF_RETURN_IF_ERROR(AllocateScopedShapedBuffer(memory_manager, backend, in CreateAndTransfer()
198 const xla::Shape& shape, XRTMemoryManager* memory_manager, in CreateUninitialized() argument
203 memory_manager, backend, device_ordinal, shape, &scoped_buffer)); in CreateUninitialized()
292 xla::StatusOr<bool> XRTTupleAllocation::SwapIn(XRTMemoryManager* memory_manager, in SwapIn() argument
308 AllocateScopedShapedBuffer(memory_manager, backend, device_ordinal(), in SwapIn()
327 XRTMemoryManager* memory_manager, xla::Backend* backend) { in PinAndSwapIn() argument
329 return SwapIn(memory_manager, backend); in PinAndSwapIn()
[all …]
Dxrt_util.cc40 explicit ScopedHandles(RefPtr<XRTMemoryManager> memory_manager) in ScopedHandles() argument
41 : memory_manager_(std::move(memory_manager)) {} in ScopedHandles()
346 XRTMemoryManager* memory_manager, in CreateExecuteOutput() argument
361 output_tensor->vec<int64>()(i) = memory_manager->Register(suballocation); in CreateExecuteOutput()
368 memory_manager->Register(std::move(output_tuple)); in CreateExecuteOutput()
374 const RefPtr<XRTMemoryManager>& memory_manager, in ExecuteChained() argument
388 ScopedHandles outputs(memory_manager); in ExecuteChained()
389 ScopedHandles results(memory_manager); in ExecuteChained()
401 XRTMemoryManager::WorkingSet working_set(memory_manager); in ExecuteChained()
Dxrt_util.h97 XRTMemoryManager* memory_manager,
108 const RefPtr<XRTMemoryManager>& memory_manager,
Dxrt_memory_manager.h53 explicit WorkingSet(RefPtr<XRTMemoryManager> memory_manager);
/external/libcxx/utils/google-benchmark/src/
Dbenchmark_runner.cc58 MemoryManager* memory_manager = nullptr; variable
314 if (memory_manager != nullptr) { in DoOneRepetition()
318 memory_manager->Start(); in DoOneRepetition()
325 memory_manager->Stop(&memory_result); in DoOneRepetition()
Dbenchmark_runner.h33 extern MemoryManager* memory_manager;
Dbenchmark.cc405 internal::memory_manager = manager; in RegisterMemoryManager()
/external/google-benchmark/src/
Dbenchmark_runner.cc58 MemoryManager* memory_manager = nullptr; variable
325 if (memory_manager != nullptr) { in DoOneRepetition()
329 memory_manager->Start(); in DoOneRepetition()
336 memory_manager->Stop(&memory_result); in DoOneRepetition()
Dbenchmark_runner.h33 extern MemoryManager* memory_manager;
Dbenchmark.cc411 internal::memory_manager = manager; in RegisterMemoryManager()
/external/llvm-project/libcxx/utils/google-benchmark/src/
Dbenchmark_runner.cc58 MemoryManager* memory_manager = nullptr; variable
314 if (memory_manager != nullptr) { in DoOneRepetition()
318 memory_manager->Start(); in DoOneRepetition()
325 memory_manager->Stop(&memory_result); in DoOneRepetition()
Dbenchmark_runner.h33 extern MemoryManager* memory_manager;
Dbenchmark.cc405 internal::memory_manager = manager; in RegisterMemoryManager()
/external/libcxx/utils/google-benchmark/include/benchmark/
Dbenchmark.h285 void RegisterMemoryManager(MemoryManager* memory_manager);
/external/llvm-project/libcxx/utils/google-benchmark/include/benchmark/
Dbenchmark.h285 void RegisterMemoryManager(MemoryManager* memory_manager);
/external/brotli/c/enc/
Dencode.c1207 MemoryManager memory_manager; in BrotliCompressBufferQuality10() local
1208 MemoryManager* m = &memory_manager; in BrotliCompressBufferQuality10()
/external/google-benchmark/include/benchmark/
Dbenchmark.h284 void RegisterMemoryManager(MemoryManager* memory_manager);