| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | pool_allocator_test.cc | 39 EXPECT_EQ(nullptr, pool.AllocateRaw(4 /*alignment*/, 0 /*num_bytes*/)); in TEST()
66 void* p0 = pool.AllocateRaw(4, 0); in TEST()
67 void* p4 = pool.AllocateRaw(4, 4); in TEST()
68 void* p12 = pool.AllocateRaw(4, 12); in TEST()
95 void* p = pool.AllocateRaw(alignment, 111); in TEST()
112 void* p = pool.AllocateRaw(4, 64 << i); in TEST()
124 void* p = pool.AllocateRaw(4, 64 << i); in TEST()
162 void* p1_16 = pool.AllocateRaw(4, 16); in TEST()
173 void* p2_16 = pool.AllocateRaw(4, 16); // Get it again. in TEST()
185 void* p3_4 = pool.AllocateRaw(4, 4); in TEST()
[all …]
|
| D | gpu_bfc_allocator_test.cc | 112 void* raw = a.AllocateRaw(1, s); in TEST_P()
145 void* raw = a.AllocateRaw(1, size); in TEST_P()
162 void* out_of_memory_ptr = a.AllocateRaw(1, (1 << 30) + 1); in TEST_P()
169 void* raw = a.AllocateRaw(1, size); in TEST_P()
276 void* raw = a.AllocateRaw(1, size); in TEST_P()
296 void* raw = a.AllocateRaw(1, size); in TEST_P()
327 void* amem = a.AllocateRaw(1, 1); in TEST_P()
328 void* bmem = b.AllocateRaw(1, 1 << 30); in TEST_P()
346 void* p = a.AllocateRaw(1, bytes); in BM_Allocation()
374 void* p = a.AllocateRaw(1, bytes); in BM_AllocationThreaded()
[all …]
|
| D | gpu_debug_allocator.cc | 88 void* GPUDebugAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::GPUDebugAllocator
90 void* allocated_ptr = base_allocator_->AllocateRaw(alignment, num_bytes); in AllocateRaw()
167 void* GPUNanResetAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::GPUNanResetAllocator
168 void* allocated_ptr = base_allocator_->AllocateRaw(alignment, num_bytes); in AllocateRaw()
|
| D | gpu_allocator_retry_test.cc | 37 void* AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw() function in tensorflow::__anon8bb5cd3e0111::FakeAllocator
38 return retry_.AllocateRaw( in AllocateRaw()
133 ptr = alloc_->AllocateRaw(16, 1); in LaunchConsumerThreads()
|
| D | gpu_debug_allocator.h | 40 void* AllocateRaw(size_t alignment, size_t num_bytes) override;
70 void* AllocateRaw(size_t alignment, size_t num_bytes) override;
|
| D | gpu_managed_allocator.h | 30 void* AllocateRaw(size_t alignment, size_t num_bytes) override;
|
| D | gpu_cudamalloc_allocator.h | 38 void* AllocateRaw(size_t alignment, size_t num_bytes) override;
|
| 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
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator_mgr_test.cc | 140 sa_instances_[0]->AllocateRaw(0 /* alignment */, 512 * sizeof(float)); in TEST_F()
142 sa_instances_[1]->AllocateRaw(0 /* alignment */, 512 * sizeof(float)); in TEST_F()
157 char* ptr0 = static_cast<char*>(inst0->AllocateRaw(0, 512 * sizeof(float))); in TEST_F()
163 char* ptr1 = static_cast<char*>(inst1->AllocateRaw(0, 9 * sizeof(float))); in TEST_F()
167 char* ptr2 = static_cast<char*>(inst2->AllocateRaw(0, 512 * sizeof(float))); in TEST_F()
210 static_cast<char*>(sa_instances_[0]->AllocateRaw(0, 512 * sizeof(float))); in TEST_F()
216 EXPECT_EQ(nullptr, sa_instances_[1]->AllocateRaw(0, 256 * sizeof(float))); in TEST_F()
219 EXPECT_EQ(nullptr, sa_instances_[1]->AllocateRaw(0, 1024 * sizeof(float))); in TEST_F()
220 void* ptr1 = sa_instances_[1]->AllocateRaw(0, 512 * sizeof(float)); in TEST_F()
222 EXPECT_EQ(nullptr, sa_instances_[0]->AllocateRaw(0, 512 * sizeof(float))); in TEST_F()
|
| D | scoped_allocator.h | 69 void* AllocateRaw(int32 field_index, size_t num_bytes) TF_LOCKS_EXCLUDED(mu_);
102 void* AllocateRaw(size_t alignment, size_t num_bytes)
104 void* AllocateRaw(size_t alignment, size_t num_bytes, in AllocateRaw() function
106 return AllocateRaw(alignment, num_bytes); in AllocateRaw()
|
| 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
231 return small_size_allocator_->AllocateRaw(alignment, num_bytes); in AllocateRaw()
234 void* ptr = large_size_allocator_->AllocateRaw(alignment, num_bytes); in AllocateRaw()
281 return cpu_allocator()->AllocateRaw(kAlignment, size); in MallocHook()
|
| 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
192 void* ptr = scoped_allocator_->AllocateRaw(field_index_, num_bytes); in AllocateRaw()
|
| D | bfc_allocator_test.cc | 62 long_lived[i] = bfc_allocator.AllocateRaw(1, kAllocSize); in BM_Allocator()
75 short_lived[i] = bfc_allocator.AllocateRaw(1, kAllocSize); in BM_Allocator()
|
| D | process_state.h | 126 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
127 void* p = a_->AllocateRaw(alignment, num_bytes); in AllocateRaw()
|
| D | allocator_retry.h | 38 void* AllocateRaw(std::function<void*(size_t alignment, size_t num_bytes,
|
| D | allocator_retry.cc | 51 void* AllocatorRetry::AllocateRaw( in AllocateRaw() function in tensorflow::AllocatorRetry
|
| /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
74 void* p1 = ta->AllocateRaw(4, 4); in TEST()
76 void* p2 = ta->AllocateRaw(4, 12); in TEST()
91 p1 = ta->AllocateRaw(4, 4); in TEST()
97 p2 = ta->AllocateRaw(4, 12); in TEST()
123 void* p1 = ta->AllocateRaw(4, 12); in TEST()
125 void* p2 = ta->AllocateRaw(4, 4); in TEST()
149 void* p1 = ta->AllocateRaw(4, 12); in TEST()
|
| D | allocator.h | 195 virtual void* AllocateRaw(size_t alignment, size_t num_bytes) = 0;
201 virtual void* AllocateRaw(size_t alignment, size_t num_bytes, in AllocateRaw() function
205 return AllocateRaw(alignment, num_bytes); in AllocateRaw()
306 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
307 return wrapped_->AllocateRaw(alignment, num_bytes); in AllocateRaw()
310 void* AllocateRaw(size_t alignment, size_t num_bytes, in AllocateRaw() function
312 return wrapped_->AllocateRaw(alignment, num_bytes, allocation_attr); in AllocateRaw()
|
| D | tracking_allocator.h | 58 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function
59 return AllocateRaw(alignment, num_bytes, AllocationAttributes()); in AllocateRaw()
61 void* AllocateRaw(size_t alignment, size_t num_bytes,
|
| D | cpu_allocator_impl.cc | 76 void* AllocateRaw(size_t alignment, size_t num_bytes) override { in AllocateRaw() function in tensorflow::__anon29597ede0111::CPUAllocator
162 return cpu_allocator_->AllocateRaw(alignment, num_bytes); in Alloc()
|
| D | tracking_allocator.cc | 32 void* TrackingAllocator::AllocateRaw( in AllocateRaw() function in tensorflow::TrackingAllocator
35 void* ptr = allocator_->AllocateRaw(alignment, num_bytes, allocation_attr); in AllocateRaw()
|
| D | allocator_test.cc | 140 void* raw = a->AllocateRaw(1, s); in TEST()
235 void* p = a->AllocateRaw(1, bytes); in BM_Allocation()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | tf_allocator_adapter.cc | 46 data = wrapped_->AllocateRaw(tensorflow::Allocator::kAllocatorAlignment, in Allocate()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_framework_c_interface.cc | 68 ->AllocateRaw(Allocator::kAllocatorAlignment, in _mlir_ciface_tf_alloc()
|