diff options
author | Anna R <annarev@google.com> | 2017-12-19 17:28:06 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-20 10:57:07 -0800 |
commit | d064a47543f51ff5a62927a76bb0fb0862d05558 (patch) | |
tree | de5429f36f01e084217c10370a0cff9c446b5e7a /tensorflow/c/c_api_internal.h | |
parent | 5d2e8e05c2ddca08e4fc7b17c88ac36a6036dd4b (diff) |
Read ApiDef from TensorFlow Go API.
PiperOrigin-RevId: 179625412
Diffstat (limited to 'tensorflow/c/c_api_internal.h')
-rw-r--r-- | tensorflow/c/c_api_internal.h | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tensorflow/c/c_api_internal.h b/tensorflow/c/c_api_internal.h index 6df77a7f9b..f8edc90a9f 100644 --- a/tensorflow/c/c_api_internal.h +++ b/tensorflow/c/c_api_internal.h @@ -24,6 +24,9 @@ limitations under the License. #include <unordered_map> #include <vector> +#ifndef __ANDROID__ +#include "tensorflow/core/framework/op_gen_lib.h" +#endif #include "tensorflow/core/common_runtime/shape_refiner.h" #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/tensor_shape.h" @@ -158,6 +161,22 @@ struct TF_Function { tensorflow::FunctionDef fdef; }; +struct TF_ApiDefMap { + explicit TF_ApiDefMap(const tensorflow::OpList& op_list) + : +#ifndef __ANDROID__ + api_def_map(op_list), +#endif + update_docs_called(false) { + } + +#ifndef __ANDROID__ + tensorflow::ApiDefMap api_def_map GUARDED_BY(lock); +#endif + bool update_docs_called GUARDED_BY(lock); + tensorflow::mutex lock; +}; + namespace tensorflow { class TensorCApi { |