Home
last modified time | relevance | path

Searched refs:TF_Buffer (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/c/
Dc_api.h187 typedef struct TF_Buffer { struct
191 } TF_Buffer; argument
195 TF_CAPI_EXPORT extern TF_Buffer* TF_NewBufferFromString(const void* proto,
199 TF_CAPI_EXPORT extern TF_Buffer* TF_NewBuffer(void);
201 TF_CAPI_EXPORT extern void TF_DeleteBuffer(TF_Buffer*);
203 TF_CAPI_EXPORT extern TF_Buffer TF_GetBuffer(TF_Buffer* buffer);
851 TF_Operation* oper, const char* attr_name, TF_Buffer* value,
859 TF_Operation* oper, const char* attr_name, TF_Buffer** values,
887 TF_Operation* oper, const char* attr_name, TF_Buffer* output_attr_value,
911 TF_Buffer* output_graph_def,
[all …]
Dc_api.cc394 TF_Buffer* TF_NewBuffer() { return new TF_Buffer{nullptr, 0, nullptr}; } in TF_NewBuffer()
396 TF_Buffer* TF_NewBufferFromString(const void* proto, size_t proto_len) { in TF_NewBufferFromString()
400 TF_Buffer* buf = new TF_Buffer; in TF_NewBufferFromString()
409 void TF_DeleteBuffer(TF_Buffer* buffer) { in TF_DeleteBuffer()
418 TF_Buffer TF_GetBuffer(TF_Buffer* buffer) { return *buffer; } in TF_GetBuffer()
649 TF_Buffer* out) { in MessageToBuffer()
834 Session* session, const char* handle, const TF_Buffer* run_options, in TF_Run_Helper()
840 const std::vector<string>& target_oper_names, TF_Buffer* run_metadata, in TF_Run_Helper()
892 void TF_Run(TF_DeprecatedSession* s, const TF_Buffer* run_options, in TF_Run()
899 TF_Buffer* run_metadata, TF_Status* status) { in TF_Run()
[all …]
Dc_api_experimental.h71 TF_CAPI_EXPORT extern TF_Buffer* TF_CreateConfig(
78 TF_CAPI_EXPORT extern TF_Buffer* TF_CreateRunOptions(
145 TF_Buffer* TFE_GetServerDef(const char* text_proto, TF_Status* status);
Dc_api_internal.h75 TF_Buffer op_list;
211 TF_Buffer* out);
Dc_api_test.cc203 TF_Buffer op_list_buf = TF_GetOpList(lib); in TEST()
212 TF_Buffer* op_list_buffer = TF_GetAllOpList(); in TEST()
276 TF_Buffer* run_options = TF_NewBufferFromString("", 0); in TEST()
277 TF_Buffer* run_metadata = TF_NewBuffer(); in TEST()
352 TF_Buffer* buf = TF_GetAllOpList(); in TEST()
672 TF_Buffer* graph_def = TF_NewBuffer(); in TEST()
849 TF_Buffer* graph_def = TF_NewBuffer(); in TEST()
900 TF_Buffer* graph_def = TF_NewBuffer(); in TEST()
1161 TF_Buffer* buffer = TF_NewBuffer(); in TEST()
1357 TF_Buffer* run_options = TF_NewBufferFromString("", 0); in TEST()
[all …]
Dpython_api.h33 TF_Buffer* attr_value_proto, TF_Status* status);
Dc_test_util.cc390 TF_Buffer* buffer = TF_NewBuffer(); in GetGraphDef()
402 TF_Buffer* buffer = TF_NewBuffer(); in GetNodeDef()
414 TF_Buffer* buffer = TF_NewBuffer(); in GetFunctionDef()
426 TF_Buffer* buffer = TF_NewBuffer(); in GetAttrValue()
Dkernels_test.cc106 TF_Buffer* buf = TF_GetRegisteredKernelsForOp(op_name, status); in TEST()
Dc_api_experimental_test.cc144 TF_Buffer* result = TFE_GetServerDef(expected_text_proto.c_str(), status); in TEST()
156 TF_Buffer* null_result = in TEST()
Dpython_api.cc30 TF_Buffer* attr_value_proto, TF_Status* status) { in SetAttr()
Dc_api_function.cc690 void TF_FunctionToFunctionDef(TF_Function* func, TF_Buffer* output_func_def, in TF_FunctionToFunctionDef()
723 TF_Buffer* output_attr_value, in TF_FunctionGetAttrValueProto()
Dc_api_function_test.cc382 TF_Buffer* attr_buf = TF_NewBuffer(); in GetAttr()
1478 TF_Buffer* attr_buf = TF_NewBuffer(); in TEST_F()
1539 TF_Buffer* buffer = TF_NewBuffer(); in TEST_F()
1584 TF_Buffer* buffer = TF_NewBuffer(); in TEST_F()
Dc_api_experimental.cc69 TF_Buffer* TF_CreateConfig(unsigned char enable_xla_compilation, in TF_CreateConfig()
106 TF_Buffer* ret = TF_NewBuffer(); in TF_CreateConfig()
111 TF_Buffer* TF_CreateRunOptions(unsigned char enable_full_trace) { in TF_CreateRunOptions()
118 TF_Buffer* ret = TF_NewBuffer(); in TF_CreateRunOptions()
8530 TF_Buffer* TFE_GetServerDef(const char* text_proto, TF_Status* status) { in TFE_GetServerDef()
8539 TF_Buffer* ret = TF_NewBuffer(); in TFE_GetServerDef()
Dwhile_loop_test.cc116 TF_Buffer* buf = TF_NewBuffer(); in GraphDebugString()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h60 void TF_Run_wrapper(TF_DeprecatedSession* session, const TF_Buffer* run_options,
63 PyObjectVector* out_values, TF_Buffer* run_outputs);
67 const TF_Buffer* callable_options,
71 const TF_Buffer* callable_options,
79 TF_Buffer* run_metadata);
82 PyObjectVector* out_values, TF_Buffer* run_metadata);
149 void TF_SessionRun_wrapper(TF_Session* session, const TF_Buffer* run_options,
154 TF_Buffer* run_metadata, TF_Status* status,
Dtf_session_helper.cc60 const TF_Buffer* run_options, PyObject* feed_dict, in TF_Run_wrapper_helper()
64 TF_Buffer* run_outputs) { in TF_Run_wrapper_helper()
163 void TF_Run_wrapper(TF_DeprecatedSession* session, const TF_Buffer* run_options, in TF_Run_wrapper()
166 PyObjectVector* out_values, TF_Buffer* run_outputs) { in TF_Run_wrapper()
174 const TF_Buffer* callable_options, int64_t* out_handle, in MakeCallableHelper()
196 const TF_Buffer* callable_options, in TF_DeprecatedSessionMakeCallable()
203 const TF_Buffer* callable_options, in TF_SessionMakeCallable()
212 PyObjectVector* out_values, TF_Buffer* run_metadata) { in RunCallableHelper()
296 TF_Buffer* run_metadata) { in TF_DeprecatedSessionRunCallable()
304 TF_Buffer* run_metadata) { in TF_SessionRunCallable()
[all …]
Dtf_session.i374 %typemap(out) TF_Buffer (TF_GetOpList,TF_GetBuffer) {
/external/tensorflow/tensorflow/java/src/main/native/
Dtensorflow_jni.cc29 TF_Buffer* buf = TF_GetAllOpList(); in Java_org_tensorflow_TensorFlow_registeredOpList()
57 TF_Buffer buf = TF_GetOpList(reinterpret_cast<TF_Library*>(handle)); in Java_org_tensorflow_TensorFlow_libraryOpList()
Dsession_jni.cc59 void TF_MaybeDeleteBuffer(TF_Buffer* buf) { in TF_MaybeDeleteBuffer()
64 typedef std::unique_ptr<TF_Buffer, decltype(&TF_MaybeDeleteBuffer)>
67 unique_tf_buffer MakeUniqueBuffer(TF_Buffer* buf) { in MakeUniqueBuffer()
Dsaved_model_bundle_jni.cc44 TF_Buffer* crun_options = nullptr; in Java_org_tensorflow_SavedModelBundle_load()
66 TF_Buffer* metagraph_def = TF_NewBuffer(); in Java_org_tensorflow_SavedModelBundle_load()
Dgraph_jni.cc101 TF_Buffer* buf = in Java_org_tensorflow_Graph_importGraphDef()
122 TF_Buffer* buf = TF_NewBuffer(); in Java_org_tensorflow_Graph_toGraphDef()
/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental.h45 TF_Buffer* buf,
Dc_api_experimental.cc40 TF_Buffer* buf, TF_Status* status) { in TFE_ProfilerSerializeToString()
Dc_api_experimental_test.cc72 TF_Buffer* profiler_result = TF_NewBuffer(); in ExecuteWithProfiling()
Dc_api.h418 TF_Buffer* buf,

12