| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_allocator.cc | 68 : memory_allocator_(memory_allocator) {} in MicroBuiltinDataAllocator()
71 return memory_allocator_->AllocateFromTail(size, alignment_hint); in Allocate()
79 SimpleMemoryAllocator* memory_allocator_; member in tflite::__anon679442d00111::MicroBuiltinDataAllocator
600 : memory_allocator_(memory_allocator), in MicroAllocator()
676 return memory_allocator_->AllocateFromTail(bytes, kBufferAlignment); in AllocatePersistentBuffer()
741 TF_LITE_ENSURE_STATUS(memory_allocator_->SetHeadBufferSize( in FinishPrepareNodeAllocations()
750 return memory_allocator_->GetUsedBytes(); in used_bytes()
761 memory_allocator_->AllocateFromTail( in AllocateNodeAndRegistrations()
785 MicroBuiltinDataAllocator builtin_data_allocator(memory_allocator_); in PrepareNodeAndRegistrationDataFromFlatbuffer()
848 memory_allocator_, error_reporter_, op->inputs(), &inputs_array)); in PrepareNodeAndRegistrationDataFromFlatbuffer()
[all …]
|
| D | micro_allocator.h | 262 SimpleMemoryAllocator* memory_allocator_; variable
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.h | 73 : memory_allocator_(allocator), in ExecutionContext()
84 : memory_allocator_(other.memory_allocator_), in ExecutionContext()
87 other.memory_allocator_ = nullptr; in ExecutionContext()
105 TRTBaseAllocator* memory_allocator_;
|
| D | trt_engine_utils.cc | 40 DCHECK(memory_allocator_) << "Internal error: Device memory with address " in ~ExecutionContext()
42 memory_allocator_->free(device_memory_); in ~ExecutionContext()
|
| /external/libchrome/base/metrics/ |
| D | persistent_histogram_allocator.h | 218 return memory_allocator_.get(); in memory_allocator()
223 uint64_t Id() const { return memory_allocator_->Id(); } in Id()
224 const char* Name() const { return memory_allocator_->Name(); } in Name()
225 const void* data() const { return memory_allocator_->data(); } in data()
226 size_t length() const { return memory_allocator_->length(); } in length()
227 size_t size() const { return memory_allocator_->size(); } in size()
228 size_t used() const { return memory_allocator_->used(); } in used()
318 std::unique_ptr<PersistentMemoryAllocator> memory_allocator_; variable
|
| D | persistent_histogram_allocator.cc | 274 : memory_allocator_(std::move(memory)), in PersistentHistogramAllocator()
275 sparse_histogram_data_manager_(memory_allocator_.get()) {} in PersistentHistogramAllocator()
287 memory_allocator_->GetAsObject<PersistentHistogramData>(ref); in GetHistogram()
288 const size_t length = memory_allocator_->GetAllocSize(ref); in GetHistogram()
321 if (memory_allocator_->IsCorrupt()) in AllocateHistogram()
331 memory_allocator_->New<PersistentHistogramData>( in AllocateHistogram()
364 memory_allocator_->Allocate(ranges_bytes, kTypeIdRangesArray); in AllocateHistogram()
367 memory_allocator_->GetAsArray<HistogramBase::Sample>( in AllocateHistogram()
379 DCHECK_EQ(kTypeIdRangesArray, memory_allocator_->GetType(ranges_ref)); in AllocateHistogram()
413 memory_allocator_->GetAsReference(histogram_data); in AllocateHistogram()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_allocations.h | 43 memory_allocator_(memory_allocator) {} in BufferAllocations()
51 return memory_allocator_; in memory_allocator()
91 se::DeviceMemoryAllocator* memory_allocator_; variable
|
| D | buffer_allocations.cc | 51 memory_allocator_->Deallocate(device_ordinal_, buffer_address); in TearDown()
|
| D | fft_thunk.h | 53 se::DeviceMemoryAllocator* memory_allocator_; variable
|
| D | fft_thunk.cc | 33 : device_ordinal_(device_ordinal), memory_allocator_(memory_allocator) {} in FftScratchAllocator()
52 memory_allocator_->Allocate(device_ordinal_, byte_size, in AllocateBytes()
|
| D | gpu_conv_algorithm_picker.cc | 58 : device_ordinal_(device_ordinal), memory_allocator_(memory_allocator) {} in ScratchAllocator()
76 se::DeviceMemoryAllocator* memory_allocator_; member in xla::gpu::__anon6437d7ed0111::ScratchAllocator
93 memory_allocator_->Allocate(device_ordinal_, byte_size, in AllocateBytes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.h | 92 return memory_allocator_.get(); in memory_allocator()
182 std::unique_ptr<se::StreamExecutorMemoryAllocator> memory_allocator_; variable
|
| D | backend.cc | 132 memory_allocator_ = absl::make_unique<se::StreamExecutorMemoryAllocator>( in Backend()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.h | 114 DeviceMemoryAllocator* memory_allocator_; variable
|
| D | redzone_allocator.cc | 60 memory_allocator_(memory_allocator), in RedzoneAllocator()
77 memory_allocator_->Allocate(device_ordinal_, in AllocateBytes()
|