Home
last modified time | relevance | path

Searched refs:OptimizedNodeName (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage.h161 const string OptimizedNodeName(const NodeScopeAndName& node) const { in OptimizedNodeName() function
164 const string OptimizedNodeName(const NodeScopeAndName& root, in OptimizedNodeName() function
168 const string OptimizedNodeName(const NodeScopeAndName& node, in OptimizedNodeName() function
175 const string node_name = OptimizedNodeName(node); in UniqueOptimizedNodeName()
180 const string node_name = OptimizedNodeName(node, rewrite_rule); in UniqueOptimizedNodeName()
Dgraph_optimizer_stage_test.cc63 TEST_F(GraphOptimizerStageTest, OptimizedNodeName) { in TEST_F() argument
75 EXPECT_EQ(stage.OptimizedNodeName(node), "a/b/c/my_opt/my_stg_Add"); in TEST_F()
76 EXPECT_EQ(stage.OptimizedNodeName(node, std::vector<string>({"Mul", "Sqrt"})), in TEST_F()
81 EXPECT_EQ(stage.OptimizedNodeName(node, rewrite), in TEST_F()
Dconstant_folding.cc327 return node_map_->NodeExists(OptimizedNodeName(node, suffix)); in OptimizedNodeExists()
330 string ConstantFolding::OptimizedNodeName(const NodeDef& node, in OptimizedNodeName() function in tensorflow::grappler::ConstantFolding
471 const string const_name = OptimizedNodeName( in MaterializeShapes()
640 OptimizedNodeName(node, strings::StrCat("-bcastargs-", j)); in MaterializeBroadcastGradientArgs()
744 string const_name = OptimizedNodeName(*node, "-reduction_indices"); in MaterializeReductionIndices()
1206 string node_name = OptimizedNodeName(node, "-folded"); in EvaluateOneFoldable()
1262 string const_out_name = OptimizedNodeName(*node, "_const"); in FoldMergeNode()
1263 string const_index_name = OptimizedNodeName(*node, "_index"); in FoldMergeNode()
2299 axis_node->set_name(OptimizedNodeName(*node, "_const_axis")); in SimplifyPack()
2356 new_node->set_name(OptimizedNodeName(*input, "_enter")); in MoveConstantsPastEnter()
[all …]
Darithmetic_optimizer.cc755 return is_div ? OptimizedNodeName(scope_and_name, "Div") in OuterNodeName()
756 : OptimizedNodeName(scope_and_name, "Mul"); in OuterNodeName()
762 return OptimizedNodeName(scope_and_name, "Add"); in InnerAddNodeName()
1662 OptimizedNodeName(root_scope_and_name, cur_tail->name()), cur_tail); in HoistChainForSplit()
1676 OptimizedNodeName(root_scope_and_name, cur_tail->name()), cur_tail); in HoistChainForSplit()
1961 const string optimized_producer_name = OptimizedNodeName( in TrySimplify()
1963 const string optimized_consumer_name = OptimizedNodeName( in TrySimplify()
2092 OptimizedNodeName(ParseNodeScopeAndName(weights->name()), in TrySimplify()
2178 const string optimized_node_name = OptimizedNodeName(matmul); in TrySimplify()
2284 const string optimized_node_name = OptimizedNodeName(matmul); in TrySimplify()
[all …]
Dconstant_folding.h59 string OptimizedNodeName(const NodeDef& node, StringPiece suffix) const;