Home
last modified time | relevance | path

Searched refs:function_lib (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Ddef_function.py32 from tensorflow.python.eager import function as function_lib unknown
513 self._function_spec = function_lib.FunctionSpec.from_function_and_signature(
638 attributes[function_lib.IMPLEMENTS_ATTRIBUTE_NAME] = attr_value
640 attributes[function_lib.IMPLEMENTS_ATTRIBUTE_NAME] = self._implements
652 attributes[function_lib.SHARED_RENDEZVOUS_ATTRIBUTE_NAME] = share
660 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 …]
/external/tensorflow/tensorflow/python/debug/lib/
Ddumping_callback.py37 from tensorflow.python.eager import function as function_lib unknown
55 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(
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_deserialization.py28 from tensorflow.python.eager import function as function_lib unknown
153 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
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc315 FunctionDefLibrary function_lib = graph_def->library(); in FunctionalizeControlFlowForGraphDef() local
322 std::swap(*graph_def->mutable_library(), function_lib); in FunctionalizeControlFlowForGraphDef()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc2271 FunctionDefLibrary function_lib; in TEST_F() local
2272 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() local
2318 function_lib.add_function()->Swap(&f); in TEST_F()
2320 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(function_lib)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_test.cc945 FunctionDefLibrary function_lib; in TEST_F() local
963 function_lib.add_function()->Swap(&func1); in TEST_F()
966 TF_CHECK_OK(s.graph()->AddFunctionLibrary(function_lib)); in TEST_F()