Searched refs:AllocationId (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | log_memory.cc | 85 allocation.set_allocation_id(allocator->AllocationId(ptr)); in RecordRawAllocation()
96 deallocation.set_allocation_id(allocator->AllocationId(ptr)); in RecordRawDeallocation()
|
| D | allocator.h | 191 virtual int64 AllocationId(const void* ptr) { return 0; } in AllocationId() function
327 int64 AllocationId(const void* ptr) override { in AllocationId() function
328 return wrapped_->AllocationId(ptr); in AllocationId()
|
| D | tracking_allocator.cc | 142 int64 TrackingAllocator::AllocationId(const void* ptr) { in AllocationId() function in tensorflow::TrackingAllocator
151 return allocator_->AllocationId(ptr); in AllocationId()
|
| D | tracking_allocator_test.cc | 94 EXPECT_EQ(1, ta->AllocationId(p1)); in TEST()
100 EXPECT_EQ(2, ta->AllocationId(p2)); in TEST()
|
| D | tracking_allocator.h | 68 int64 AllocationId(const void* ptr) override;
|
| D | tensor.cc | 85 int64 id = alloc_->AllocationId(data_ptr); in FillAllocationDescription()
97 LogMemory::RecordTensorDeallocation(alloc_->AllocationId(data()), in RecordDeallocation()
|
| D | op_kernel.cc | 775 int64 alloc_id = a->AllocationId(t->tensor_data().data()); in allocate_persistent()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator.cc | 130 int64 GPUDebugAllocator::AllocationId(const void* ptr) { in AllocationId() function in tensorflow::GPUDebugAllocator
131 return base_allocator_->AllocationId(static_cast<const char*>(ptr) - in AllocationId()
|
| D | gpu_debug_allocator.h | 45 int64 AllocationId(const void* ptr) override;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator.h | 110 int64 AllocationId(const void* ptr) override { return 0; } in AllocationId() function
|
| D | direct_session_with_tracking_alloc_test.cc | 121 EXPECT_EQ(13, cm->AllocationId(node, 0)); in TEST()
123 EXPECT_EQ(14, cm->AllocationId(node, 0)); in TEST()
|
| D | bfc_allocator.h | 70 int64 AllocationId(const void* ptr) override;
|
| D | bfc_allocator.cc | 545 int64 BFCAllocator::AllocationId(const void* ptr) { in AllocationId() function in tensorflow::BFCAllocator
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.cc | 357 int64 CostModel::AllocationId(const Node* node, int slot) const { in AllocationId() function in tensorflow::CostModel
511 int64 alloc_id = AllocationId(n, i); in AddToCostGraphDef()
514 int64 input_alloc_id = AllocationId(e->src(), e->src_output()); in AddToCostGraphDef()
|
| D | costmodel.h | 154 int64 AllocationId(const Node* node, int output_slot) const;
|