diff options
| author |  Mingsheng Hong <hongm@google.com> | 2018-02-07 12:22:55 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-07 12:26:52 -0800 |
| commit | 65f676426d808cf20abd6a4a30ad48668ee5fdf2 (patch) | |
| tree | 84707b8b38781cb4703e46656f24737d86ac1d2d /tensorflow/c/c_test_util.h | |
| parent | 90ce80131a8b5213d9f3eb9649d63921db7874a4 (diff) | |
Initial XLA support for TF eager. This is prerequisite for TF compiler's XLA support.
This CL adds XLA support for the following TFE_Op's:
1. A TF op such as MatMul, with full support of constant and resource params.
2. A TF_Function as TFE_Op, where the function must have no constant and
resource params. Removing this restriction requires more discussion and will be
deferred to a later time.
PiperOrigin-RevId: 184877345
Diffstat (limited to 'tensorflow/c/c_test_util.h')
| -rw-r--r-- | tensorflow/c/c_test_util.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/c/c_test_util.h b/tensorflow/c/c_test_util.h index 6acc2fec00..77520be010 100644 --- a/tensorflow/c/c_test_util.h +++ b/tensorflow/c/c_test_util.h @@ -69,6 +69,9 @@ TF_Operation* AddWithCtrlDependency(TF_Operation* l, TF_Operation* r, TF_Operation* Add(TF_Output l, TF_Output r, TF_Graph* graph, TF_Status* s, const char* name = "add"); +TF_Operation* Min(TF_Operation* l, TF_Operation* r, TF_Graph* graph, + TF_Status* s, const char* name = "min"); + TF_Operation* Neg(TF_Operation* n, TF_Graph* graph, TF_Status* s, const char* name = "neg"); |