Searched refs:record_persistent_memory_allocation (Results 1 – 9 of 9) sorted by relevance
76 ctx->record_persistent_memory_allocation(var->tensor()->AllocatedBytes()); in Compute()107 context->record_persistent_memory_allocation( in Compute()151 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()
67 ctx->record_persistent_memory_allocation( in Compute()
41 context->record_persistent_memory_allocation(resource_->MemoryUsed()); in Compute()
844 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()879 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()954 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
87 ctx->record_persistent_memory_allocation(tensor_.AllocatedBytes()); in Compute()
81 ctx->record_persistent_memory_allocation(resource->MemoryUsed()); in Compute()216 ctx->record_persistent_memory_allocation(resource->MemoryUsed()); in Compute()330 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
776 record_persistent_memory_allocation(alloc_size, alloc_id); in allocate_persistent()900 void OpKernelContext::record_persistent_memory_allocation(int64 size, in record_persistent_memory_allocation() function in tensorflow::OpKernelContext
1197 void record_persistent_memory_allocation(int64 size, int64 alloc_id = -1)