tree b3ddb8959c5b48df6178fb39f9e26c78c0015a09
parent 1ffc9f5263ef1d858e4e79c8abec88f5cd15d52f
author Jacques Pienaar <jpienaar@google.com> 1588645422 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1588645719 -0700

Compute values as shapes where requested by shape function

Detect cases where shape functions require input operand as shape. If the operand is not known, then attempt to compute all the dimensions of the required shape (e.g., this allows computing [?,1] as output rather than failing to compute the entire output). This adds a simple partial computation evaluator of inputs. These are currently local functions but the intention is to extract them into general traits rather than code the op specific behavior here.

PiperOrigin-RevId: 309865790
Change-Id: I4458f4694e775132ba218cfc5f292edcd0001c64