Searched refs:RegisterFunctionBodyOutputs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 39 Status RegisterFunctionBodyOutputs(const OpRegistrationData& registration, in RegisterFunctionBodyOutputs() function
45 connectivity->RegisterFunctionBodyOutputs(node.name(), in RegisterFunctionBodyOutputs()
50 Status RegisterFunctionBodyOutputs(const FunctionLibraryDefinition& flib, in RegisterFunctionBodyOutputs() function
55 return RegisterFunctionBodyOutputs(*registration, node, connectivity); in RegisterFunctionBodyOutputs()
93 void GrapplerFunctionConnectivity::RegisterFunctionBodyOutputs( in RegisterFunctionBodyOutputs() function in tensorflow::grappler::GrapplerFunctionConnectivity
592 TF_RETURN_IF_ERROR(RegisterFunctionBodyOutputs(*registration, func_def_node, in MakeGrapplerFunctionItem()
674 RegisterFunctionBodyOutputs(flib, func_body_node, connectivity)); in RegisterGrapplerFunctionConnectivity()
|
| D | functions_test.cc | 107 connectivity.RegisterFunctionBodyOutputs("Add", {{"z", {0, 1}}}); in TEST_F()
108 connectivity.RegisterFunctionBodyOutputs("Func", in TEST_F()
173 connectivity.RegisterFunctionBodyOutputs("Add", {{"z", {0, 1}}}); in TEST_F()
174 connectivity.RegisterFunctionBodyOutputs("Func", in TEST_F()
|
| D | functions.h | 91 void RegisterFunctionBodyOutputs(const string& node_name,
|