diff options
| author |  Mingsheng Hong <hongm@google.com> | 2018-03-09 19:07:44 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-09 19:11:36 -0800 |
| commit | a9bb191793e8e8c924b6a19f645610809b1dae62 (patch) | |
| tree | d39702d206596b1518465ba49e0e484ab8979b22 /tensorflow/c/c_test_util.h | |
| parent | 54d785dfdcb4eb6758741e20a6d111fda577dc99 (diff) | |
Unified test util PlaceHolderFloat() into PlaceHolder(), and extended the latter
to take a TF_DataType param.
PiperOrigin-RevId: 188570493
Diffstat (limited to 'tensorflow/c/c_test_util.h')
| -rw-r--r-- | tensorflow/c/c_test_util.h | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/tensorflow/c/c_test_util.h b/tensorflow/c/c_test_util.h index 8cf060f73f..d87c57fd51 100644 --- a/tensorflow/c/c_test_util.h +++ b/tensorflow/c/c_test_util.h @@ -46,12 +46,9 @@ TF_Tensor* DoubleTensor(double v); TF_Tensor* FloatTensor(float v); -// TODO(hongm): Change Placeholder() to take in a TF_DataType parameter, and -// unify with PlaceholderFloat. TF_Operation* Placeholder(TF_Graph* graph, TF_Status* s, - const char* name = "feed"); -TF_Operation* PlaceholderFloat(TF_Graph* graph, TF_Status* s, - const char* name = "feed"); + const char* name = "feed", + TF_DataType dtype = TF_INT32); TF_Operation* Const(TF_Tensor* t, TF_Graph* graph, TF_Status* s, const char* name = "const"); |