Searched refs:_FuncGraph (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 679 class _FuncGraph(ops.Graph): class
693 super(_FuncGraph, self).__init__(*args, **kwargs)
818 return super(_FuncGraph, self)._create_op_internal(
945 func_graph = _FuncGraph(name, capture_by_value, allowlisted_stateful_ops,
1272 if isinstance(g, _FuncGraph):
1288 if isinstance(g, _FuncGraph):
1304 if isinstance(g, _FuncGraph):
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 385 isinstance(graph, framework_function._FuncGraph)) # pylint: disable=protected-access
392 assert isinstance(func_graph, framework_function._FuncGraph) # pylint: disable=protected-access
511 … assert isinstance(curr_graph, framework_function._FuncGraph) # pylint: disable=protected-access
|
| D | script_ops.py | 325 if isinstance(graph, function._FuncGraph): # pylint: disable=protected-access
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 1966 graph, function._FuncGraph): # pylint: disable=protected-access
2022 graph, function._FuncGraph): # pylint: disable=protected-access
|
| D | tpu.py | 2094 if isinstance(graph, function._FuncGraph): # pylint: disable=protected-access
|