tree 5ecca16171e062f34b1685e3035ddc657ecb1d3d
parent 9202a44c3085bfd3316b8e14fa8684dc18e5b8e2
author Kibeom Kim <kkb@google.com> 1605725046 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1605744066 -0800

Add a tag to `tensorflow::Graph` that indicates where it's originated from.

- `ConstructionContext::kDirectSession` From `tensorflow::DirectSession`, tf1 session API.
- `ConstructionContext::kFunctionDef`: From `FunctionDef`, @tf.function.
- `ConstructionContext::kUnknown`: Not tracked.

It can be accessed via `Graph::GetConstructionContext()`

PiperOrigin-RevId: 343109880
Change-Id: I7b3488648855c9d86d4fb4a202bd66cf7182191a