| /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()
208 output.scalar<int64>()() = memory_manager->Register(allocation); in Compute()
241 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
244 xla_shape_, memory_manager.get(), in Compute()
249 output.scalar<int64>()() = memory_manager->Register(allocation); in Compute()
345 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
348 literal, memory_manager.get(), device_ref.backend(), in Compute()
353 output.scalar<int64>()() = memory_manager->Register(allocation); in Compute()
395 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute() local
[all …]
|
| D | tpu_execute_op.cc | 172 OpKernelContext* context, XRTMemoryManager* memory_manager, in AllocateOutputTensors() argument
184 return CreateExecuteOutput(context, memory_manager, std::move(output_tuple), in AllocateOutputTensors()
232 XRTMemoryManager* memory_manager, const TPUExecutableInfoProto& executable, in ExecuteTPUProgram() argument
244 return memory_manager->Run<xla::ExecutionOutput>( in ExecuteTPUProgram()
342 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
343 XRTMemoryManager::WorkingSet working_set(memory_manager); in DoWork()
359 context, node_context.get(), memory_manager.get(), executable, in DoWork()
367 context, memory_manager.get(), node_context.get(), stream, config_proto, in DoWork()
429 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
459 ExecuteTPUProgram(context, node_context.get(), memory_manager.get(), in DoWork()
[all …]
|
| D | xrt_execute_op.cc | 377 OpKernelContext* context, XRTMemoryManager* memory_manager, in ExecuteComputation() argument
391 return memory_manager->Run<RefPtr<XRTTupleAllocation>>( in ExecuteComputation()
397 OpKernelContext* context, const RefPtr<XRTMemoryManager>& memory_manager, in ExecuteComputation() argument
403 XRTMemoryManager::WorkingSet working_set(memory_manager); in ExecuteComputation()
408 return ExecuteComputation(context, memory_manager.get(), device_ref, in ExecuteComputation()
476 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
491 ExecuteComputation(context, memory_manager, &device_ref, executable, in DoWork()
495 return CreateExecuteOutput(context, memory_manager.get(), in DoWork()
559 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork() local
571 context, memory_manager.get(), &device_ref, executable, input_tuples, in DoWork()
[all …]
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLDepthwiseConvolutionLayer.h | 48 CLDepthwiseConvolutionLayer(std::shared_ptr<IMemoryManager> memory_manager = nullptr);
146 … CLDepthwiseConvolutionLayerInternal3x3(std::shared_ptr<IMemoryManager> memory_manager = nullptr);
209 void set_memory_group(std::shared_ptr<IMemoryManager> memory_manager) in set_memory_group() argument
211 _memory_group = MemoryGroup(std::move(memory_manager)); in set_memory_group()
246 … CLDepthwiseConvolutionLayerGeneric(std::shared_ptr<IMemoryManager> memory_manager = nullptr);
309 void set_memory_group(std::shared_ptr<IMemoryManager> memory_manager) in set_memory_group() argument
311 _memory_group = MemoryGroup(std::move(memory_manager)); in set_memory_group()
|
| /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/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEFFT2D.cpp | 37 NEFFT2D::NEFFT2D(std::shared_ptr<IMemoryManager> memory_manager) in NEFFT2D() argument
38 …: _memory_group(memory_manager), _first_pass_func(memory_manager), _second_pass_func(memory_manage… in NEFFT2D()
|
| D | NEArgMinMaxLayer.cpp | 40 NEArgMinMaxLayer::NEArgMinMaxLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEArgMinMaxLayer() argument
43 ARM_COMPUTE_UNUSED(memory_manager); in NEArgMinMaxLayer()
|
| D | NEGaussian5x5.cpp | 39 NEGaussian5x5::NEGaussian5x5(std::shared_ptr<IMemoryManager> memory_manager) in NEGaussian5x5() argument
40 …: _memory_group(std::move(memory_manager)), _kernel_hor(), _kernel_vert(), _tmp(), _border_handler… in NEGaussian5x5()
|
| D | NENormalizationLayer.cpp | 39 NENormalizationLayer::NENormalizationLayer(std::shared_ptr<IMemoryManager> memory_manager) in NENormalizationLayer() argument
40 : _memory_group(std::move(memory_manager)), _norm_kernel(), _multiply_f(), _input_squared() in NENormalizationLayer()
|
| D | NEL2NormalizeLayer.cpp | 40 NEL2NormalizeLayer::NEL2NormalizeLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEL2NormalizeLayer() argument
41 : _memory_group(std::move(memory_manager)), _reduce_func(), _normalize_kernel(), _sumsq() in NEL2NormalizeLayer()
|
| D | NERNNLayer.cpp | 51 NERNNLayer::NERNNLayer(std::shared_ptr<IMemoryManager> memory_manager) in NERNNLayer() argument
52 …: _memory_group(std::move(memory_manager)), _gemm_state_f(), _add_f(), _activation(), _fully_conne… in NERNNLayer()
|
| D | NEHOGGradient.cpp | 37 NEHOGGradient::NEHOGGradient(std::shared_ptr<IMemoryManager> memory_manager) // NOLINT in NEHOGGradient() argument
38 : _memory_group(std::move(memory_manager)), in NEHOGGradient()
|
| D | NEInstanceNormalizationLayer.cpp | 36 …nceNormalizationLayer::NEInstanceNormalizationLayer(std::shared_ptr<IMemoryManager> memory_manager) in NEInstanceNormalizationLayer() argument
37 …: _memory_group(std::move(memory_manager)), _normalization_kernel(), _is_nchw(false), _permute_inp… in NEInstanceNormalizationLayer()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLFFT2D.cpp | 35 CLFFT2D::CLFFT2D(std::shared_ptr<IMemoryManager> memory_manager) in CLFFT2D() argument
36 …: _memory_group(memory_manager), _first_pass_func(memory_manager), _second_pass_func(memory_manage… in CLFFT2D()
|
| D | CLGaussian5x5.cpp | 40 CLGaussian5x5::CLGaussian5x5(std::shared_ptr<IMemoryManager> memory_manager) in CLGaussian5x5() argument
41 : _memory_group(std::move(memory_manager)), in CLGaussian5x5()
|
| D | CLL2NormalizeLayer.cpp | 44 CLL2NormalizeLayer::CLL2NormalizeLayer(std::shared_ptr<IMemoryManager> memory_manager) in CLL2NormalizeLayer() argument
45 : _memory_group(std::move(memory_manager)), in CLL2NormalizeLayer()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.h | 84 XRTMemoryManager* memory_manager,
93 XRTMemoryManager* memory_manager,
143 static Status MakeTuple(XRTMemoryManager* memory_manager,
165 xla::StatusOr<bool> SwapIn(XRTMemoryManager* memory_manager,
173 xla::StatusOr<bool> PinAndSwapIn(XRTMemoryManager* memory_manager,
|
| D | xrt_state.cc | 73 XRTMemoryManager* memory_manager, xla::Backend* backend, int device_ordinal, in AllocateScopedShapedBuffer() argument
104 memory_manager->Allocate(backend, device_ordinal, size, allocator)); in AllocateScopedShapedBuffer()
171 const xla::LiteralBase& literal, XRTMemoryManager* memory_manager, in CreateAndTransfer() argument
176 TF_RETURN_IF_ERROR(AllocateScopedShapedBuffer(memory_manager, backend, in CreateAndTransfer()
198 const xla::Shape& shape, XRTMemoryManager* memory_manager, in CreateUninitialized() argument
202 TF_RETURN_IF_ERROR(AllocateScopedShapedBuffer(memory_manager, backend, in CreateUninitialized()
293 XRTMemoryManager* memory_manager, xla::Backend* backend, in SwapIn() argument
309 AllocateScopedShapedBuffer(memory_manager, backend, device_ordinal(), in SwapIn()
327 XRTMemoryManager* memory_manager, xla::Backend* backend, in PinAndSwapIn() argument
330 return SwapIn(memory_manager, backend, allocator); in PinAndSwapIn()
[all …]
|
| D | xrt_memory_manager.cc | 101 Status CompactAllocations(XRTMemoryManager* memory_manager, in CompactAllocations() argument
136 it->tuple->SwapIn(memory_manager, backend, allocator); in CompactAllocations()
190 RefPtr<XRTMemoryManager> memory_manager) in WorkingSet() argument
191 : memory_manager_(std::move(memory_manager)) {} in WorkingSet()
212 XRTMemoryManager* memory_manager = nullptr; in Get() local
214 *container, *name, &memory_manager, [](XRTMemoryManager** ret) { in Get()
218 return memory_manager; in Get()
|
| D | xrt_util.cc | 40 explicit ScopedHandles(RefPtr<XRTMemoryManager> memory_manager) in ScopedHandles() argument
41 : memory_manager_(std::move(memory_manager)) {} in ScopedHandles()
348 XRTMemoryManager* memory_manager, in CreateExecuteOutput() argument
363 output_tensor->vec<int64>()(i) = memory_manager->Register(suballocation); in CreateExecuteOutput()
370 memory_manager->Register(std::move(output_tuple)); in CreateExecuteOutput()
376 const RefPtr<XRTMemoryManager>& memory_manager, in ExecuteChained() argument
391 ScopedHandles outputs(memory_manager); in ExecuteChained()
392 ScopedHandles results(memory_manager); in ExecuteChained()
404 XRTMemoryManager::WorkingSet working_set(memory_manager); in ExecuteChained()
|
| /external/ComputeLibrary/src/runtime/GLES_COMPUTE/functions/ |
| D | GCNormalizationLayer.cpp | 36 GCNormalizationLayer::GCNormalizationLayer(std::shared_ptr<IMemoryManager> memory_manager) in GCNormalizationLayer() argument
37 …: _memory_group(std::move(memory_manager)), _squared_input(), _norm_kernel(), _multiply_kernel(), … in GCNormalizationLayer()
|
| D | GCSoftmaxLayer.cpp | 32 GCSoftmaxLayer::GCSoftmaxLayer(std::shared_ptr<IMemoryManager> memory_manager) in GCSoftmaxLayer() argument
33 …: _memory_group(std::move(memory_manager)), _max_kernel(), _shift_exp_sum_kernel(), _norm_kernel()… in GCSoftmaxLayer()
|
| /external/ComputeLibrary/arm_compute/graph/backends/ |
| D | FusedConvolutionBatchNormalizationFunction.h | 45 …FusedConvolutionBatchNormalizationFunction(std::shared_ptr<IMemoryManager> memory_manager = nullpt…
46 : _conv_layer(memory_manager), _fused_batch_norm_layer(), _fused_bias(), _is_prepared(false) in _conv_layer() argument
|
| D | FusedDepthwiseConvolutionBatchNormalizationFunction.h | 45 …wiseConvolutionBatchNormalizationFunction(std::shared_ptr<IMemoryManager> memory_manager = nullptr)
46 … : _depth_conv_layer(memory_manager), _fused_batch_norm_layer(), _fused_bias(), _is_prepared(false) in _depth_conv_layer() argument
|
| /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()
|