Searched refs:function_lib (Results 1 – 6 of 6) sorted by relevance
32 from tensorflow.python.eager import function as function_lib unknown513 self._function_spec = function_lib.FunctionSpec.from_function_and_signature(638 attributes[function_lib.IMPLEMENTS_ATTRIBUTE_NAME] = attr_value640 attributes[function_lib.IMPLEMENTS_ATTRIBUTE_NAME] = self._implements652 attributes[function_lib.SHARED_RENDEZVOUS_ATTRIBUTE_NAME] = share660 return function_lib.defun_with_attributes(769 self._function_spec = function_lib.FunctionSpec.from_function_and_signature(950 return function_lib.defun(fn_with_cond)(canon_args, canon_kwds,1077 @function_lib.defun(autograph=False)1133 @function_lib.defun[all …]
37 from tensorflow.python.eager import function as function_lib unknown55 compat.as_bytes(function_lib._FORWARD_PREFIX),56 compat.as_bytes(function_lib._BACKWARD_PREFIX),57 compat.as_bytes(function_lib._INFERENCE_PREFIX))866 function_lib.add_function_callback(897 function_lib.remove_function_callback(
28 from tensorflow.python.eager import function as function_lib unknown153 return function_lib.FunctionSpec(fullargspec=fullargspec,369 func = function_lib.ConcreteFunction(func_graph)528 _FUNCTION_WRAPPER_NAME_REGEX = r"^%s(.*)_\d+$" % (function_lib._INFERENCE_PREFIX
315 FunctionDefLibrary function_lib = graph_def->library(); in FunctionalizeControlFlowForGraphDef() local322 std::swap(*graph_def->mutable_library(), function_lib); in FunctionalizeControlFlowForGraphDef()
2271 FunctionDefLibrary function_lib; in TEST_F() local2272 function_lib.add_function()->Swap(&f); in TEST_F()2274 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(function_lib)); in TEST_F()2317 FunctionDefLibrary function_lib; in TEST_F() local2318 function_lib.add_function()->Swap(&f); in TEST_F()2320 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(function_lib)); in TEST_F()
945 FunctionDefLibrary function_lib; in TEST_F() local963 function_lib.add_function()->Swap(&func1); in TEST_F()966 TF_CHECK_OK(s.graph()->AddFunctionLibrary(function_lib)); in TEST_F()