Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dtopological_sort.cc37 std::vector<GraphView::Edge> ephemeral_edges; in MakeEphemeralEdges() local
38 ephemeral_edges.reserve(extra_dependencies.size()); in MakeEphemeralEdges()
40 ephemeral_edges.emplace_back( in MakeEphemeralEdges()
44 return ephemeral_edges; in MakeEphemeralEdges()
/external/tensorflow/tensorflow/core/grappler/
Dgraph_topology_view.cc43 const absl::Span<const GraphView::Edge> ephemeral_edges, in InitializeFromGraph() argument
64 for (const GraphView::Edge& edge : ephemeral_edges) { in InitializeFromGraph()
145 const absl::Span<const GraphView::Edge> ephemeral_edges) { in InitializeFromGraph() argument
146 return InitializeFromGraph(graph, ephemeral_edges, in InitializeFromGraph()
Dgraph_topology_view.h59 absl::Span<const GraphView::Edge> ephemeral_edges,
62 absl::Span<const GraphView::Edge> ephemeral_edges);
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc410 Status AddEphemeralEdges(absl::Span<const NodeTypeIdEdge> ephemeral_edges);
566 absl::Span<const NodeTypeIdEdge> ephemeral_edges) { in AddEphemeralEdges() argument
568 for (const NodeTypeIdEdge& edge : ephemeral_edges) { in AddEphemeralEdges()
1313 std::vector<NodeTypeIdEdge> ephemeral_edges; in Optimize() local
1319 FindTensorListImplicitFloat32Edges(cluster, &ephemeral_edges); in Optimize()
1321 TF_RETURN_IF_ERROR(graph_type_view_.AddEphemeralEdges(ephemeral_edges)); in Optimize()