Searched refs:allocation_description (Results 1 – 15 of 15) sorted by relevance
5 import "tensorflow/core/framework/allocation_description.proto";23 AllocationDescription allocation_description = 4; field
5 import "tensorflow/core/framework/allocation_description.proto";
69 AllocationDescription allocation_description; in GetAllocatedBytes() local70 FillAllocationDescription(&allocation_description); in GetAllocatedBytes()71 if (allocation_description.allocated_bytes() > 0) { in GetAllocatedBytes()72 *out_bytes = allocation_description.allocated_bytes(); in GetAllocatedBytes()
118 "allocation_description.proto",1438 srcs = ["allocation_description.proto"],
118 .allocation_description() in AddMemoryStats()121 .allocation_description() in AddMemoryStats()124 output.tensor_description().allocation_description().ptr(); in AddMemoryStats()
118 const char* allocation_description) const { in PrintRecordedAllocation()127 allocation.count, allocation_description); in PrintRecordedAllocation()
104 const char* allocation_description) const;
357 .allocation_description() in BuildCostModel()363 .allocation_description() in BuildCostModel()
185 .allocation_description() in InferFromTrace()
452 .allocation_description() in GetStatsStringFromRunMetadata()
511 allocation = output.tensor_description.allocation_description
121 no.tensor_description().allocation_description().requested_bytes(); in MergeFromStats()
293 tot += no.tensor_description().allocation_description().requested_bytes(); in DetailText()
193 "//tensorflow/core/framework:allocation_description.proto",
1962 output.tensor_description.allocation_description.ptr)