| 47 | |
| 48 | |
| 49 | class LayerWithMetrics(keras.layers.Layer): |
| 50 | |
| 51 | def build(self, input_shape): |
| 52 | self.mean = keras.metrics.Mean(name='mean_object') |
| 53 | |
| 54 | def call(self, inputs): |
| 55 | self.add_metric( |
| 56 | math_ops.reduce_mean(inputs), name='mean_tensor', aggregation='mean') |
| 57 | self.add_metric(self.mean(inputs)) |
| 58 | return inputs |
| 59 | |
| 60 | |
| 61 | class LayerWithTrainingArg(keras.layers.Layer): |
no outgoing calls