Searched refs:RemoveNodeAttr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer.cc | 370 mutation->RemoveNodeAttr(transpose, "Tperm"); in EraseCancellableNodesAroundPad()
391 mutation->RemoveNodeAttr(node, kAttrOutputShape); in EraseOutputShapeAttrs()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view.h | 337 void RemoveNodeAttr(MutableNodeView* node, absl::string_view attr_name);
340 void RemoveNodeAttr(const MutationNewNode& node, absl::string_view attr_name);
|
| D | graph_view_test.cc | 1633 mutation->RemoveNodeAttr(new_node, "attr_2"); in TEST_F()
1673 mutation->RemoveNodeAttr(d_node, "attr_d_2"); in TEST_F()
1952 mutation->RemoveNodeAttr(a_node, "T"); in TEST_F()
2426 mutation->RemoveNodeAttr(graph_view.GetNode(j), "_some_random_attr"); in BM_MutableGraphViewClearAttrs()
|
| D | graph_view.cc | 409 void Mutation::RemoveNodeAttr(MutableNodeView* node, in RemoveNodeAttr() function in tensorflow::grappler::utils::Mutation
416 void Mutation::RemoveNodeAttr(const MutationNewNode& node, in RemoveNodeAttr() function in tensorflow::grappler::utils::Mutation
|