Searched refs:record_persistent_memory_allocation (Results 1 – 9 of 9) 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()
87 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()154 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
78 ctx->record_persistent_memory_allocation( in Compute()157 ctx->record_persistent_memory_allocation(table->MemoryUsed() + in Compute()
962 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()995 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()1068 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
41 context->record_persistent_memory_allocation(resource_->MemoryUsed()); in Compute()
92 ctx->record_persistent_memory_allocation(tensor_.AllocatedBytes()); in Compute()
271 ctx->record_persistent_memory_allocation(resource->MemoryUsed() - in Compute()
1190 void record_persistent_memory_allocation(int64_t size, int64_t alloc_id = -1)
1072 void OpKernelContext::record_persistent_memory_allocation(int64_t size, in record_persistent_memory_allocation() function in tensorflow::OpKernelContext