| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view.cc | 66 bool HasSingleFanoutNode(const GraphView& graph_view, const NodeDef* node, in HasSingleFanoutNode()
68 const auto output = GraphView::OutputPort(node, port); in HasSingleFanoutNode()
72 bool HasFanouts(const GraphView& graph_view, const NodeDef* node, int port) { in HasFanouts()
73 const auto output = GraphView::OutputPort(node, port); in HasFanouts()
77 bool HasControlFanin(const GraphView& graph_view, const NodeDef* node) { in HasControlFanin()
78 const auto control_port = GraphView::InputPort(node, Graph::kControlSlot); in HasControlFanin()
82 bool HasControlFanout(const GraphView& graph_view, const NodeDef* node) { in HasControlFanout()
83 const auto control_port = GraphView::OutputPort(node, Graph::kControlSlot); in HasControlFanout()
87 bool HasControlFaninOrFanout(const GraphView& graph_view, const NodeDef* node) { in HasControlFaninOrFanout()
|
| D | graph_view_test.cc | 42 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 …]
|
| D | graph_topology_view.cc | 43 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()
|
| D | graph_view.h | 400 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);
|
| D | mutable_graph_view.h | 53 const GraphView::OutputPort& port) const;
57 const GraphView::InputPort& port) const;
60 const OutputPort GetRegularFanin(const GraphView::InputPort& port) const;
|
| D | graph_topology_view.h | 59 absl::Span<const GraphView::Edge> ephemeral_edges,
62 absl::Span<const GraphView::Edge> ephemeral_edges);
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view.h | 39 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 …]
|
| D | topological_sort.cc | 35 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()
|
| D | graph_view_test.cc | 52 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 …]
|
| D | frame.cc | 144 Status FrameView::InferFromGraphView(const utils::GraphView& graph_view) { in InferFromGraphView()
155 utils::GraphView graph_view(&graph, &status); in InferFromGraph()
|
| D | frame.h | 42 Status InferFromGraphView(const utils::GraphView& graph_view);
|
| D | frame_test.cc | 30 ::testing::Types<GraphDef, utils::GraphView, utils::MutableGraphView>;
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | compute_batch_size_op.cc | 74 const grappler::GraphView& graph, in GetInputNode()
77 grappler::GraphView::InputPort input_port = in GetInputNode()
103 grappler::GraphView graph_view(&graph_def); in Compute()
116 const grappler::GraphView& graph) { in GetBatchSizeFromBatchNode()
153 int64 GetBatchSize(const NodeDef& node, const grappler::GraphView& graph) { in GetBatchSize()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_view.h | 148 friend class GraphView;
207 class GraphView {
209 GraphView() : space_(nullptr) {} in GraphView() function
210 ~GraphView();
250 TF_DISALLOW_COPY_AND_ASSIGN(GraphView);
|
| D | graph_view.cc | 48 GraphView::~GraphView() { in ~GraphView()
83 size_t GraphView::NodeItemBytes(const Node* n) { in NodeItemBytes()
122 char* GraphView::InitializeNode(char* ptr, const Node* n) { in InitializeNode()
245 Status GraphView::Initialize(const Graph* g) { in Initialize()
294 void GraphView::SetScopedAllocatorAttrs( in SetScopedAllocatorAttrs()
404 Status GraphView::SetAllocAttrs(const Graph* g, const Device* device) { in SetAllocAttrs()
|
| D | immutable_executor_state.h | 89 const GraphView& graph_view() const { return gview_; } in graph_view()
133 GraphView gview_;
|
| D | simple_propagator_state.cc | 74 const GraphView& gview = immutable_state_.graph_view(); in PropagateOutputs()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_test.cc | 243 utils::GraphView graph_view(&output, &status); in TEST_F()
288 utils::GraphView graph_view(&output, &status); in TEST_F()
307 utils::GraphView graph_view(&output, &status); in TEST_F()
330 utils::GraphView graph_view(&output, &status); in TEST_F()
349 utils::GraphView graph_view(&output, &status); in TEST_F()
372 utils::GraphView graph_view(&output, &status); in TEST_F()
394 utils::GraphView graph_view_original(&item.graph, &status); in TEST_F()
403 utils::GraphView graph_view(&output, &status); in TEST_F()
428 utils::GraphView graph_view(&output, &status); in TEST_F()
474 utils::GraphView graph_view(&output, &status); in TEST_F()
[all …]
|
| D | pin_to_host_optimizer.cc | 91 Status IsNodeOutputPortHostFriendly(const GraphView& graph, in IsNodeOutputPortHostFriendly()
222 Status IsNodeHostCandidate(const GraphView& graph, GraphProperties* properties, in IsNodeHostCandidate()
244 for (const GraphView::OutputPort& fanin : in IsNodeHostCandidate()
309 GraphView graph(optimized_graph); in Optimize()
357 for (const GraphView::InputPort& fanout : graph.GetFanouts(*node, false)) { in Optimize()
|
| D | model_pruner.cc | 38 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()
|
| D | loop_optimizer_test.cc | 111 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 …]
|
| D | auto_mixed_precision_test.cc | 74 GraphView optimized_view(&optimized_graph); in VerifyGraphsEquivalent()
190 GraphView output_view(&output); in TestSimpleUnaryInferOp()
231 GraphView output_view(&output); in TEST_F()
267 GraphView output_view(&output); in TEST_F()
310 GraphView output_view(&output); in TEST_F()
355 GraphView output_view(&output); in TEST_F()
396 GraphView output_view(&output); in TEST_F()
441 GraphView output_view(&output); in TEST_F()
484 GraphView output_view(&output); in TEST_F()
541 GraphView output_view(&output); in TEST_F()
[all …]
|
| D | loop_optimizer.cc | 603 GraphView::InputPort switch_loopcond_port(&switch_node, 1); in CheckForDeadFanout()
619 GraphView::InputPort switch_input_port(&switch_node, 0); in CheckForDeadFanout()
769 absl::flat_hash_set<GraphView::OutputPort> identity_switches; in RemoveDeadBranches()
791 GraphView::OutputPort dead(&node, dead_fanout); in RemoveDeadBranches()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_timeline_test.cc | 63 TEST_F(TFProfTimelineTest, GraphView) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 617 const GraphView& graph, in SymbolicShapeRefiner()
628 const GraphView& graph() const { return graph_; } in graph()
975 const GraphView::InputPort port(node, dst_input); in UpdateNode()
976 const GraphView::OutputPort fanin = graph_.GetRegularFanin(port); in UpdateNode()
1999 const GraphView& graph_;
2235 for (const GraphView::Edge fanin : shape_refiner->graph().GetFaninEdges( in UpdateMerge()
2277 GraphView::InputPort port(node, 0); in UpdateEnter()
2278 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(port); in UpdateEnter()
2449 GraphView::InputPort inp(enqueue_node, i); in UpdateEnqueue()
2450 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(inp); in UpdateEnqueue()
[all …]
|