Searched refs:function_lib (Results 1 – 6 of 6) sorted by relevance
77 from tensorflow.python.eager import function as function_lib unknown579 self._function_spec = function_lib.FunctionSpec.from_function_and_signature(696 attributes[function_lib.IMPLEMENTS_ATTRIBUTE_NAME] = attr_value698 attributes[function_lib.IMPLEMENTS_ATTRIBUTE_NAME] = self._implements710 attributes[function_lib.SHARED_RENDEZVOUS_ATTRIBUTE_NAME] = share718 return function_lib.defun_with_attributes(832 self._function_spec = function_lib.FunctionSpec.from_function_and_signature(1037 return function_lib.defun(fn_with_cond)(canon_args, canon_kwds,1105 @function_lib.defun(autograph=False)1161 @function_lib.defun[all …]
36 from tensorflow.python.eager import function as function_lib unknown54 compat.as_bytes(function_lib._FORWARD_PREFIX),55 compat.as_bytes(function_lib._BACKWARD_PREFIX),56 compat.as_bytes(function_lib._INFERENCE_PREFIX))862 function_lib.add_function_callback(893 function_lib.remove_function_callback(
28 from tensorflow.python.eager import function as function_lib unknown157 return function_lib.FunctionSpec(fullargspec=fullargspec,407 func = function_lib.ConcreteFunction(func_graph, attrs=copy.attr)610 _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()
2425 FunctionDefLibrary function_lib; in TEST_F() local2426 function_lib.add_function()->Swap(&f); in TEST_F()2428 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(function_lib)); in TEST_F()2471 FunctionDefLibrary function_lib; in TEST_F() local2472 function_lib.add_function()->Swap(&f); in TEST_F()2474 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(function_lib)); in TEST_F()
949 FunctionDefLibrary function_lib; in TEST_F() local967 function_lib.add_function()->Swap(&func1); in TEST_F()970 TF_CHECK_OK(s.graph()->AddFunctionLibrary(function_lib)); in TEST_F()