Home
last modified time | relevance | path

Searched refs:framework_function (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dgradients_util.py28 from tensorflow.python.framework import function as framework_function unknown
373 isinstance(graph, framework_function._FuncGraph)) # pylint: disable=protected-access
380 assert isinstance(func_graph, framework_function._FuncGraph) # pylint: disable=protected-access
522 … assert isinstance(curr_graph, framework_function._FuncGraph) # pylint: disable=protected-access
H A Dgradients_test.py28 from tensorflow.python.framework import function as framework_function unknown
511 return framework_function.Defun(dtypes.float32, dtypes.float32, **
549 grad_func = framework_function.Defun(dtypes.float32, dtypes.float32,
572 grad_func = framework_function.Defun(dtypes.float32, dtypes.float32,
599 @framework_function.Defun()
1189 @framework_function.Defun(dtypes.float32)
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dload_v1_in_v2_test.py27 from tensorflow.python.framework import function as framework_function unknown
590 @framework_function.Defun(dtypes.int64)
594 @framework_function.Defun(dtypes.int64)
598 @framework_function.Defun(dtypes.int64)
H A Dload_test.py44 from tensorflow.python.framework import function as framework_function unknown
1914 @framework_function.Defun(dtypes.float32)