Home
last modified time | relevance | path

Searched refs:memory_allocator (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_allocations.h40 se::DeviceMemoryAllocator* memory_allocator) in BufferAllocations() argument
43 memory_allocator_(memory_allocator) {} in BufferAllocations()
50 se::DeviceMemoryAllocator* memory_allocator() const { in memory_allocator() function
Dgpu_executable.cc306 se::DeviceMemoryAllocator* const memory_allocator, int device_ordinal, in BufferForAllocation() argument
346 memory_allocator->Allocate(device_ordinal, buffer_size)); in BufferForAllocation()
377 se::DeviceMemoryAllocator* const memory_allocator, in GenerateBufferAllocations() argument
390 BufferForAllocation(arguments, globals, allocation, memory_allocator, in GenerateBufferAllocations()
395 return {{buffers, executor->device_ordinal(), memory_allocator}}; in GenerateBufferAllocations()
421 se::DeviceMemoryAllocator* const memory_allocator = run_options->allocator(); in ExecuteAsyncOnStreamImpl() local
424 !memory_allocator->AllowsAsynchronousDeallocation(); in ExecuteAsyncOnStreamImpl()
438 ExecutionOutput result(/*on_device_shape=*/output_shape_, memory_allocator, in ExecuteAsyncOnStreamImpl()
443 memory_allocator, executor)); in ExecuteAsyncOnStreamImpl()
522 memory_allocator->Allocate(device_ordinal, allocation_size)); in ExecuteAsyncOnStreamImpl()
Dgpu_executable.h162 se::DeviceMemoryAllocator* const memory_allocator,
169 se::DeviceMemoryAllocator* const memory_allocator, int device_ordinal,
Dfft_thunk.cc32 int device_ordinal, se::DeviceMemoryAllocator* memory_allocator) in FftScratchAllocator() argument
33 : device_ordinal_(device_ordinal), memory_allocator_(memory_allocator) {} in FftScratchAllocator()
127 buffer_allocations.memory_allocator()); in ExecuteOnStream()
Dcudnn_fused_conv_rewriter_test.cc41 backend().default_stream_executor(), backend().memory_allocator()) in GetOptimizedHlo()
337 backend().default_stream_executor(), backend().memory_allocator()) in TEST_F()
Dfft_thunk.h42 se::DeviceMemoryAllocator* memory_allocator);
/external/libchrome/base/metrics/
Dpersistent_histogram_allocator_unittest.cc40 allocator_ = GlobalHistogramAllocator::Get()->memory_allocator(); in CreatePersistentHistogramAllocator()
170 GlobalHistogramAllocator::Get()->memory_allocator()->Name()); in TEST_F()
176 GlobalHistogramAllocator::Get()->memory_allocator()->Name()); in TEST_F()
182 GlobalHistogramAllocator::Get()->memory_allocator()->Name()); in TEST_F()
261 const_cast<void*>(new_allocator->memory_allocator()->data()), in TEST_F()
262 new_allocator->memory_allocator()->size(), 0, 0, "", false)); in TEST_F()
308 const_cast<void*>(new_allocator->memory_allocator()->data()), in TEST_F()
309 new_allocator->memory_allocator()->size(), 0, 0, "", false)); in TEST_F()
Dpersistent_histogram_allocator.cc259 : allocator_(allocator), memory_iter_(allocator->memory_allocator()) {} in Iterator()
914 PersistentMemoryAllocator* memory_allocator = in ReleaseForTesting() local
915 histogram_allocator->memory_allocator(); in ReleaseForTesting()
920 PersistentMemoryAllocator::Iterator iter(memory_allocator); in ReleaseForTesting()
964 memory_allocator()->SetMemoryState(PersistentMemoryAllocator::MEMORY_DELETED); in DeletePersistentLocation()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc94 se::DeviceMemoryAllocator* memory_allocator, int device_ordinal) { in MemoryForAllocation() argument
115 memory_allocator->Allocate(device_ordinal, buffer_size)); in MemoryForAllocation()
128 se::DeviceMemoryAllocator* memory_allocator, int device_ordinal, in CreateBufferTable() argument
138 buffers[i], MemoryForAllocation(allocation, arguments, memory_allocator, in CreateBufferTable()
352 se::DeviceMemoryAllocator* memory_allocator = run_options->allocator(); in ExecuteAsyncOnStream() local
355 CreateBufferTable(memory_allocator, stream->parent()->device_ordinal(), in ExecuteAsyncOnStream()
Dcpu_executable.h105 se::DeviceMemoryAllocator* memory_allocator, int device_ordinal,
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc96 shape, on_device_shape, backend->memory_allocator(), device_ordinal); in AllocateScopedShapedBuffer()
188 device_ordinal, backend->memory_allocator(), in CreateAndTransfer()
191 ->InitializeFromShapedBuffer(shaped_buffer, backend->memory_allocator(), in CreateAndTransfer()
211 device_ordinal, backend->memory_allocator(), in CreateUninitialized()
214 ->InitializeFromShapedBuffer(shaped_buffer, backend->memory_allocator(), in CreateUninitialized()
224 auto allocator = backend->memory_allocator(); in CreateFromBuffer()
318 InitializeFromShapedBuffer(shaped_buffer, backend->memory_allocator(), in SwapIn()
459 auto allocator = backend->memory_allocator(); in MakeTuple()
/external/tensorflow/tensorflow/lite/micro/
Dmicro_allocator.h114 static MicroAllocator* Create(SimpleMemoryAllocator* memory_allocator,
187 MicroAllocator(SimpleMemoryAllocator* memory_allocator,
Dmicro_allocator.cc67 explicit MicroBuiltinDataAllocator(SimpleMemoryAllocator* memory_allocator) in MicroBuiltinDataAllocator() argument
68 : memory_allocator_(memory_allocator) {} in MicroBuiltinDataAllocator()
598 MicroAllocator::MicroAllocator(SimpleMemoryAllocator* memory_allocator, in MicroAllocator() argument
600 : memory_allocator_(memory_allocator), in MicroAllocator()
615 MicroAllocator* MicroAllocator::Create(SimpleMemoryAllocator* memory_allocator, in Create() argument
617 TFLITE_DCHECK(memory_allocator != nullptr); in Create()
620 uint8_t* allocator_buffer = memory_allocator->AllocateFromTail( in Create()
623 new (allocator_buffer) MicroAllocator(memory_allocator, error_reporter); in Create()
Drecording_micro_allocator.h99 RecordingMicroAllocator(RecordingSimpleMemoryAllocator* memory_allocator,
/external/tensorflow/tensorflow/compiler/jit/
Dxla_tensor.cc51 client->backend().memory_allocator(), in AllocateShapedBuffer()
59 client->backend().memory_allocator()->Allocate( in AllocateShapedBuffer()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbuffer_donation_test.cc82 se::StreamExecutorMemoryAllocator memory_allocator( in RunAndCheck() local
86 run_options.set_allocator(&memory_allocator); in RunAndCheck()
102 argument_literal.shape(), &memory_allocator, in RunAndCheck()
116 &memory_allocator); in RunAndCheck()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer_test.cc38 client->backend().memory_allocator()->Allocate( in MakeArray()
46 client->backend().memory_allocator(), /*device_ordinal=*/0, in MakeArray()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc278 se::DeviceMemoryAllocator* const allocator = backend->memory_allocator(); in BuildComputationInputs()
365 memory_allocator(allocator) {} in OutputBuffers()
372 memory_allocator->Deallocate(buffers.device_ordinal(), buffer); in ~OutputBuffers()
385 se::DeviceMemoryAllocator* const memory_allocator; member
437 node_context->backend()->memory_allocator(); in AllocateOutputTensors()
689 se::DeviceMemoryAllocator* const allocator = backend->memory_allocator(); in DoWork()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner.cc52 literal.shape(), backend().memory_allocator(), in TransferLiteralToDevice()
241 argument->shape(), backend().memory_allocator(), device)); in ExecuteReplicatedImpl()
434 backend().memory_allocator())); in CreateExecutable()
439 backend().memory_allocator()); in CreateExecutable()
448 run_options.set_allocator(backend().memory_allocator()); in GetServiceRunOptionsForDevice()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc88 executable_run_options.set_allocator(backend_->memory_allocator()); in RunMlirModule()
112 auto* allocator = backend_->memory_allocator(); in RunMlirModuleWithHostBuffers()
Dgpu_convolution_regression_test.cc31 backend().default_stream_executor(), backend().memory_allocator()); in CheckForHloText()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dredzone_allocator.h46 RedzoneAllocator(Stream* stream, DeviceMemoryAllocator* memory_allocator,
Dredzone_allocator.cc49 DeviceMemoryAllocator* memory_allocator, in RedzoneAllocator() argument
60 memory_allocator_(memory_allocator), in RedzoneAllocator()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc159 run_options.set_allocator(backend_->memory_allocator()); in RunHelper()
403 allocator = backend().memory_allocator(); in LiteralToShapedBuffer()
454 shape, backend().memory_allocator(), device_ordinal)); in TransferToLocalServer()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles.cc235 TRTBaseAllocator* memory_allocator) { in CreateExecutionContexts() argument
243 ExecutionContext::Create(engine, memory_allocator); in CreateExecutionContexts()

12