(self, x, y)
| 62 | input_signature=(tensor_spec.TensorSpec([None, 2], dtypes.float32), |
| 63 | tensor_spec.TensorSpec([None], dtypes.float32))) |
| 64 | def call(self, x, y): |
| 65 | with backprop.GradientTape() as tape: |
| 66 | loss = math_ops.reduce_mean((self.dense(x) - y) ** 2.) |
| 67 | trainable_variables = self.dense.trainable_variables |
| 68 | gradients = tape.gradient(loss, trainable_variables) |
| 69 | self.optimizer.apply_gradients(zip(gradients, trainable_variables)) |
| 70 | return {"loss": loss} |
| 71 | |
| 72 | |
| 73 | def _import_and_infer( |
no test coverage detected