Home
last modified time | relevance | path

Searched refs:RecordTensorAllocation (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dlog_memory.h70 static void RecordTensorAllocation(const string& kernel_name, int64 step_id,
Dlog_memory.cc48 void LogMemory::RecordTensorAllocation(const string& kernel_name, in RecordTensorAllocation() function in tensorflow::LogMemory
Dtensor.cc741 LogMemory::RecordTensorAllocation("Unknown", LogMemory::UNKNOWN_STEP_ID, in Tensor()
756 LogMemory::RecordTensorAllocation("Unknown (with attributes)", in Tensor()
884 LogMemory::RecordTensorAllocation("Unknown (from Proto)", in FromProto()
Dop_kernel.cc247 LogMemory::RecordTensorAllocation( in allocate_temp()
717 LogMemory::RecordTensorAllocation(params_->op_kernel->name(), in allocate_tensor()