Home
last modified time | relevance | path

Searched defs:memory_manager (Results 1 – 25 of 89) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/
H A Dxrt_state_ops.h200 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
241 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
345 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
395 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
465 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
509 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
562 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
655 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
689 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
737 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
H A Dxrt_execute_op.cc379 OpKernelContext* context, XRTMemoryManager* memory_manager, in ExecuteComputation()
399 OpKernelContext* context, const RefPtr<XRTMemoryManager>& memory_manager, in ExecuteComputation()
478 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
561 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
H A Dtpu_execute_op.cc172 OpKernelContext* context, XRTMemoryManager* memory_manager, in AllocateOutputTensors()
232 XRTMemoryManager* memory_manager, const TPUExecutableInfoProto& executable, in ExecuteTPUProgram()
342 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
429 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/
H A Dxrt_state.cc73 XRTMemoryManager* memory_manager, xla::Backend* backend, int device_ordinal, in AllocateScopedShapedBuffer()
171 const xla::LiteralBase& literal, XRTMemoryManager* memory_manager, in CreateAndTransfer()
198 const xla::Shape& shape, XRTMemoryManager* memory_manager, in CreateUninitialized()
293 XRTMemoryManager* memory_manager, xla::Backend* backend, in SwapIn()
327 XRTMemoryManager* memory_manager, xla::Backend* backend, in PinAndSwapIn()
456 XRTMemoryManager* memory_manager, xla::Backend* backend, int device_ordinal, in MakeTuple()
H A Dxrt_memory_manager.cc101 Status CompactAllocations(XRTMemoryManager* memory_manager, in CompactAllocations()
190 RefPtr<XRTMemoryManager> memory_manager) in WorkingSet()
212 XRTMemoryManager* memory_manager = nullptr; in Get() local
H A Dxrt_util.cc40 explicit ScopedHandles(RefPtr<XRTMemoryManager> memory_manager) in ScopedHandles()
349 XRTMemoryManager* memory_manager, in CreateExecuteOutput()
378 const RefPtr<XRTMemoryManager>& memory_manager, in ExecuteChained()
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEConvolutionLayer.cpp45 std::shared_ptr<IMemoryManager> memory_manager{}; member
54 NEConvolutionLayer::NEConvolutionLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEConvolutionLayer()
H A DNEDepthwiseConvolutionLayer.cpp62 …ternal::NEDepthwiseConvolutionLayerOptimizedInternal(std::shared_ptr<IMemoryManager> memory_manage… in NEDepthwiseConvolutionLayerOptimizedInternal()
295 …pthwiseConvolutionLayer::NEDepthwiseConvolutionLayer(std::shared_ptr<IMemoryManager> memory_manage… in NEDepthwiseConvolutionLayer()
H A DNEArgMinMaxLayer.cpp39 NEArgMinMaxLayer::NEArgMinMaxLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEArgMinMaxLayer()
H A DNENormalizationLayer.cpp39 NENormalizationLayer::NENormalizationLayer(std::shared_ptr<IMemoryManager> memory_manager) in NENormalizationLayer()
H A DNEFFT2D.cpp35 NEFFT2D::NEFFT2D(std::shared_ptr<IMemoryManager> memory_manager) in NEFFT2D()
H A DNEInstanceNormalizationLayer.cpp36 …anceNormalizationLayer::NEInstanceNormalizationLayer(std::shared_ptr<IMemoryManager> memory_manage… in NEInstanceNormalizationLayer()
H A DNEL2NormalizeLayer.cpp40 NEL2NormalizeLayer::NEL2NormalizeLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEL2NormalizeLayer()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLConvolutionLayer.cpp46 std::shared_ptr<IMemoryManager> memory_manager{}; member
55 CLConvolutionLayer::CLConvolutionLayer(std::shared_ptr<IMemoryManager> memory_manager) in CLConvolutionLayer()
H A DCLFFT2D.cpp37 CLFFT2D::CLFFT2D(std::shared_ptr<IMemoryManager> memory_manager) in CLFFT2D()
/aosp_15_r20/external/executorch/examples/mediatek/executor_runner/llama_runner/
H A DModelChunk.cpp55 std::unique_ptr<MemoryManager> memory_manager; member
565 auto& memory_manager = modelInstance->memory_manager; in CreateModelInstance() local
/aosp_15_r20/external/executorch/runtime/executor/
H A Dtensor_parser.h39 MemoryManager* memory_manager) { in parseListOptionalType()
H A Dtensor_parser_aten.cpp33 MemoryManager* memory_manager, in parseTensor()
H A Dtensor_parser_portable.cpp29 MemoryManager* memory_manager, in parseTensor()
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/backends/
H A DFusedConvolutionBatchNormalizationWithPostOpsFunction.h47 : _conv_layer(memory_manager), _fused_batch_norm_layer(), _fused_bias(), _is_prepared(false) in _conv_layer() argument
H A DFusedDepthwiseConvolutionBatchNormalizationFunction.h46 … : _depth_conv_layer(memory_manager), _fused_batch_norm_layer(), _fused_bias(), _is_prepared(false) in _depth_conv_layer() argument
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/
H A DMemoryGroup.h72 inline MemoryGroup::MemoryGroup(std::shared_ptr<IMemoryManager> memory_manager) noexcept in MemoryGroup()
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuDirectConv3d.cpp38 CpuDirectConv3d::CpuDirectConv3d(std::shared_ptr<IMemoryManager> memory_manager) in CpuDirectConv3d()
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CL/functions/
H A DCLDepthwiseConvolutionLayer.h113 void set_memory_group(std::shared_ptr<IMemoryManager> memory_manager) in set_memory_group()
/aosp_15_r20/external/executorch/examples/portable/executor_runner/
H A Dexecutor_runner.cpp147 MemoryManager memory_manager(&method_allocator, &planned_memory); in main() local

1234