diff options
author | 2017-09-28 13:08:27 -0700 | |
---|---|---|
committer | 2017-09-28 13:16:12 -0700 | |
commit | 4f3956698fd8d0aeffb6c4e40fef05664e4ff3cc (patch) | |
tree | cda9080022177ffc286735f247030d576ce499be /tensorflow/c/c_api_function.cc | |
parent | b1728aa3c5d2d8545acea781f1e2d6ffeccf3f7a (diff) |
Use void* intead of TF_Buffer in TF_FunctionImportFunctionDef
void* is more common (and more convenient) for passing in serialized
protobufs in c_api.h.
PiperOrigin-RevId: 170386128
Diffstat (limited to 'tensorflow/c/c_api_function.cc')
-rw-r--r-- | tensorflow/c/c_api_function.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/c/c_api_function.cc b/tensorflow/c/c_api_function.cc index 7924c31a5f..dcb818b88b 100644 --- a/tensorflow/c/c_api_function.cc +++ b/tensorflow/c/c_api_function.cc @@ -548,10 +548,10 @@ void TF_FunctionToFunctionDef(TF_Function* func, TF_Buffer* output_func_def, status->status = MessageToBuffer(func->fdef, output_func_def); } -TF_Function* TF_FunctionImportFunctionDef(const TF_Buffer* func_def, +TF_Function* TF_FunctionImportFunctionDef(const void* proto, size_t proto_len, TF_Status* status) { TF_Function* func = new TF_Function(); - if (!func->fdef.ParseFromArray(func_def->data, func_def->length)) { + if (!func->fdef.ParseFromArray(proto, proto_len)) { status->status = InvalidArgument( "Invalid FunctionDef given to TF_FunctionImportFunctionDef"); TF_DeleteFunction(func); |