/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | buffer_allocations.h | 40 se::DeviceMemoryAllocator* memory_allocator) in BufferAllocations() argument 43 memory_allocator_(memory_allocator) {} in BufferAllocations() 50 se::DeviceMemoryAllocator* memory_allocator() const { in memory_allocator() function
|
D | gpu_executable.cc | 306 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()
|
D | gpu_executable.h | 162 se::DeviceMemoryAllocator* const memory_allocator, 169 se::DeviceMemoryAllocator* const memory_allocator, int device_ordinal,
|
D | fft_thunk.cc | 32 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()
|
D | cudnn_fused_conv_rewriter_test.cc | 41 backend().default_stream_executor(), backend().memory_allocator()) in GetOptimizedHlo() 337 backend().default_stream_executor(), backend().memory_allocator()) in TEST_F()
|
D | fft_thunk.h | 42 se::DeviceMemoryAllocator* memory_allocator);
|
/external/libchrome/base/metrics/ |
D | persistent_histogram_allocator_unittest.cc | 40 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()
|
D | persistent_histogram_allocator.cc | 259 : 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/ |
D | cpu_executable.cc | 94 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()
|
D | cpu_executable.h | 105 se::DeviceMemoryAllocator* memory_allocator, int device_ordinal,
|
/external/tensorflow/tensorflow/compiler/xrt/ |
D | xrt_state.cc | 96 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/ |
D | micro_allocator.h | 114 static MicroAllocator* Create(SimpleMemoryAllocator* memory_allocator, 187 MicroAllocator(SimpleMemoryAllocator* memory_allocator,
|
D | micro_allocator.cc | 67 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()
|
D | recording_micro_allocator.h | 99 RecordingMicroAllocator(RecordingSimpleMemoryAllocator* memory_allocator,
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_tensor.cc | 51 client->backend().memory_allocator(), in AllocateShapedBuffer() 59 client->backend().memory_allocator()->Allocate( in AllocateShapedBuffer()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | buffer_donation_test.cc | 82 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/ |
D | tracked_device_buffer_test.cc | 38 client->backend().memory_allocator()->Allocate( in MakeArray() 46 client->backend().memory_allocator(), /*device_ordinal=*/0, in MakeArray()
|
/external/tensorflow/tensorflow/core/tpu/kernels/ |
D | tpu_execute_op.cc | 278 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/ |
D | hlo_runner.cc | 52 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/ |
D | mlir_gpu_test_base.cc | 88 executable_run_options.set_allocator(backend_->memory_allocator()); in RunMlirModule() 112 auto* allocator = backend_->memory_allocator(); in RunMlirModuleWithHostBuffers()
|
D | gpu_convolution_regression_test.cc | 31 backend().default_stream_executor(), backend().memory_allocator()); in CheckForHloText()
|
/external/tensorflow/tensorflow/stream_executor/gpu/ |
D | redzone_allocator.h | 46 RedzoneAllocator(Stream* stream, DeviceMemoryAllocator* memory_allocator,
|
D | redzone_allocator.cc | 49 DeviceMemoryAllocator* memory_allocator, in RedzoneAllocator() argument 60 memory_allocator_(memory_allocator), in RedzoneAllocator()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | local_client.cc | 159 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/ |
D | trt_shape_optimization_profiles.cc | 235 TRTBaseAllocator* memory_allocator) { in CreateExecutionContexts() argument 243 ExecutionContext::Create(engine, memory_allocator); in CreateExecutionContexts()
|