Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xrt/kernels/
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
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
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
/external/ImageMagick/coders/
Djxl.c109 static inline void JXLSetMemoryManager(JxlMemoryManager *memory_manager, in JXLSetMemoryManager()
195 memory_manager; in ReadJXLImage() local
510 memory_manager; in WriteJXLImage() local
/external/tensorflow/tensorflow/compiler/xrt/
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()
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
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()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEConvolutionLayer.cpp45 std::shared_ptr<IMemoryManager> memory_manager{}; member
54 NEConvolutionLayer::NEConvolutionLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEConvolutionLayer()
DNEArgMinMaxLayer.cpp39 NEArgMinMaxLayer::NEArgMinMaxLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEArgMinMaxLayer()
DNENormalizationLayer.cpp39 NENormalizationLayer::NENormalizationLayer(std::shared_ptr<IMemoryManager> memory_manager) in NENormalizationLayer()
DNEFFT2D.cpp35 NEFFT2D::NEFFT2D(std::shared_ptr<IMemoryManager> memory_manager) in NEFFT2D()
DNEInstanceNormalizationLayer.cpp36 …anceNormalizationLayer::NEInstanceNormalizationLayer(std::shared_ptr<IMemoryManager> memory_manage… in NEInstanceNormalizationLayer()
DNEL2NormalizeLayer.cpp40 NEL2NormalizeLayer::NEL2NormalizeLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEL2NormalizeLayer()
DNEPooling3dLayer.cpp46 NEPooling3dLayer::NEPooling3dLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEPooling3dLayer()
DNERNNLayer.cpp39 NERNNLayer::NERNNLayer(std::shared_ptr<IMemoryManager> memory_manager) in NERNNLayer()
/external/ComputeLibrary/arm_compute/graph/backends/
DFusedConvolutionBatchNormalizationWithPostOpsFunction.h47 : _conv_layer(memory_manager), _fused_batch_norm_layer(), _fused_bias(), _is_prepared(false) in _conv_layer() argument
DFusedDepthwiseConvolutionBatchNormalizationFunction.h46 … : _depth_conv_layer(memory_manager), _fused_batch_norm_layer(), _fused_bias(), _is_prepared(false) in _depth_conv_layer() argument
DFusedConvolutionBatchNormalizationFunction.h46 : _conv_layer(memory_manager), _fused_batch_norm_layer(), _fused_bias(), _is_prepared(false) in _conv_layer() argument
/external/ComputeLibrary/arm_compute/runtime/
DMemoryGroup.h72 inline MemoryGroup::MemoryGroup(std::shared_ptr<IMemoryManager> memory_manager) noexcept in MemoryGroup()
/external/ComputeLibrary/src/cpu/operators/
DCpuDirectConv3d.cpp38 CpuDirectConv3d::CpuDirectConv3d(std::shared_ptr<IMemoryManager> memory_manager) in CpuDirectConv3d()
DCpuDirectConv2d.cpp38 CpuDirectConv2d::CpuDirectConv2d(std::shared_ptr<IMemoryManager> memory_manager) in CpuDirectConv2d()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLConvolutionLayer.cpp46 std::shared_ptr<IMemoryManager> memory_manager{}; member
55 CLConvolutionLayer::CLConvolutionLayer(std::shared_ptr<IMemoryManager> memory_manager) in CLConvolutionLayer()
DCLFFT2D.cpp37 CLFFT2D::CLFFT2D(std::shared_ptr<IMemoryManager> memory_manager) in CLFFT2D()
DCLL2NormalizeLayer.cpp45 CLL2NormalizeLayer::CLL2NormalizeLayer(std::shared_ptr<IMemoryManager> memory_manager) in CLL2NormalizeLayer()
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLDepthwiseConvolutionLayer.h113 void set_memory_group(std::shared_ptr<IMemoryManager> memory_manager) in set_memory_group()

123