diff options
Diffstat (limited to 'tensorflow/c/c_test_util.h')
| -rw-r--r-- | tensorflow/c/c_test_util.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/c/c_test_util.h b/tensorflow/c/c_test_util.h index a927739d46..91f96b0e5d 100644 --- a/tensorflow/c/c_test_util.h +++ b/tensorflow/c/c_test_util.h @@ -42,6 +42,8 @@ TF_Tensor* Int32Tensor(const std::vector<int32_t>& values); TF_Tensor* Int32Tensor(int32_t v); +TF_Tensor* DoubleTensor(double v); + TF_Operation* Placeholder(TF_Graph* graph, TF_Status* s, const char* name = "feed"); @@ -51,6 +53,9 @@ TF_Operation* Const(TF_Tensor* t, TF_Graph* graph, TF_Status* s, TF_Operation* ScalarConst(int32_t v, TF_Graph* graph, TF_Status* s, const char* name = "scalar"); +TF_Operation* ScalarConst(double v, TF_Graph* graph, TF_Status* s, + const char* name = "scalar"); + TF_Operation* Add(TF_Operation* l, TF_Operation* r, TF_Graph* graph, TF_Status* s, const char* name = "add"); |