Home
last modified time | relevance | path

Searched refs:MemoryUsed (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dlookup_table_init_op.cc79 memory_used_before = table->MemoryUsed(); in Compute()
83 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
139 memory_used_before = table->MemoryUsed(); in Compute()
145 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
Dtyped_queue.h41 int64 MemoryUsed() const override;
111 int64 TypedQueue<SubQueue>::MemoryUsed() const { in MemoryUsed() function
Dlookup_table_op.cc136 int64 MemoryUsed() const override { in MemoryUsed() function in tensorflow::lookup::MutableHashTableOfScalars
274 int64 MemoryUsed() const override { in MemoryUsed() function in tensorflow::lookup::MutableHashTableOfTensors
560 int64 MemoryUsed() const override { in MemoryUsed() function in tensorflow::lookup::MutableDenseHashTable
840 memory_used_before = table->MemoryUsed(); in Compute()
844 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
875 memory_used_before = table->MemoryUsed(); in Compute()
879 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
950 memory_used_before = table->MemoryUsed(); in Compute()
954 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
Dlookup_table_op.h68 container->MemoryUsed() + table_handle_.AllocatedBytes()); in Compute()
261 int64 MemoryUsed() const override { in MemoryUsed() function
Dqueue_op.cc41 context->record_persistent_memory_allocation(resource_->MemoryUsed()); 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()
320 memory_used_before = table->MemoryUsed(); in Compute()
330 ctx->record_persistent_memory_allocation(table->MemoryUsed() - in Compute()
/external/tensorflow/tensorflow/core/util/
Dpresized_cuckoo_map.h145 int64 MemoryUsed() const { in MemoryUsed() function
/external/tensorflow/tensorflow/core/framework/
Dresource_mgr.h83 virtual int64 MemoryUsed() const { return 0; } in MemoryUsed() function