MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / temp_memory_allocated

Method temp_memory_allocated

tensorflow/core/framework/op_kernel.cc:1013–1016  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1011}
1012
1013int64 OpKernelContext::temp_memory_allocated() const {
1014 mutex_lock l(stats_mu_);
1015 return temp_memory_allocated_;
1016}
1017
1018void OpKernelContext::record_persistent_memory_allocation(int64 size,
1019 int64 alloc_id) {

Callers 3

SetMemoryMethod · 0.80
UpdateStatsFunction · 0.80
gpu_device.ccFile · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected