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()
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()
41 context->record_persistent_memory_allocation(resource_->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()
89 ctx->record_persistent_memory_allocation(tensor_.AllocatedBytes()); in Compute()
1179 void record_persistent_memory_allocation(int64_t size, int64_t alloc_id = -1)
991 void OpKernelContext::record_persistent_memory_allocation(int64_t size, in record_persistent_memory_allocation() function in tensorflow::OpKernelContext