Searched refs:record_persistent_memory_allocation (Results 1 – 8 of 8) sorted by relevance
90 ctx->record_persistent_memory_allocation(var->tensor()->AllocatedBytes()); in Compute()122 context->record_persistent_memory_allocation( in Compute()168 context->record_persistent_memory_allocation( in Compute()
84 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()147 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
76 ctx->record_persistent_memory_allocation( in Compute()
41 context->record_persistent_memory_allocation(resource_->MemoryUsed()); in Compute()
882 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()915 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()988 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
90 ctx->record_persistent_memory_allocation(tensor_.AllocatedBytes()); in Compute()
849 record_persistent_memory_allocation(alloc_size, alloc_id); in allocate_persistent()853 record_persistent_memory_allocation(t->TotalBytes()); in allocate_persistent()1061 void OpKernelContext::record_persistent_memory_allocation(int64 size, in record_persistent_memory_allocation() function in tensorflow::OpKernelContext
1229 void record_persistent_memory_allocation(int64 size, int64 alloc_id = -1)