Searched refs:AddSimpleNode (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | loop_optimizer_test.cc | 54 void AddSimpleNode(const string& name, const string& op, in AddSimpleNode() function in tensorflow::grappler::LoopOptimizerTest
84 AddSimpleNode("In", "Identity", {}, &graph); in TEST_F()
87 AddSimpleNode("InvariantAdd", "Add", {"InvariantEnter", "InvariantEnter"}, in TEST_F()
89 AddSimpleNode("VariantAdd", "Add", {"InvariantAdd", "Identity"}, &graph); in TEST_F()
91 AddSimpleNode("Merge", "Merge", {"VariantEnter", "NextIteration"}, &graph); in TEST_F()
92 AddSimpleNode("Less/y", "Const", {"^Identity"}, &graph); in TEST_F()
93 AddSimpleNode("Less", "Less", {"VariantAdd", "Less/y"}, &graph); in TEST_F()
94 AddSimpleNode("LoopCond", "LoopCond", {"Less"}, &graph); in TEST_F()
95 AddSimpleNode("Switch", "Switch", {"Merge", "LoopCond"}, &graph); in TEST_F()
96 AddSimpleNode("Identity", "Identity", {"Switch:1"}, &graph); in TEST_F()
[all …]
|
| D | auto_mixed_precision_test.cc | 124 NodeDef* AddSimpleNode(const string& name, const string& op, in AddSimpleNode() function in tensorflow::grappler::__anond48c34ab0111::AutoMixedPrecisionTest
|