Home
last modified time | relevance | path

Searched refs:TF_Function (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/c/
Dc_api_function.cc151 TF_Function* TF_GraphToFunctionWithControlOutputs( in TF_GraphToFunctionWithControlOutputs()
206 TF_Function* tf_function = new TF_Function(); in TF_GraphToFunctionWithControlOutputs()
226 TF_Function* TF_GraphToFunction(const TF_Graph* fn_body, const char* fn_name, in TF_GraphToFunction()
240 const char* TF_FunctionName(TF_Function* func) { in TF_FunctionName()
244 void TF_GraphCopyFunction(TF_Graph* g, const TF_Function* func, in TF_GraphCopyFunction()
245 const TF_Function* grad, TF_Status* status) { in TF_GraphCopyFunction()
272 int TF_GraphGetFunctions(TF_Graph* g, TF_Function** funcs, int max_func, in TF_GraphGetFunctions()
281 TF_Function* func = new TF_Function(); in TF_GraphGetFunctions()
289 void TF_FunctionToFunctionDef(TF_Function* func, TF_Buffer* output_func_def, in TF_FunctionToFunctionDef()
294 TF_Function* TF_FunctionImportFunctionDef(const void* proto, size_t proto_len, in TF_FunctionImportFunctionDef()
[all …]
Dc_api.h186 typedef struct TF_Function TF_Function; typedef
897 const TF_Function* func,
898 const TF_Function* grad,
915 TF_CAPI_EXPORT extern int TF_GraphGetFunctions(TF_Graph* g, TF_Function** funcs,
1121 TF_CAPI_EXPORT extern TF_Function* TF_GraphToFunction(
1141 TF_CAPI_EXPORT extern TF_Function* TF_GraphToFunctionWithControlOutputs(
1153 TF_CAPI_EXPORT extern const char* TF_FunctionName(TF_Function* func);
1161 TF_CAPI_EXPORT extern void TF_FunctionToFunctionDef(TF_Function* func,
1173 TF_CAPI_EXPORT extern TF_Function* TF_FunctionImportFunctionDef(
1181 TF_CAPI_EXPORT extern void TF_FunctionSetAttrValueProto(TF_Function* func,
[all …]
Dc_api_function_test.cc393 TF_Function* func_;
1213 void DefineFunction(const char* name, TF_Function** func, in DefineFunction()
1322 TF_Function* grad_func; in TEST_F()
1364 TF_Function* func1; in TEST_F()
1365 TF_Function* func2; in TEST_F()
1366 TF_Function* grad_func; in TEST_F()
1394 TF_Function* func; in TEST_F()
1395 TF_Function* grad_func; in TEST_F()
1433 TF_Function* grad_func1; in TEST_F()
1434 TF_Function* grad_func2; in TEST_F()
[all …]
Dc_api_internal.h158 struct TF_Function { struct
Dc_api_experimental.h119 TF_CAPI_EXPORT extern char* TF_FunctionDebugString(TF_Function* func,
Dc_api_experimental.cc58 typedef std::unique_ptr<TF_Function, decltype(&TF_DeleteFunction)>
192 char* TF_FunctionDebugString(TF_Function* func, size_t* len) { in TF_FunctionDebugString()
232 TF_Function* func = TF_FunctionImportFunctionDef( in CreateFunctionsFromTextProto()
Dc_test_util.h120 bool GetFunctionDef(TF_Function* func, tensorflow::FunctionDef* func_def);
Dc_test_util.cc412 bool GetFunctionDef(TF_Function* func, tensorflow::FunctionDef* func_def) { in GetFunctionDef()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h206 TF_Function* TF_GraphToFunction_wrapper(
Dtf_session_wrapper.cc899 [](TF_Function* func, const char* attr_name, py::bytes proto) { in PYBIND11_MODULE()
912 [](TF_Function* graph, TF_Buffer* output_func_def) { in PYBIND11_MODULE()
922 [](TF_Graph* graph, const TF_Function* func, const TF_Function* grad) { in PYBIND11_MODULE()
Dtf_session_helper.cc579 TF_Function* TF_GraphToFunction_wrapper( in TF_GraphToFunction_wrapper()
/external/tensorflow/tensorflow/c/eager/
Dc_api.h409 TF_Function* function,
Dc_api_experimental_test.cc262 TF_Function* fn = in TEST()
Dc_api.cc923 void TFE_ContextAddFunction(TFE_Context* ctx, TF_Function* function, in TFE_ContextAddFunction()
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc81 PYBIND11_MAKE_OPAQUE(TF_Function);
629 py::class_<TF_Function> TF_Function_class(m, "TF_Function"); in PYBIND11_MODULE()
740 m.def("TFE_ContextAddFunction", [](py::handle& ctx, TF_Function* func) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_test.cc493 std::unique_ptr<TF_Function, decltype(&TF_DeleteFunction)> function( in RegisterCollectiveMulFunction()