Home
last modified time | relevance | path

Searched refs:GraphView (Results 1 – 25 of 33) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dgraph_view_test.cc42 GraphView graph_view(&graph_def); in TEST_F()
77 GraphView graph_view(&graph_def); in TEST_F()
109 GraphView graph_view(&graph_def); in TEST_F()
132 GraphView graph(&item.graph); in TEST_F()
134 GraphView::InputPort input = graph.GetInputPort("AddN", 0); in TEST_F()
137 GraphView::OutputPort fanin = graph.GetRegularFanin(input); in TEST_F()
148 GraphView::OutputPort output = graph.GetOutputPort("AddN", 0); in TEST_F()
190 GraphView graph(&item.graph); in TEST_F()
192 GraphView::OutputPort output = graph.GetOutputPort("a", -1); in TEST_F()
208 GraphView::InputPort input = graph.GetInputPort("d", -1); in TEST_F()
[all …]
H A Dgraph_view.cc33 bool HasSingleFanoutNode(const GraphView& graph_view, const NodeDef* node, in HasSingleFanoutNode()
35 const auto output = GraphView::OutputPort(node, port); in HasSingleFanoutNode()
39 bool HasFanouts(const GraphView& graph_view, const NodeDef* node, int port) { in HasFanouts()
40 const auto output = GraphView::OutputPort(node, port); in HasFanouts()
44 bool HasControlFanin(const GraphView& graph_view, const NodeDef* node) { in HasControlFanin()
45 const auto control_port = GraphView::InputPort(node, Graph::kControlSlot); in HasControlFanin()
49 bool HasControlFanout(const GraphView& graph_view, const NodeDef* node) { in HasControlFanout()
50 const auto control_port = GraphView::OutputPort(node, Graph::kControlSlot); in HasControlFanout()
54 bool HasControlFaninOrFanout(const GraphView& graph_view, const NodeDef* node) { in HasControlFaninOrFanout()
H A Dgraph_view.h400 class GraphView
403 explicit GraphView(const GraphDef* graph) : GraphViewInternal(graph) { in GraphView() function
410 bool HasSingleFanoutNode(const GraphView& graph_view, const NodeDef* node,
414 bool HasFanouts(const GraphView& graph_view, const NodeDef* node, int port = 0);
416 bool HasControlFanin(const GraphView& graph_view, const NodeDef* node);
418 bool HasControlFanout(const GraphView& graph_view, const NodeDef* node);
420 bool HasControlFaninOrFanout(const GraphView& graph_view, const NodeDef* node);
H A Dgraph_topology_view.cc43 const absl::Span<const GraphView::Edge> ephemeral_edges, in InitializeFromGraph()
64 for (const GraphView::Edge& edge : ephemeral_edges) { in InitializeFromGraph()
145 const absl::Span<const GraphView::Edge> ephemeral_edges) { in InitializeFromGraph()
152 return InitializeFromGraph(graph, absl::Span<GraphView::Edge>(), in InitializeFromGraph()
157 return InitializeFromGraph(graph, absl::Span<GraphView::Edge>(), in InitializeFromGraph()
H A Dmutable_graph_view.h53 const GraphView::OutputPort& port) const;
57 const GraphView::InputPort& port) const;
60 const OutputPort GetRegularFanin(const GraphView::InputPort& port) const;
H A Dgraph_topology_view.h59 absl::Span<const GraphView::Edge> ephemeral_edges,
62 absl::Span<const GraphView::Edge> ephemeral_edges);
H A Dmutable_graph_view_test.cc89 GraphView immutable_graph(mutable_graph.graph()); in CheckGraph()
96 const absl::flat_hash_set<GraphView::Edge>& immutable_edges) { in CheckGraph()
99 GraphView::Edge immutable_edge( in CheckGraph()
116 GraphView::OutputPort immutable_fanin(fanin.node, fanin.port_id); in CheckGraph()
126 GraphView::InputPort immutable_fanout(fanout.node, fanout.port_id); in CheckGraph()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dgraph_view.h39 class GraphView; variable
44 class FaninView : public internal::NodeIndexAndPortIndex<NodeView, GraphView> {
48 FaninView(GraphView* graph_view, int node_index, int port_index) in FaninView()
55 friend class GraphView; variable
61 class FanoutView : public internal::NodeIndexAndPortIndex<NodeView, GraphView> {
65 FanoutView(GraphView* graph_view, int node_index, int port_index) in FanoutView()
72 friend class GraphView; variable
80 GraphView, true> {
82 explicit NodeView(GraphView* graph_view, int node_index) in NodeView()
109 friend class GraphView; variable
[all …]
H A Dtopological_sort.cc35 std::vector<GraphView::Edge> MakeEphemeralEdges( in MakeEphemeralEdges()
37 std::vector<GraphView::Edge> ephemeral_edges; in MakeEphemeralEdges()
41 GraphView::OutputPort(dep.from, Graph::kControlSlot), in MakeEphemeralEdges()
42 GraphView::InputPort(dep.to, Graph::kControlSlot)); in MakeEphemeralEdges()
H A Dgraph_view_test.cc52 return std::is_same<T, class GraphView>::value ? "GraphView" in GetGraphViewTypeAsString()
56 using GraphViewTypes = ::testing::Types<GraphView, MutableGraphView>;
318 ::testing::Types<std::pair<FanoutView, GraphView>,
415 if (std::is_same<GraphViewType, GraphView>::value) { in TYPED_TEST()
431 ::testing::Types<std::pair<FaninView, GraphView>,
548 if (std::is_same<GraphViewType, GraphView>::value) { in TYPED_TEST()
601 if (std::is_same<TypeParam, GraphView>::value) { in TYPED_TEST()
650 if (std::is_same<TypeParam, GraphView>::value) { in TYPED_TEST()
798 GraphView expected_graph_view(&expected_graph, &s); in CompareGraphViewWithGraph()
2407 BM_GraphViewTConstruction<GraphView>(state); in BM_GraphViewConstruction()
[all …]
H A Dframe.cc144 Status FrameView::InferFromGraphView(const utils::GraphView& graph_view) { in InferFromGraphView()
155 utils::GraphView graph_view(&graph, &status); in InferFromGraph()
H A Dframe.h42 Status InferFromGraphView(const utils::GraphView& graph_view);
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_test.cc243 utils::GraphView graph_view(&output, &status); in TEST_F()
270 utils::GraphView graph_view(&output, &status); in TEST_F()
289 utils::GraphView graph_view(&output, &status); in TEST_F()
312 utils::GraphView graph_view(&output, &status); in TEST_F()
331 utils::GraphView graph_view(&output, &status); in TEST_F()
354 utils::GraphView graph_view(&output, &status); in TEST_F()
376 utils::GraphView graph_view_original(&item.graph, &status); in TEST_F()
385 utils::GraphView graph_view(&output, &status); in TEST_F()
410 utils::GraphView graph_view(&output, &status); in TEST_F()
444 utils::GraphView graph_view(&output, &status); in TEST_F()
[all …]
H A Dmodel_pruner.cc38 bool IsTrivialIdentity(const NodeDef& node, const GraphView& graph_view) { in IsTrivialIdentity()
60 bool IsTrivialOp(const NodeDef& node, const GraphView& graph_view) { in IsTrivialOp()
82 const GraphView& graph_view) { in RemovalIncreasesEdgeCount()
104 bool CanRemoveNode(const NodeDef& node, const GraphView& graph_view, in CanRemoveNode()
149 const GraphView& graph_view) { in ForwardInputsInternal()
183 const GraphView& graph_view) { in ForwardInputs()
453 GraphView graph_view(pruned_graph); in Optimize()
H A Dpin_to_host_optimizer.cc90 Status IsNodeOutputPortHostFriendly(const GraphView& graph, in IsNodeOutputPortHostFriendly()
221 Status IsNodeHostCandidate(const GraphView& graph, GraphProperties* properties, in IsNodeHostCandidate()
243 for (const GraphView::OutputPort& fanin : in IsNodeHostCandidate()
308 GraphView graph(optimized_graph); in Optimize()
356 for (const GraphView::InputPort& fanout : graph.GetFanouts(*node, false)) { in Optimize()
H A Dauto_mixed_precision_test.cc74 GraphView optimized_view(&optimized_graph); in VerifyGraphsEquivalent()
195 GraphView output_view(&output); in TestSimpleUnaryInferOp()
236 GraphView output_view(&output); in TEST_F()
272 GraphView output_view(&output); in TEST_F()
315 GraphView output_view(&output); in TEST_F()
367 GraphView output_view(&output); in TEST_F()
412 GraphView output_view(&output); in TEST_F()
453 GraphView output_view(&output); in TEST_F()
498 GraphView output_view(&output); in TEST_F()
541 GraphView output_view(&output); in TEST_F()
[all …]
H A Dloop_optimizer_test.cc111 utils::GraphView view(&graph, &status); in TEST_F()
131 utils::GraphView view(&output, &status); in TEST_F()
178 utils::GraphView view(&graph, &status); in TEST_F()
198 utils::GraphView view(&output, &status); in TEST_F()
245 utils::GraphView view(&graph, &status); in TEST_F()
260 utils::GraphView view(&output, &status); in TEST_F()
319 utils::GraphView view(&graph, &status); in TEST_F()
344 utils::GraphView view(&output, &status); in TEST_F()
412 utils::GraphView view(&graph, &status); in TEST_F()
432 utils::GraphView view(&output, &status); in TEST_F()
[all …]
H A Dloop_optimizer.cc603 GraphView::InputPort switch_loopcond_port(&switch_node, 1); in CheckForDeadFanout()
619 GraphView::InputPort switch_input_port(&switch_node, 0); in CheckForDeadFanout()
766 absl::flat_hash_set<GraphView::OutputPort> identity_switches; in RemoveDeadBranches()
788 GraphView::OutputPort dead(&node, dead_fanout); in RemoveDeadBranches()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dcompute_batch_size_op.cc76 const grappler::GraphView& graph, in GetInputNode()
79 grappler::GraphView::InputPort input_port = in GetInputNode()
105 grappler::GraphView graph_view(&graph_def); in Compute()
118 const grappler::GraphView& graph) { in GetBatchSizeFromBatchNode()
155 int64_t GetBatchSize(const NodeDef& node, const grappler::GraphView& graph) { in GetBatchSize()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgraph_view.h150 friend class GraphView;
209 class GraphView {
211 GraphView() : space_(nullptr) {} in GraphView() function
212 ~GraphView();
252 TF_DISALLOW_COPY_AND_ASSIGN(GraphView);
H A Dgraph_view.cc50 GraphView::~GraphView() { in ~GraphView()
85 size_t GraphView::NodeItemBytes(const Node* n) { in NodeItemBytes()
124 char* GraphView::InitializeNode(char* ptr, const Node* n) { in InitializeNode()
247 Status GraphView::Initialize(const Graph* g) { in Initialize()
311 void GraphView::SetScopedAllocatorAttrs( in SetScopedAllocatorAttrs()
421 Status GraphView::SetAllocAttrs(const Graph* g, const Device* device) { in SetAllocAttrs()
H A Dimmutable_executor_state.h89 const GraphView& graph_view() const { return gview_; } in graph_view()
133 GraphView gview_;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Dpick_subgraphs.cc88 struct GraphView { struct
110 std::unordered_map<InferenceDeviceType, GraphView,
246 GraphView current_graph_view; in ProcessSubgraph()
372 std::queue<const GraphView*> processing_queue; in MakeDecisions()
374 const GraphView* preferred_graph_view; in MakeDecisions()
391 const GraphView* current = processing_queue.front(); in MakeDecisions()
402 const GraphView& input_subgraph_view = in MakeDecisions()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties.cc646 const GraphView& graph, in SymbolicShapeRefiner()
657 const GraphView& graph() const { return graph_; } in graph()
1004 const GraphView::InputPort port(node, dst_input); in UpdateNode()
1005 const GraphView::OutputPort fanin = graph_.GetRegularFanin(port); in UpdateNode()
2043 const GraphView& graph_;
2279 for (const GraphView::Edge fanin : shape_refiner->graph().GetFaninEdges( in UpdateMerge()
2321 GraphView::InputPort port(node, 0); in UpdateEnter()
2322 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(port); in UpdateEnter()
2493 GraphView::InputPort inp(enqueue_node, i); in UpdateEnqueue()
2494 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(inp); in UpdateEnqueue()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/internal/
H A Dtfprof_timeline_test.cc60 TEST_F(TFProfTimelineTest, GraphView) { in TEST_F() argument

12