Searched refs:record_persistent_memory_allocation (Results 1 – 8 of 8) sorted by relevance
76 ctx->record_persistent_memory_allocation(var->tensor()->AllocatedBytes()); in Compute()107 context->record_persistent_memory_allocation( in Compute()152 context->record_persistent_memory_allocation( in Compute()
83 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()145 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()
860 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()893 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()966 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
91 ctx->record_persistent_memory_allocation(tensor_.AllocatedBytes()); in Compute()
886 record_persistent_memory_allocation(alloc_size, alloc_id); in allocate_persistent()890 record_persistent_memory_allocation(t->TotalBytes()); in allocate_persistent()1057 void OpKernelContext::record_persistent_memory_allocation(int64 size, in record_persistent_memory_allocation() function in tensorflow::OpKernelContext
1249 void record_persistent_memory_allocation(int64 size, int64 alloc_id = -1)