| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | graph_utils.h | 55 MutableGraphView* graph);
58 NodeDef* AddScalarPlaceholder(DataType dtype, MutableGraphView* graph);
62 NodeDef* AddScalarConstNode(T v, MutableGraphView* graph) { in AddScalarConstNode()
71 NodeDef* AddScalarConstNode(bool v, MutableGraphView* graph);
73 NodeDef* AddScalarConstNode(double v, MutableGraphView* graph);
75 NodeDef* AddScalarConstNode(float v, MutableGraphView* graph);
77 NodeDef* AddScalarConstNode(int v, MutableGraphView* graph);
79 NodeDef* AddScalarConstNode(int64 v, MutableGraphView* graph);
81 NodeDef* AddScalarConstNode(StringPiece v, MutableGraphView* graph);
125 NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph);
[all …]
|
| D | graph_utils_test.cc | 42 MutableGraphView graph(&graph_def); in TEST()
50 MutableGraphView graph(&graph_def); in TEST()
58 MutableGraphView graph(&graph_def); in TEST()
66 MutableGraphView graph(&graph_def); in TEST()
74 MutableGraphView graph(&graph_def); in TEST()
82 MutableGraphView graph(&graph_def); in TEST()
90 MutableGraphView graph(&graph_def); in TEST()
99 MutableGraphView graph(&graph_def); in TEST()
108 MutableGraphView graph(&graph_def); in TEST()
119 MutableGraphView graph(&graph_def); in TEST()
[all …]
|
| D | graph_utils.cc | 74 MutableGraphView* graph) { in AddScalarConstNodeHelper()
94 NodeDef* AddScalarPlaceholder(DataType dtype, MutableGraphView* graph) { in AddScalarPlaceholder()
107 MutableGraphView* graph) { in AddNode()
125 NodeDef* AddScalarConstNode(bool v, MutableGraphView* graph) { in AddScalarConstNode()
131 NodeDef* AddScalarConstNode(double v, MutableGraphView* graph) { in AddScalarConstNode()
137 NodeDef* AddScalarConstNode(float v, MutableGraphView* graph) { in AddScalarConstNode()
143 NodeDef* AddScalarConstNode(int v, MutableGraphView* graph) { in AddScalarConstNode()
149 NodeDef* AddScalarConstNode(int64 v, MutableGraphView* graph) { in AddScalarConstNode()
155 NodeDef* AddScalarConstNode(StringPiece v, MutableGraphView* graph) { in AddScalarConstNode()
271 NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph) { in GetInputNode()
[all …]
|
| D | noop_elimination.cc | 37 bool IsTakeAll(const NodeDef& take_node, const MutableGraphView& graph) { in IsTakeAll()
51 bool IsSkipNone(const NodeDef& skip_node, const MutableGraphView& graph) { in IsSkipNone()
57 bool IsRepeatOne(const NodeDef& repeat_node, const MutableGraphView& graph) { in IsRepeatOne()
64 const MutableGraphView& graph) { in IsPrefetchZero()
99 bool IsMapIdentity(const NodeDef& map_node, const MutableGraphView& graph) { in IsMapIdentity()
131 bool IsNoOp(const NodeDef& node, const MutableGraphView& graph) { in IsNoOp()
144 MutableGraphView graph(output); in OptimizeAndCollectStats()
|
| D | noop_elimination_test.cc | 39 MutableGraphView *graph) { in MakeUnaryNode()
47 MutableGraphView *graph) { in MakeUnaryNonConstNode()
54 NodeDef *MakeCacheNode(string input_node, MutableGraphView *graph) { in MakeCacheNode()
62 NodeDef *MakeRangeNode(MutableGraphView *graph) { in MakeRangeNode()
81 MutableGraphView graph(&item.graph); in TEST_P()
121 MutableGraphView graph(&item.graph); in TEST_P()
171 MutableGraphView graph(&item.graph); in TEST_P()
228 MutableGraphView graph(&item.graph); in TEST_P()
|
| D | auto_shard.cc | 140 Status AddShardNode(MutableGraphView* graph, const NodeDef& add_before, in AddShardNode()
205 Status AddShuffleDataset(MutableGraphView* graph, const NodeDef& add_before, in AddShuffleDataset()
234 Status AddShuffleDatasetV2(MutableGraphView* graph, const NodeDef& add_before, in AddShuffleDatasetV2()
257 Status AddShuffleDatasetV3(MutableGraphView* graph, const NodeDef& add_before, in AddShuffleDatasetV3()
306 Status RemoveShuffleDataset(MutableGraphView* graph, const NodeDef& node, in RemoveShuffleDataset()
331 Status RemoveShuffleDatasetV2(MutableGraphView* graph, const NodeDef& node, in RemoveShuffleDatasetV2()
353 Status RemoveShuffleDatasetV3(MutableGraphView* graph, const NodeDef& node, in RemoveShuffleDatasetV3()
380 Status ProcessDatasetSourceNode(MutableGraphView* graph, const NodeDef& node, in ProcessDatasetSourceNode()
424 FunctionLibraryDefinition* flib, MutableGraphView* graph, in FindFuncAndTensorSliceDataset()
451 MutableGraphView* graph, in RecursivelyHandleOp()
[all …]
|
| D | map_vectorization_test.cc | 71 FunctionDef* AddMapFn(MutableGraphView* graph) { in AddMapFn()
84 NodeDef* AddMapNode(MutableGraphView* graph, const string& input_dataset, in AddMapNode()
118 NodeDef* AddPrefetchNode(MutableGraphView* graph, const string& input_dataset, in AddPrefetchNode()
131 NodeDef* AddBatchNode(MutableGraphView* graph, const string& input_dataset, in AddBatchNode()
160 NodeDef* AddRangeNode(MutableGraphView* graph) { in AddRangeNode()
292 MutableGraphView graph(&item.graph); in TEST_P()
341 NodeDef* AddMapAndBatchNode(MutableGraphView* graph, in AddMapAndBatchNode()
369 MutableGraphView graph(&item.graph); in TEST_P()
401 MutableGraphView graph(&item.graph); in TEST_P()
494 NodeDef* AddArbitraryInputNode(MutableGraphView* graph, in AddArbitraryInputNode()
[all …]
|
| D | shuffle_and_repeat_fusion.cc | 48 MutableGraphView* graph, GraphDef* output, in FuseShuffleV1AndRepeat()
80 MutableGraphView* graph, GraphDef* output, in FuseShuffleV2AndRepeat()
119 MutableGraphView* graph, GraphDef* output, in FuseShuffleV3AndRepeat()
158 MutableGraphView graph(output); in OptimizeAndCollectStats()
|
| D | hoist_random_uniform.cc | 41 MutableGraphView* graph) { in MakeStatelessMap()
69 MutableGraphView* graph) { in MakeRandomDataset()
90 NodeDef MakeBatchTwo(const NodeDef& random_dataset, MutableGraphView* graph) { in MakeBatchTwo()
114 MutableGraphView* graph) { in MakeZipNode()
240 MutableGraphView graph(output); in OptimizeAndCollectStats()
|
| D | map_and_filter_fusion.cc | 47 MutableGraphView* graph) { in MakeFusedNode()
89 MutableGraphView* graph, FunctionDefLibrary* library) { in MakeFilterNode()
123 const FunctionDef& fused_map_func, MutableGraphView* graph, in MakeMapNode()
175 MutableGraphView graph(output); in OptimizeAndCollectStats()
|
| D | slack.cc | 80 Status Slack::RecursivelyHandleOp(const MutableGraphView& graph, in RecursivelyHandleOp()
117 MutableGraphView graph(output); in OptimizeAndCollectStats()
|
| D | latency_all_edges.cc | 40 Status MakeLatencyNode(const NodeDef& node, MutableGraphView* graph, in MakeLatencyNode()
85 MutableGraphView graph(output); in OptimizeAndCollectStats()
|
| D | batch_parallelization.cc | 38 NodeDef MakeParallelBatch(const string& name, MutableGraphView* graph) { in MakeParallelBatch()
70 MutableGraphView graph(output); in OptimizeAndCollectStats()
|
| D | map_parallelization.cc | 37 NodeDef MakeParallelMap(const string& name, MutableGraphView* graph) { in MakeParallelMap()
67 MutableGraphView graph(output); in OptimizeAndCollectStats()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view.h | 130 class MutableGraphView; variable
139 MutableGraphView> {
143 MutableFaninView(MutableGraphView* graph_view, int node_index, int port_index) in MutableFaninView()
146 explicit MutableFaninView(MutableGraphView* graph_view, int node_index, in MutableFaninView()
165 friend class MutableGraphView; variable
174 MutableGraphView> {
178 MutableFanoutView(MutableGraphView* graph_view, int node_index, in MutableFanoutView()
182 explicit MutableFanoutView(MutableGraphView* graph_view, int node_index, in MutableFanoutView()
198 friend class MutableGraphView; variable
209 MutableGraphView, false> {
[all …]
|
| D | graph_view_internal_test.cc | 44 const MutableGraphView* graph_view) { in GetUpdatedNodeNames()
53 using MutableNodeViewDiff = NodeViewDiff<MutableGraphView>;
59 MutableGraphView graph_view(&graph, &s); in TEST()
83 MutableGraphView graph_view(&graph, &s); in TEST()
107 MutableGraphView graph_view(&graph, &s); in TEST()
131 MutableGraphView graph_view(&graph, &s); in TEST()
177 MutableGraphView graph_view(&graph, &s); in TEST()
216 MutableGraphView graph_view(&graph, &s); in TEST()
300 MutableGraphView graph_view(&graph, &s); in TEST()
335 MutableGraphView graph_view(&graph, &s); in TEST()
[all …]
|
| D | graph_view_test.cc | 56 using GraphViewTypes = ::testing::Types<GraphView, MutableGraphView>;
209 MutableGraphView graph_view(&graph, &s); in TEST()
319 std::pair<MutableFanoutView, MutableGraphView>>;
432 std::pair<MutableFaninView, MutableGraphView>>;
795 void CompareGraphViewWithGraph(MutableGraphView* graph_view, in CompareGraphViewWithGraph()
974 MutableGraphView graph_view(&graph, &s); in TEST_F()
1011 MutableGraphView graph_view(&graph, &s); in TEST_F()
1033 MutableGraphView graph_view(&graph, &s); in TEST_F()
1059 MutableGraphView graph_view(&graph, &s); in TEST_F()
1080 MutableGraphView graph_view(&graph, &s); in TEST_F()
[all …]
|
| D | graph_view.cc | 223 Mutation::Mutation(MutableGraphView* graph_view) : graph_view_(graph_view) {} in Mutation()
464 MutableGraphView::MutableGraphView(GraphDef* graph, Status* status) in MutableGraphView() function in tensorflow::grappler::utils::MutableGraphView
490 Mutation* MutableGraphView::GetMutationBuilder() { return &mutation_; } in GetMutationBuilder()
492 bool MutableGraphView::AddUniqueNodeInternal(NodeDef* node) { in AddUniqueNodeInternal()
502 Status MutableGraphView::CheckFaninsInternal( in CheckFaninsInternal()
540 void MutableGraphView::AddFaninsInternal( in AddFaninsInternal()
606 Status MutableGraphView::GetNodeNamesAndPartitionUpdatedNodes( in GetNodeNamesAndPartitionUpdatedNodes()
692 Status MutableGraphView::RemovedOrMissingNodeFanoutsWellFormed( in RemovedOrMissingNodeFanoutsWellFormed()
770 Status MutableGraphView::CheckNodeNamesAndFanins( in CheckNodeNamesAndFanins()
807 Status MutableGraphView::CheckKernelRegisteredForNodes() { in CheckKernelRegisteredForNodes()
[all …]
|
| D | pattern_utils_test.cc | 169 MutableGraphView graph_view(&graph, &status); in TEST_F()
251 MutableGraphView graph_view(&graph, &status); in TEST_F()
337 MutableGraphView graph_view(&graph, &status); in TEST_F()
358 MutableGraphView graph_view(&graph, &status); in TEST_F()
394 MutableGraphView graph_view(&graph, &status); in TEST_F()
415 MutableGraphView graph_view(&graph, &status); in TEST_F()
|
| D | pattern_utils.h | 181 SubGraphMatcher(MutableGraphView* graph_view) : graph_view_(graph_view){}; in SubGraphMatcher()
190 MutableGraphView* graph_view_;
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view_test.cc | 36 void CompareNodeFanins(const MutableGraphView& graph, NodeDef* node, in CompareNodeFanins()
48 MutableGraphView::InputPort input_port(node, port); in CompareNodeFanins()
49 MutableGraphView::OutputPort output_port = in CompareNodeFanins()
56 void CompareNodeFanouts(const MutableGraphView& graph, NodeDef* node, in CompareNodeFanouts()
63 MutableGraphView::InputPort input_port(graph.GetNode(tensor_id.node()), in CompareNodeFanouts()
69 void CheckNode(const MutableGraphView& graph, absl::string_view node_name, in CheckNode()
88 void CheckGraph(const MutableGraphView& mutable_graph) { in CheckGraph()
95 [](const absl::flat_hash_set<MutableGraphView::Edge>& mutable_edges, in CheckGraph()
152 MutableGraphView graph(&graph_def); in TEST()
174 MutableGraphView graph(&graph_def); in TEST()
[all …]
|
| D | mutable_graph_view.cc | 55 bool IsOutputPortControlling(const MutableGraphView::OutputPort& port) { in IsOutputPortControlling()
61 bool IsIdentityConsumingSwitch(const MutableGraphView& graph, in IsIdentityConsumingSwitch()
80 bool CanDedupControlWithRegularInput(const MutableGraphView& graph, in CanDedupControlWithRegularInput()
89 bool CanDedupControlWithRegularInput(const MutableGraphView& graph, in CanDedupControlWithRegularInput()
98 bool HasRegularFaninNode(const MutableGraphView& graph, const NodeDef& node, in HasRegularFaninNode()
111 absl::flat_hash_map<MutableGraphView::OutputPort,
112 absl::flat_hash_set<MutableGraphView::InputPort>>;
114 void SwapControlledFanoutInputs(const MutableGraphView& graph, in SwapControlledFanoutInputs()
134 MutableGraphView::OutputPort port; in SwapRegularFanoutInputs()
151 void SwapFanoutInputs(const MutableGraphView& graph, FanoutsMap* fanouts, in SwapFanoutInputs()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 72 MutableGraphView graph(optimized_graph); in Optimize()
79 for (MutableGraphView::InputPort fanout : in Optimize()
80 graph.GetFanout(MutableGraphView::OutputPort(&node, 0))) { in Optimize()
90 const MutableGraphView::OutputPort reduce_indices = in Optimize()
91 graph.GetRegularFanin(MutableGraphView::InputPort(fanout.node, 1)); in Optimize()
142 MutableGraphView graph(optimized_graph); in Optimize()
148 const MutableGraphView::OutputPort input1 = in Optimize()
149 graph.GetRegularFanin(MutableGraphView::InputPort(&node, 0)); in Optimize()
150 const MutableGraphView::OutputPort input2 = in Optimize()
151 graph.GetRegularFanin(MutableGraphView::InputPort(&node, 1)); in Optimize()
|
| D | memory_optimizer.cc | 518 MutableGraphView view(&item->graph); in SchedulingPass()
885 static bool IsSwappable(const MutableGraphView& graph, in IsSwappable()
886 MutableGraphView::OutputPort output) { in IsSwappable()
912 MutableGraphView::InputPort input; in IsSwappable()
915 MutableGraphView::OutputPort fanin = graph.GetRegularFanin(input); in IsSwappable()
924 const NodeDef* node, int input_id, const MutableGraphView& view, in FindSwapOutTrigger()
928 MutableGraphView::InputPort swap; in FindSwapOutTrigger()
931 MutableGraphView::OutputPort generator = view.GetRegularFanin(swap); in FindSwapOutTrigger()
936 const absl::flat_hash_set<MutableGraphView::InputPort>& fanout = in FindSwapOutTrigger()
955 static bool IsSwappable(MutableGraphView::InputPort input) { in IsSwappable()
[all …]
|
| D | loop_optimizer.cc | 597 Status CheckForDeadFanout(const MutableGraphView& view, in CheckForDeadFanout()
771 MutableGraphView view(optimized_graph); in RemoveDeadBranches()
794 SetVector<MutableGraphView::InputPort, absl::Hash<MutableGraphView::Port>> in RemoveDeadBranches()
796 for (const MutableGraphView::InputPort& port : view.GetFanout(dead)) { in RemoveDeadBranches()
810 MutableGraphView::InputPort dead = zombie_inputs.PopBack(); in RemoveDeadBranches()
829 MutableGraphView::OutputPort value_index(dead.node, 1); in RemoveDeadBranches()
830 const absl::flat_hash_set<MutableGraphView::InputPort>& index_fanout = in RemoveDeadBranches()
857 for (const MutableGraphView::InputPort& port : in RemoveDeadBranches()
868 for (const MutableGraphView::InputPort& dead_fanout : in RemoveDeadBranches()
|