HomeSort by relevance Sort by last modified time
    Searched defs:BatchNormalization (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/python/keras/_impl/keras/layers/
normalization.py 31 @tf_export('keras.layers.BatchNormalization')
32 class BatchNormalization(tf_normalization_layers.BatchNormalization, Layer):
44 set `axis=1` in `BatchNormalization`.
92 super(BatchNormalization, self).__init__(
113 output = super(BatchNormalization, self).call(inputs, training=training)
136 base_config = super(BatchNormalization, self).get_config()
  /external/tensorflow/tensorflow/python/layers/
normalization.py 45 @tf_export('layers.BatchNormalization')
46 class BatchNormalization(base.Layer):
145 super(BatchNormalization, self).__init__(
686 `data_format="channels_first"`, set `axis=1` in `BatchNormalization`.
    [all...]

Completed in 79 milliseconds