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()
Darithmetic_optimizer.cc781 return is_div ? OptimizedNodeName(scope_and_name, "Div") in OuterNodeName()
782 : OptimizedNodeName(scope_and_name, "Mul"); in OuterNodeName()
788 return OptimizedNodeName(scope_and_name, "AddV2"); in InnerAddNodeName()
1169 const string optimized_node_name = OptimizedNodeName(transpose); in TrySimplify()
1741 OptimizedNodeName(root_scope_and_name, cur_tail->name()), cur_tail); in HoistChainForSplit()
1755 OptimizedNodeName(root_scope_and_name, cur_tail->name()), cur_tail); in HoistChainForSplit()
2064 const string optimized_producer_name = OptimizedNodeName( in TrySimplify()
2066 const string optimized_consumer_name = OptimizedNodeName( in TrySimplify()
2195 OptimizedNodeName(ParseNodeScopeAndName(weights->name()), in TrySimplify()
2289 const string optimized_node_name = OptimizedNodeName(matmul); in TrySimplify()
[all …]
Dconstant_folding.cc362 return node_map_->NodeExists(OptimizedNodeName(node, suffix)); in OptimizedNodeExists()
365 string ConstantFolding::OptimizedNodeName(const NodeDef& node, in OptimizedNodeName() function in tensorflow::grappler::ConstantFolding
524 const string const_name = OptimizedNodeName( in MaterializeShapes()
694 OptimizedNodeName(node, strings::StrCat("-bcastargs-", j)); in MaterializeBroadcastGradientArgs()
799 string const_name = OptimizedNodeName(*node, "-reduction_indices"); in MaterializeReductionIndices()
1368 string node_name = OptimizedNodeName(node, "-folded"); in EvaluateOneFoldable()
1424 string const_out_name = OptimizedNodeName(*node, "_const"); in FoldMergeNode()
1425 string const_index_name = OptimizedNodeName(*node, "_index"); in FoldMergeNode()
1824 const string const_name = OptimizedNodeName( in ReplaceOperationWithBroadcastTo()
2423 const string axis_node_name = OptimizedNodeName(*node, "_const_axis"); in SimplifyPack()
[all …]
Dconstant_folding.h69 string OptimizedNodeName(const NodeDef& node, StringPiece suffix) const;