Home
last modified time | relevance | path

Searched refs:AllocatedBytes (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtyped_queue.h87 return sq.size() * sq.front().AllocatedBytes(); in SizeOf()
95 return sq.size() * sq.front().AllocatedBytes(); in SizeOf()
105 return sq.size() * (sizeof(TensorPair) + sq.top().second.AllocatedBytes()); in SizeOf()
Dvariable_ops.cc76 ctx->record_persistent_memory_allocation(var->tensor()->AllocatedBytes()); in Compute()
108 tmp_var->val.AllocatedBytes()); in Compute()
152 -static_cast<int64>(tmpvar.AllocatedBytes())); in Compute()
Dlookup_table_op.h68 container->MemoryUsed() + table_handle_.AllocatedBytes()); in Compute()
Dlookup_table_op.cc562 return sizeof(MutableDenseHashTable) + key_buckets_.AllocatedBytes() + in MemoryUsed()
563 value_buckets_.AllocatedBytes() + empty_key_.AllocatedBytes(); in MemoryUsed()
Dconstant_op.cc87 ctx->record_persistent_memory_allocation(tensor_.AllocatedBytes()); in Compute()
/external/tensorflow/tensorflow/core/kernels/data/
Dwindow_dataset.cc44 int64 AllocatedBytes() const override { in AllocatedBytes() function in tensorflow::data::__anon8ba9a6530111::WindowDataset
/external/tensorflow/tensorflow/core/framework/
Ddataset.cc325 allocated_bytes += dataset->AllocatedBytes(); in GetAllocatedBytes()
327 allocated_bytes += tensor.AllocatedBytes(); in GetAllocatedBytes()
Dop_kernel.h258 int64 AllocatedBytes() const { return tensor_.AllocatedBytes(); } in AllocatedBytes() function
Ddataset.h646 virtual int64 AllocatedBytes() const { return 0; } in AllocatedBytes() function
Dtensor.h232 size_t AllocatedBytes() const;
Dtensor.cc916 size_t Tensor::AllocatedBytes() const { in AllocatedBytes() function in tensorflow::Tensor