from-300-fix-layer-build-issue-and-computational-optimizations

[david-thrower]

From #300

TLDR

Optimizations and addressing warnings...

Issue 1: Layer build warnings / layers missing a proper build method.
Issue 2: Make sure any functions where it would be beneficial are decorated with @tf.function
Issue 3: TensorFlow tf.function Retracing Warning

# Instead of this:
@tf.function
def my_func(x, my_scalar):
    return x * my_scalar
for i in range(10):
    my_func(data, i) # Retraces 10 times

# Do this:
@tf.function
def my_func(x, my_scalar_tensor):
    return x * my_scalar_tensor
for i in range(10):
    my_func(data, tf.constant(i)) # Traces once