Python tensorflow CNN framework