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