| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator.cc | 58 void* ScopedAllocator::AllocateRaw(int32_t field_index, size_t num_bytes) { in AllocateRaw() function in tensorflow::ScopedAllocator
191 void* ScopedAllocatorInstance::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::ScopedAllocatorInstance
|
| D | mkl_cpu_allocator.h | 64 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
220 inline void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
|
| D | allocator_retry.cc | 51 void* AllocatorRetry::AllocateRaw( in AllocateRaw() function in tensorflow::AllocatorRetry
|
| D | scoped_allocator.h | 105 void* AllocateRaw(size_t alignment, size_t num_bytes, in AllocateRaw() function
|
| D | process_state.h | 127 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
|
| D | pool_allocator.cc | 95 void* PoolAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::PoolAllocator
|
| D | bfc_allocator.h | 77 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tracking_allocator_test.cc | 30 void* AllocateRaw(size_t /*alignment*/, size_t num_bytes) override { in AllocateRaw() function in tensorflow::TestableSizeTrackingAllocator
56 void* AllocateRaw(size_t /*alignment*/, size_t num_bytes) override { in AllocateRaw() function in tensorflow::NoMemoryAllocator
|
| D | allocator.h | 131 virtual void* AllocateRaw(size_t alignment, size_t num_bytes, in AllocateRaw() function
250 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
254 void* AllocateRaw(size_t alignment, size_t num_bytes, in AllocateRaw() function
|
| D | tracking_allocator.h | 58 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
|
| D | tracking_allocator.cc | 32 void* TrackingAllocator::AllocateRaw( in AllocateRaw() function in tensorflow::TrackingAllocator
|
| D | cpu_allocator_impl.cc | 78 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function in tensorflow::__anon99240fbc0111::CPUAllocator
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_simple_allocator.cc | 25 void* PluggableDeviceSimpleAllocator::AllocateRaw(size_t alignment, in AllocateRaw() function in tensorflow::PluggableDeviceSimpleAllocator
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_managed_allocator.cc | 30 void* GpuManagedAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::GpuManagedAllocator
|
| D | gpu_cudamalloc_allocator.cc | 36 void* GPUcudaMallocAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::GPUcudaMallocAllocator
|
| D | gpu_debug_allocator.cc | 88 void* GPUDebugAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::GPUDebugAllocator
167 void* GPUNanResetAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::GPUNanResetAllocator
|
| D | gpu_allocator_retry_test.cc | 37 void* AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::__anon79a5365c0111::FakeAllocator
|
| D | gpu_cudamallocasync_allocator.cc | 271 void* GpuCudaMallocAsyncAllocator::AllocateRaw(size_t alignment, in AllocateRaw() function in tensorflow::GpuCudaMallocAsyncAllocator
|
| /external/tensorflow/tensorflow/core/tfrt/common/ |
| D | async_value_tensor.cc | 58 void* AsyncValueAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::AsyncValueAllocator
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | immutable_constant_op.cc | 39 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function in tensorflow::__anone9758bd00111::MemmappedTensorAllocator
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 44 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function in tensorflow::XlaCompilationAllocator
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 46 void* XlaDeviceAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::XlaDeviceAllocator
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_execute_compat.cc | 948 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function in tensorflow::tfd::__anone1369dfd0711::TestAllocator
954 void* AllocateRaw(size_t alignment, size_t num_bytes, in AllocateRaw() function in tensorflow::tfd::__anone1369dfd0711::TestAllocator
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | light_outside_compilation.cc | 248 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function in tensorflow::__anonfdc132ad0311::WriteIntoXlaBufferAllocator
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 51 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function in tensorflow::__anonba1791850111::FakeAllocator
|