Searched defs:memory_allocator (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_allocations.h | 40 se::DeviceMemoryAllocator* 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()
377 se::DeviceMemoryAllocator* const memory_allocator, in GenerateBufferAllocations()
421 se::DeviceMemoryAllocator* const memory_allocator = run_options->allocator(); in ExecuteAsyncOnStreamImpl() local
|
| D | fft_thunk.cc | 32 int device_ordinal, se::DeviceMemoryAllocator* memory_allocator) in FftScratchAllocator()
|
| D | gpu_conv_algorithm_picker.cc | 57 se::DeviceMemoryAllocator* memory_allocator) in ScratchAllocator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 94 se::DeviceMemoryAllocator* memory_allocator, int device_ordinal) { in MemoryForAllocation()
128 se::DeviceMemoryAllocator* memory_allocator, int device_ordinal, in CreateBufferTable()
352 se::DeviceMemoryAllocator* memory_allocator = run_options->allocator(); in ExecuteAsyncOnStream() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.h | 91 se::DeviceMemoryAllocator* memory_allocator() const { in memory_allocator() function
|
| D | shaped_buffer.h | 176 se::DeviceMemoryAllocator* memory_allocator() const { return allocator_; } in memory_allocator() function
|
| /external/libchrome/base/metrics/ |
| D | persistent_histogram_allocator.h | 217 PersistentMemoryAllocator* memory_allocator() { in memory_allocator() function
|
| D | persistent_histogram_allocator.cc | 914 PersistentMemoryAllocator* memory_allocator = in ReleaseForTesting() local
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_allocator.cc | 67 explicit MicroBuiltinDataAllocator(SimpleMemoryAllocator* memory_allocator) in MicroBuiltinDataAllocator()
598 MicroAllocator::MicroAllocator(SimpleMemoryAllocator* memory_allocator, in MicroAllocator()
615 MicroAllocator* MicroAllocator::Create(SimpleMemoryAllocator* memory_allocator, in Create()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.cc | 49 DeviceMemoryAllocator* memory_allocator, in RedzoneAllocator()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles.cc | 235 TRTBaseAllocator* memory_allocator) { in CreateExecutionContexts()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 82 se::StreamExecutorMemoryAllocator memory_allocator( in RunAndCheck() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 385 se::DeviceMemoryAllocator* const memory_allocator; member
|