tensorflow optimization example