Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dgradients_util.py32 from tensorflow.python.framework import function as framework_function unknown
383 isinstance(graph, framework_function._FuncGraph)) # pylint: disable=protected-access
390 assert isinstance(func_graph, framework_function._FuncGraph) # pylint: disable=protected-access
509 … assert isinstance(curr_graph, framework_function._FuncGraph) # pylint: disable=protected-access
Dgradients_test.py31 from tensorflow.python.framework import function as framework_function unknown
513 return framework_function.Defun(dtypes.float32, dtypes.float32, **
551 grad_func = framework_function.Defun(dtypes.float32, dtypes.float32,
574 grad_func = framework_function.Defun(dtypes.float32, dtypes.float32,
601 @framework_function.Defun()
1191 @framework_function.Defun(dtypes.float32)
/external/tensorflow/tensorflow/python/saved_model/
Dload_v1_in_v2_test.py31 from tensorflow.python.framework import function as framework_function unknown
594 @framework_function.Defun(dtypes.int64)
598 @framework_function.Defun(dtypes.int64)
602 @framework_function.Defun(dtypes.int64)
Dload_test.py44 from tensorflow.python.framework import function as framework_function unknown
1903 @framework_function.Defun(dtypes.float32)