STT-tensorflow/tensorflow/python/keras/layers/ops
Thomas O'Malley 70387ab55b Reduce Dense.__call__ overhead by ~15%
- Creates a layers/ops directory to contain functional implementations of Keras
  layer ops (this should make it easier to XLA compile these ops, as well as potentially being useful to expose stateless versions of the layer ops)
- Moves Dense op to this directory and implements a functional version of
  Dense.call with reduced Python overhead
- Uses this op in Dense layer
- Cleans up input_dim and activity_regularizer handling in Dense and Layer
- Adds a microbenchmark for Dense

Also fixes two small issues:
- compute_dtype_object was not always correctly set in Layer
- activity_regularizer can now be passed as a str to any Layer class

PiperOrigin-RevId: 314256375
Change-Id: I769cef6d67aa117f6cb75dc34c0594e748284af2
2020-06-01 20:22:44 -07:00
..
__init__.py
BUILD Reduce Dense.__call__ overhead by ~15% 2020-06-01 20:22:44 -07:00
core.py