Commit 1adbc5aa authored by Allen Lavoie's avatar Allen Lavoie Committed by Alex Passos
Browse files

Add a single positional argument mode for shape inference in subclassed Models.

Allows fit() when call's signature looks something like call(x, training=True).

Calling conventions are "inputs", single positional, and multiple positional. Right now the distinction between "inputs" and single positional calling conventions is the text of one error message. Both support shape inference (which just hasn't been implemented for multiple positional input arguments yet).

PiperOrigin-RevId: 198815483
parent 4b6cde9b
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment