Home
last modified time | relevance | path

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

1234

/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
508 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
561 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
654 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
688 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
736 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.cc377 OpKernelContext* context, XRTMemoryManager* memory_manager, in ExecuteComputation()
397 OpKernelContext* context, const RefPtr<XRTMemoryManager>& memory_manager, in ExecuteComputation()
476 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
559 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
/external/ComputeLibrary/src/runtime/GLES_COMPUTE/functions/
DGCNormalizationLayer.cpp36 GCNormalizationLayer::GCNormalizationLayer(std::shared_ptr<IMemoryManager> memory_manager) in GCNormalizationLayer()
DGCSoftmaxLayer.cpp32 GCSoftmaxLayer::GCSoftmaxLayer(std::shared_ptr<IMemoryManager> memory_manager) in GCSoftmaxLayer()
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLDepthwiseConvolutionLayer.h209 void set_memory_group(std::shared_ptr<IMemoryManager> memory_manager) in set_memory_group()
309 void set_memory_group(std::shared_ptr<IMemoryManager> memory_manager) in set_memory_group()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEGaussian5x5.cpp39 NEGaussian5x5::NEGaussian5x5(std::shared_ptr<IMemoryManager> memory_manager) in NEGaussian5x5()
DNEFastCorners.cpp44 NEFastCorners::NEFastCorners(std::shared_ptr<IMemoryManager> memory_manager) in NEFastCorners()
DNESobel7x7.cpp40 NESobel7x7::NESobel7x7(std::shared_ptr<IMemoryManager> memory_manager) in NESobel7x7()
DNENormalizationLayer.cpp39 NENormalizationLayer::NENormalizationLayer(std::shared_ptr<IMemoryManager> memory_manager) in NENormalizationLayer()
DNEArgMinMaxLayer.cpp40 NEArgMinMaxLayer::NEArgMinMaxLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEArgMinMaxLayer()
DNESobel5x5.cpp40 NESobel5x5::NESobel5x5(std::shared_ptr<IMemoryManager> memory_manager) in NESobel5x5()
DNEHOGGradient.cpp37 NEHOGGradient::NEHOGGradient(std::shared_ptr<IMemoryManager> memory_manager) // NOLINT in NEHOGGradient()
DNEFFT2D.cpp37 NEFFT2D::NEFFT2D(std::shared_ptr<IMemoryManager> memory_manager) in NEFFT2D()
DNEInstanceNormalizationLayer.cpp36 …anceNormalizationLayer::NEInstanceNormalizationLayer(std::shared_ptr<IMemoryManager> memory_manage… in NEInstanceNormalizationLayer()
DNEDirectConvolutionLayer.cpp39 NEDirectConvolutionLayer::NEDirectConvolutionLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEDirectConvolutionLayer()
DNEL2NormalizeLayer.cpp40 NEL2NormalizeLayer::NEL2NormalizeLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEL2NormalizeLayer()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLGaussian5x5.cpp40 CLGaussian5x5::CLGaussian5x5(std::shared_ptr<IMemoryManager> memory_manager) in CLGaussian5x5()
DCLSobel7x7.cpp38 CLSobel7x7::CLSobel7x7(std::shared_ptr<IMemoryManager> memory_manager) in CLSobel7x7()
DCLSobel5x5.cpp38 CLSobel5x5::CLSobel5x5(std::shared_ptr<IMemoryManager> memory_manager) in CLSobel5x5()
DCLHOGGradient.cpp35 CLHOGGradient::CLHOGGradient(std::shared_ptr<IMemoryManager> memory_manager) in CLHOGGradient()
/external/ComputeLibrary/arm_compute/runtime/
DMemoryGroup.h72 inline MemoryGroup::MemoryGroup(std::shared_ptr<IMemoryManager> memory_manager) noexcept in MemoryGroup()

1234