(inputShape: Shape|Shape[])
| 495 | } |
| 496 | |
| 497 | override build(inputShape: Shape|Shape[]): void { |
| 498 | nameScope(this.forwardLayer.name, () => { |
| 499 | this.forwardLayer.build(inputShape); |
| 500 | }); |
| 501 | nameScope(this.backwardLayer.name, () => { |
| 502 | this.backwardLayer.build(inputShape); |
| 503 | }); |
| 504 | this.built = true; |
| 505 | } |
| 506 | |
| 507 | override computeMask(inputs: Tensor|Tensor[], mask?: Tensor|Tensor[]): Tensor |
| 508 | |Tensor[] { |