Searched refs:AllocationId (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | log_memory.cc | 84 allocation.set_allocation_id(allocator->AllocationId(ptr)); in RecordRawAllocation()
95 deallocation.set_allocation_id(allocator->AllocationId(ptr)); in RecordRawDeallocation()
|
| D | allocator.h | 267 virtual int64 AllocationId(const void* ptr) const { return 0; } in AllocationId() function
333 int64 AllocationId(const void* ptr) const override { in AllocationId() function
334 return wrapped_->AllocationId(ptr); in AllocationId()
|
| D | tracking_allocator.cc | 142 int64 TrackingAllocator::AllocationId(const void* ptr) const { 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 | 67 int64 AllocationId(const void* ptr) const override;
|
| D | tensor.cc | 107 int64 id = alloc_->AllocationId(data_ptr); in FillAllocationDescription()
119 LogMemory::RecordTensorDeallocation(alloc_->AllocationId(data()), in RecordDeallocation()
|
| D | op_kernel.cc | 848 int64 alloc_id = a->AllocationId(t->tensor_data().data()); in allocate_persistent()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator.cc | 131 int64 GPUDebugAllocator::AllocationId(const void* ptr) const { in AllocationId() function in tensorflow::GPUDebugAllocator
132 return base_allocator_->AllocationId(static_cast<const char*>(ptr) - in AllocationId()
|
| D | gpu_debug_allocator.h | 45 int64 AllocationId(const void* ptr) const override;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator.h | 112 int64 AllocationId(const void* ptr) const override { return 0; } in AllocationId() function
|
| D | direct_session_with_tracking_alloc_test.cc | 121 EXPECT_EQ(3, cm->AllocationId(node, 0)); in TEST()
123 EXPECT_EQ(4, cm->AllocationId(node, 0)); in TEST()
|
| D | bfc_allocator.h | 74 int64 AllocationId(const void* ptr) const override;
|
| D | bfc_allocator.cc | 906 int64 BFCAllocator::AllocationId(const void* ptr) const { in AllocationId() function in tensorflow::BFCAllocator
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.cc | 359 int64 CostModel::AllocationId(const Node* node, int slot) const { in AllocationId() function in tensorflow::CostModel
514 int64 alloc_id = AllocationId(n, i); in AddToCostGraphDef()
517 int64 input_alloc_id = AllocationId(e->src(), e->src_output()); in AddToCostGraphDef()
|
| D | costmodel.h | 162 int64 AllocationId(const Node* node, int output_slot) const;
|