| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator.cc | 58 void* ScopedAllocator::AllocateRaw(int32 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 | 66 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
225 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 | 104 void* AllocateRaw(size_t alignment, size_t num_bytes, in AllocateRaw() function
|
| D | process_state.h | 126 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 | 59 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
|
| D | bfc_allocator.cc | 253 void* BFCAllocator::AllocateRaw(size_t unused_alignment, size_t num_bytes, in AllocateRaw() function in tensorflow::BFCAllocator
|
| /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 | 39 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::__anon55289c970111::FakeAllocator
|
| /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 | 201 virtual void* AllocateRaw(size_t alignment, size_t num_bytes, in AllocateRaw() function
306 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
310 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 | 76 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function in tensorflow::__anon435a67570111::CPUAllocator
|
| D | tensor_test.cc | 1155 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function in tensorflow::__anona5e40b380111::DummyCPUAllocator
|
| /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::__anonda76d6cb0111::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 | 42 void* XlaDeviceAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::XlaDeviceAllocator
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 182 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function in xla::CpuAllocator
|