/external/tensorflow/tensorflow/c/ |
python_api.h | 31 TF_Buffer* attr_value_proto, TF_Status* status);
|
c_api.h | 179 // TF_Buffer holds a pointer to a block of data and its associated length. 183 // By default, TF_Buffer itself does not do any memory management of the 186 typedef struct TF_Buffer { 190 } TF_Buffer; 194 TF_CAPI_EXPORT extern TF_Buffer* TF_NewBufferFromString(const void* proto, 198 TF_CAPI_EXPORT extern TF_Buffer* TF_NewBuffer(); 200 TF_CAPI_EXPORT extern void TF_DeleteBuffer(TF_Buffer*); 202 TF_CAPI_EXPORT extern TF_Buffer TF_GetBuffer(TF_Buffer* buffer); [all...] |
c_api_internal.h | 71 TF_Buffer op_list; 195 Status MessageToBuffer(const tensorflow::protobuf::Message& in, TF_Buffer* out);
|
python_api.cc | 29 TF_Buffer* attr_value_proto, TF_Status* status) {
|
c_api.cc | 310 TF_Buffer* TF_NewBuffer() { return new TF_Buffer{nullptr, 0, nullptr}; } 312 TF_Buffer* TF_NewBufferFromString(const void* proto, size_t proto_len) { 316 TF_Buffer* buf = new TF_Buffer; 325 void TF_DeleteBuffer(TF_Buffer* buffer) { 333 TF_Buffer TF_GetBuffer(TF_Buffer* buffer) { return *buffer; } 620 TF_Buffer* out) { 622 return InvalidArgument("Passing non-empty TF_Buffer is invalid.") [all...] |
c_test_util.cc | 342 TF_Buffer* buffer = TF_NewBuffer(); 354 TF_Buffer* buffer = TF_NewBuffer(); 366 TF_Buffer* buffer = TF_NewBuffer(); 378 TF_Buffer* buffer = TF_NewBuffer();
|
c_api_test.cc | 197 TF_Buffer op_list_buf = TF_GetOpList(lib); 255 TF_Buffer* run_options = TF_NewBufferFromString("", 0); 256 TF_Buffer* run_metadata = TF_NewBuffer(); 331 TF_Buffer* buf = TF_GetAllOpList(); 651 TF_Buffer* graph_def = TF_NewBuffer(); 828 TF_Buffer* graph_def = TF_NewBuffer(); [all...] |
c_api_function.cc | 584 void TF_FunctionToFunctionDef(TF_Function* func, TF_Buffer* output_func_def, 617 TF_Buffer* output_attr_value,
|
c_api_function_test.cc | 379 TF_Buffer* attr_buf = TF_NewBuffer(); [all...] |
while_loop_test.cc | 116 TF_Buffer* buf = TF_NewBuffer();
|
/external/tensorflow/tensorflow/python/client/ |
tf_session_helper.h | 57 void TF_Run_wrapper(TF_DeprecatedSession* session, const TF_Buffer* run_options, 60 PyObjectVector* out_values, TF_Buffer* run_outputs); 122 void TF_SessionRun_wrapper(TF_Session* session, const TF_Buffer* run_options, 127 TF_Buffer* run_metadata, TF_Status* out_status,
|
tf_session_helper.cc | 46 const TF_Buffer* run_options, PyObject* feed_dict, 50 TF_Buffer* run_outputs) { 149 void TF_Run_wrapper(TF_DeprecatedSession* session, const TF_Buffer* run_options, 152 PyObjectVector* out_values, TF_Buffer* run_outputs) { 193 const TF_Buffer* run_options, 198 TF_Buffer* run_metadata, 279 void TF_SessionRun_wrapper(TF_Session* session, const TF_Buffer* run_options, 284 TF_Buffer* run_metadata, TF_Status* out_status,
|
tf_session.i | 346 // Typemap for functions that return a TF_Buffer struct. This typemap creates a 347 // Python string from the TF_Buffer and returns it. The TF_Buffer.data string 348 // is not expected to be NULL-terminated, and TF_Buffer.length does not count 350 %typemap(out) TF_Buffer (TF_GetOpList,TF_GetBuffer) {
|
/external/tensorflow/tensorflow/java/src/main/native/ |
tensorflow_jni.cc | 29 TF_Buffer* buf = TF_GetAllOpList(); 57 TF_Buffer buf = TF_GetOpList(reinterpret_cast<TF_Library*>(handle));
|
saved_model_bundle_jni.cc | 31 TF_Buffer* crun_options = nullptr; 53 TF_Buffer* metagraph_def = TF_NewBuffer();
|
graph_jni.cc | 92 TF_Buffer* buf = 113 TF_Buffer* buf = TF_NewBuffer();
|
session_jni.cc | 89 void TF_MaybeDeleteBuffer(TF_Buffer* buf) { 94 typedef std::unique_ptr<TF_Buffer, decltype(&TF_MaybeDeleteBuffer)> 97 unique_tf_buffer MakeUniqueBuffer(TF_Buffer* buf) {
|
/external/tensorflow/tensorflow/c/eager/ |
c_api.h | 274 TF_Buffer* buf,
|
c_api.cc | [all...] |
c_api_test.cc | 605 TF_Buffer* b = TF_NewBuffer(); [all...] |
/external/tensorflow/tensorflow/contrib/tensorrt/segment/ |
segment_test.cc | 53 TF_Buffer* buffer = TF_NewBuffer();
|