Home
last modified time | relevance | path

Searched refs:AddOrUpdateNodeAttr (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view.h328 void AddOrUpdateNodeAttr(MutableNodeView* node, absl::string_view attr_name,
332 void AddOrUpdateNodeAttr(const MutationNewNode& node,
Dgraph_view_test.cc1629 mutation->AddOrUpdateNodeAttr(new_node, "attr_3", attr_3); in TEST_F()
1632 mutation->AddOrUpdateNodeAttr(new_node, "attr_1", attr_1); in TEST_F()
1636 mutation->AddOrUpdateNodeAttr(new_node, "T", attr_4); in TEST_F()
1669 mutation->AddOrUpdateNodeAttr(d_node, "attr_d_1", attr_d_1); in TEST_F()
1672 mutation->AddOrUpdateNodeAttr(d_node, "attr_e_3", attr_e_3); in TEST_F()
1676 mutation->AddOrUpdateNodeAttr(d_node, "T", attr_e_4); in TEST_F()
Dgraph_view.cc393 void Mutation::AddOrUpdateNodeAttr(MutableNodeView* node, in AddOrUpdateNodeAttr() function in tensorflow::grappler::utils::Mutation
401 void Mutation::AddOrUpdateNodeAttr(const MutationNewNode& node, in AddOrUpdateNodeAttr() function in tensorflow::grappler::utils::Mutation
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.cc412 context->graph_view->GetMutationBuilder()->AddOrUpdateNodeAttr( in UpdateFanoutEdgesWithOp()
698 mutation->AddOrUpdateNodeAttr(node, kAttrDataFormat, data_format_attr); in UpdateNode()
708 mutation->AddOrUpdateNodeAttr(node, attr_name, attr_copy); in UpdateNode()
726 mutation->AddOrUpdateNodeAttr(node, kAttrExplicitPaddings, in UpdateNode()
1809 context->graph_view->GetMutationBuilder()->AddOrUpdateNodeAttr( in UpdateSqueezeDims()
1867 context->graph_view->GetMutationBuilder()->AddOrUpdateNodeAttr(node, mask, in PermuteMask()
Dgeneric_layout_optimizer.cc363 mutation->AddOrUpdateNodeAttr(paddings, "value", permuted_paddings_tensor); in EraseCancellableNodesAroundPad()
Dgeneric_layout_optimizer_transposer_test.cc644 context.graph_view->GetMutationBuilder()->AddOrUpdateNodeAttr( in TEST_F()