/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
D | xrt_state_ops.h | 200 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 …]
|
D | tpu_execute_op.cc | 171 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 …]
|
D | xrt_execute_op.cc | 375 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/ |
D | jxl.c | 109 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/ |
D | xrt_state.h | 84 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,
|
D | xrt_memory_manager.cc | 101 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()
|
D | xrt_state.cc | 73 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 …]
|
D | xrt_util.cc | 40 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()
|
D | xrt_util.h | 97 XRTMemoryManager* memory_manager, 108 const RefPtr<XRTMemoryManager>& memory_manager,
|
D | xrt_memory_manager.h | 53 explicit WorkingSet(RefPtr<XRTMemoryManager> memory_manager);
|
/external/libcxx/utils/google-benchmark/src/ |
D | benchmark_runner.cc | 58 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()
|
D | benchmark_runner.h | 33 extern MemoryManager* memory_manager;
|
D | benchmark.cc | 405 internal::memory_manager = manager; in RegisterMemoryManager()
|
/external/google-benchmark/src/ |
D | benchmark_runner.cc | 58 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()
|
D | benchmark_runner.h | 33 extern MemoryManager* memory_manager;
|
D | benchmark.cc | 411 internal::memory_manager = manager; in RegisterMemoryManager()
|
/external/llvm-project/libcxx/utils/google-benchmark/src/ |
D | benchmark_runner.cc | 58 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()
|
D | benchmark_runner.h | 33 extern MemoryManager* memory_manager;
|
D | benchmark.cc | 405 internal::memory_manager = manager; in RegisterMemoryManager()
|
/external/libcxx/utils/google-benchmark/include/benchmark/ |
D | benchmark.h | 285 void RegisterMemoryManager(MemoryManager* memory_manager);
|
/external/llvm-project/libcxx/utils/google-benchmark/include/benchmark/ |
D | benchmark.h | 285 void RegisterMemoryManager(MemoryManager* memory_manager);
|
/external/brotli/c/enc/ |
D | encode.c | 1207 MemoryManager memory_manager; in BrotliCompressBufferQuality10() local 1208 MemoryManager* m = &memory_manager; in BrotliCompressBufferQuality10()
|
/external/google-benchmark/include/benchmark/ |
D | benchmark.h | 284 void RegisterMemoryManager(MemoryManager* memory_manager);
|