tf.PrePostProcessWrapper module

tf.PrePostProcessWrapper module

class tf.PrePostProcessWrapper.PrePostProcessWrapper(layer, pre_process_config=[], post_process_config=[], **kwargs)

Bases: object

Helper class that allows for a flexible specification of pre- and post-process operations wrapped around a given layer.