| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view_internal_test.cc | 188 RemoveRegularFanin(&diff, 0); in TEST()
191 RemoveRegularFanin(&diff, 2); in TEST()
203 RemoveRegularFanin(&diff, 0); in TEST()
212 TEST(MutableNodeViewDiffTest, RemoveRegularFanin) { in TEST() argument
228 RemoveRegularFanin(&diff, -1); in TEST()
231 RemoveRegularFanin(&diff, 3); in TEST()
240 RemoveRegularFanin(&diff, 4); in TEST()
253 RemoveRegularFanin(&diff, 3); in TEST()
257 RemoveRegularFanin(&diff, 4); in TEST()
270 RemoveRegularFanin(&diff, 4); in TEST()
[all …]
|
| D | graph_view.h | 305 void RemoveRegularFanin(MutableNodeView* node, int index);
308 void RemoveRegularFanin(const MutationNewNode& node, int index);
|
| D | graph_view_internal.h | 532 inline bool RemoveRegularFanin(NodeViewDiff<GraphViewT>* diff, int index) { in RemoveRegularFanin() function
841 inline void RemoveRegularFanin(NewNode<GraphViewT>* new_node, int index) { in RemoveRegularFanin() function
|
| D | graph_view_test.cc | 1318 mutation->RemoveRegularFanin(d_node, 1); in TEST_F()
1319 mutation->RemoveRegularFanin(d_node, 2); in TEST_F()
1354 mutation->RemoveRegularFanin(d_node, 2); in TEST_F()
1623 mutation->RemoveRegularFanin(new_node, 1); in TEST_F()
1805 mutation->RemoveRegularFanin(b_node, 1); in TEST_F()
1841 mutation->RemoveRegularFanin(new_node_1_node, 1); in TEST_F()
1874 mutation->RemoveRegularFanin(new_node_2_node, 1); in TEST_F()
|
| D | graph_view.cc | 346 void Mutation::RemoveRegularFanin(MutableNodeView* node, int index) { in RemoveRegularFanin() function in tensorflow::grappler::utils::Mutation
348 return internal::RemoveRegularFanin(diff, index); in RemoveRegularFanin()
352 void Mutation::RemoveRegularFanin(const MutationNewNode& node, int index) { in RemoveRegularFanin() function in tensorflow::grappler::utils::Mutation
355 internal::RemoveRegularFanin(&new_nodes_[node.index_], index); in RemoveRegularFanin()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view.h | 185 Status RemoveRegularFanin(absl::string_view node_name, const TensorId& fanin);
|
| D | mutable_graph_view_test.cc | 1212 Status s = graph.RemoveRegularFanin(node_name, fanin_to_remove); in TestRemoveRegularFanin()
1229 TEST(MutableGraphViewTest, RemoveRegularFanin) { in TEST() argument
2343 TF_EXPECT_OK(graph.RemoveRegularFanin("c", {"a", 2})); in TEST()
2360 TF_EXPECT_OK(graph.RemoveRegularFanin("b", {"a", 1})); in TEST()
|
| D | mutable_graph_view.cc | 1102 Status MutableGraphView::RemoveRegularFanin(absl::string_view node_name, in RemoveRegularFanin() function in tensorflow::grappler::MutableGraphView
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer.cc | 371 mutation->RemoveRegularFanin(transpose, 1); in EraseCancellableNodesAroundPad()
|