Home
last modified time | relevance | path

Searched refs:record_persistent_memory_allocation (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dvariable_ops.cc76 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()
Dlookup_table_init_op.cc83 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
145 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
Dlookup_table_op.h67 ctx->record_persistent_memory_allocation( in Compute()
Dqueue_op.cc41 context->record_persistent_memory_allocation(resource_->MemoryUsed()); in Compute()
Dlookup_table_op.cc844 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()
Dconstant_op.cc87 ctx->record_persistent_memory_allocation(tensor_.AllocatedBytes()); in Compute()
/external/tensorflow/tensorflow/core/kernels/lookup_tables/
Dop_kernel_templates.h81 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()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc776 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
Dop_kernel.h1197 void record_persistent_memory_allocation(int64 size, int64 alloc_id = -1)