Searched refs:RecordTensorOutput (Results 1 – 3 of 3) sorted by relevance
83 static void RecordTensorOutput(const string& kernel_name, int64 step_id,
66 void LogMemory::RecordTensorOutput(const string& kernel_name, in RecordTensorOutput() function in tensorflow::LogMemory
1886 LogMemory::RecordTensorOutput(ctx->op_kernel().name(), in ProcessOutputs()1904 LogMemory::RecordTensorOutput(ctx->op_kernel().name(), in ProcessOutputs()