| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view_internal_test.cc | 127 TEST(MutableNodeViewDiffTest, AddOrUpdateRegularFanin) { in TEST() argument
143 AddOrUpdateRegularFanin(&diff, -1, {"a", 0}); in TEST()
148 AddOrUpdateRegularFanin(&diff, 0, {"a", 2}); in TEST()
153 AddOrUpdateRegularFanin(&diff, 0, {"a", 3}); in TEST()
158 AddOrUpdateRegularFanin(&diff, 4, {"b", 4}); in TEST()
163 AddOrUpdateRegularFanin(&diff, 3, {"c", 4}); in TEST()
168 AddOrUpdateRegularFanin(&diff, 5, {"c", 5}); in TEST()
195 AddOrUpdateRegularFanin(&diff, 1, {"c", 1}); in TEST()
199 AddOrUpdateRegularFanin(&diff, 0, {"c", 0}); in TEST()
207 AddOrUpdateRegularFanin(&diff, 2, {"c", 2}); in TEST()
[all …]
|
| D | graph_view_test.cc | 1020 mutation->AddOrUpdateRegularFanin(new_node, 1, {"valid", 2}); in TEST_F()
1230 mutation->AddOrUpdateRegularFanin(d_node, 1, {"d", 3}); in TEST_F()
1363 mutation->AddOrUpdateRegularFanin(d_node, 0, {"b", 1}); in TEST_F()
1552 mutation->AddOrUpdateRegularFanin(d_node, 3, {"c", 6}); in TEST_F()
1553 mutation->AddOrUpdateRegularFanin(d_node, 1, {"new_node_1", 5}); in TEST_F()
1622 mutation->AddOrUpdateRegularFanin(new_node, 0, {"b", 6}); in TEST_F()
1831 mutation->AddOrUpdateRegularFanin(d_node, 1, {"c", 5}); in TEST_F()
1840 mutation->AddOrUpdateRegularFanin(new_node_1_node, 0, {"c", 5}); in TEST_F()
1842 mutation->AddOrUpdateRegularFanin(new_node_1_node, 1, {"a", 6}); in TEST_F()
1860 mutation->AddOrUpdateRegularFanin(d_node, 3, {"new_node_2", 6}); in TEST_F()
[all …]
|
| D | graph_view.h | 297 void AddOrUpdateRegularFanin(MutableNodeView* node, int index,
301 void AddOrUpdateRegularFanin(const MutationNewNode& node, int index,
|
| D | graph_view_internal.h | 489 inline bool AddOrUpdateRegularFanin(NodeViewDiff<GraphViewT>* diff, int index, in AddOrUpdateRegularFanin() function
827 inline void AddOrUpdateRegularFanin(NewNode<GraphViewT>* new_node, int index, in AddOrUpdateRegularFanin() function
|
| D | graph_view.cc | 331 void Mutation::AddOrUpdateRegularFanin(MutableNodeView* node, int index, in AddOrUpdateRegularFanin() function in tensorflow::grappler::utils::Mutation
334 return internal::AddOrUpdateRegularFanin(diff, index, fanin); in AddOrUpdateRegularFanin()
338 void Mutation::AddOrUpdateRegularFanin(const MutationNewNode& node, int index, in AddOrUpdateRegularFanin() function in tensorflow::grappler::utils::Mutation
343 internal::AddOrUpdateRegularFanin(&new_nodes_[node.index_], index, fanin); in AddOrUpdateRegularFanin()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer.cc | 252 mutation->AddOrUpdateRegularFanin(regular_fanout.node_view(), in EraseCancellableNodes()
|
| D | generic_layout_optimizer_transposer.cc | 550 mutation->AddOrUpdateRegularFanin(added_node, 0, in UpdateEdge()
554 mutation->AddOrUpdateRegularFanin(dst_node, dst_port, {added_node_name, 0}); in UpdateEdge()
1315 mutation->AddOrUpdateRegularFanin(node, vector_index, in MaybeReshapeVectorFanin()
|