Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dgradients_util.py477 src_graph=None): argument
501 unconnected_gradients, src_graph)
505 if src_graph is None:
506 src_graph = ops.get_default_graph()
516 curr_graph = src_graph
606 src_graph._is_function(op.type) or is_partitioned_call)
616 func_call = src_graph._get_function( # pylint: disable=protected-access
621 if not func_call and hasattr(src_graph, "outer_graph"):
622 graph = src_graph.outer_graph
632 func_call = src_graph._get_function(op.type) # pylint: disable=protected-access
[all …]
H A Dcond_v2.py423 src_graph=func_graph)
H A Dwhile_v2.py754 ys, xs, grad_ys=grad_ys, src_graph=func_graph,
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dgraph_partition.cc1073 GraphDef* src_graph = &(*partitions)[opts.node_to_loc(src)]; in Partition() local
1074 if (src_graph == dst_graph && !NeedSameDeviceSendRecv(edge, g_info)) { in Partition()
1133 NodeDef* dummy = AddDummyConst(opts, src_graph, edge, &status); in Partition()
1155 NodeDef* send = AddSend(opts, g_info, src_graph, edge, send_from, in Partition()
1166 if (src_graph == dst_graph) { in Partition()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dfunction.py618 src_graph=self._func_graph)
834 src_graph=self._func_graph)
1036 src_graph=forward_wrapper.graph)
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api.cc1865 tensorflow::Status CopyGraph(Graph* src_graph, Graph* dst_graph, in CopyGraph() argument
1875 src_graph->ToGraphDef(&gdef); in CopyGraph()