Home
last modified time | relevance | path

Searched refs:AddPrefixToNodeName (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_parallel.cc214 string new_name = AddPrefixToNodeName(new_node->input(i), prefix); in AddSharedNodes()
227 new_node->set_name(AddPrefixToNodeName(new_node->name(), prefix)); in AddOneReplica()
233 string new_name = AddPrefixToNodeName(new_node->input(i), prefix); in AddOneReplica()
250 string fetch = AddPrefixToNodeName(item_->fetch[i], prefix); in BuildGraph()
Dremapper.cc1464 AddPrefixToNodeName(x_format + "Shape", fused_node.name()); in IsCpuCompatibleDataType()
1493 AddPrefixToNodeName(x_format + "ShapedScale", fused_node.name())); in IsCpuCompatibleDataType()
1506 AddPrefixToNodeName(x_format + "ShapedOffset", fused_node.name())); in IsCpuCompatibleDataType()
1519 AddPrefixToNodeName(x_format + "ShapedMean", fused_node.name())); in IsCpuCompatibleDataType()
1532 AddPrefixToNodeName(x_format + "ShapedVariance", fused_node.name())); in IsCpuCompatibleDataType()
1553 AddPrefixToNodeName("Const", fused_node.name()); in IsCpuCompatibleDataType()
1562 AddPrefixToNodeName("VarPlusEpsilon", fused_node.name()); in IsCpuCompatibleDataType()
1573 const string inv_name = AddPrefixToNodeName("Inv", fused_node.name()); in IsCpuCompatibleDataType()
1583 const string scaled_name = AddPrefixToNodeName("Scaled", fused_node.name()); in IsCpuCompatibleDataType()
1594 const string a_name = AddPrefixToNodeName("Mul", fused_node.name()); in IsCpuCompatibleDataType()
[all …]
Dmemory_optimizer.cc356 new_trigger_node->set_name(AddPrefixToNodeName( in AddRecomputeControlDependencyNodes()
388 return AddPrefixToNodeName(original_node_name, kRecomputedNodePrefix); in RecomputedOrOriginalNodeName()
418 AddPrefixToNodeName(original_node->name(), kRecomputedNodePrefix)); in RecomputeSubgraph()
Dloop_optimizer.cc132 AddPrefixToNodeName(node->name(), kLoopOptimizer); in HandleConst()
231 new_enter->set_name(AddPrefixToNodeName( in HandleInvariantNode()
Ddependency_optimizer.cc683 group_name = AddPrefixToNodeName( in GroupCrossDeviceControlEdges()
Dconstant_folding_test.cc1655 EXPECT_NE(AddPrefixToNodeName("s-matshapes-0", kConstantFoldingConst), in TEST_F()
1657 EXPECT_NE(AddPrefixToNodeName("s-matshapes-1", kConstantFoldingConst), in TEST_F()
1669 EXPECT_EQ(AddPrefixToNodeName("s-matshapes-2", kConstantFoldingConst), in TEST_F()
1680 AddPrefixToNodeName("s-matshapes-2", kConstantFoldingConst)) { in TEST_F()
1728 EXPECT_NE(AddPrefixToNodeName("s-matshapes-0", kConstantFoldingConst), in TEST_F()
1740 EXPECT_EQ(AddPrefixToNodeName("s-matshapes-1", kConstantFoldingConst), in TEST_F()
Darithmetic_optimizer_test.cc57 return AddPrefixToNodeName(name, kHoistFactorOptimizerMul, ""); in HoistMulName()
62 return AddPrefixToNodeName(name, kHoistFactorOptimizerDiv, ""); in HoistDivName()
67 return AddPrefixToNodeName(name, kHoistFactorOptimizerAdd, ""); in HoistAddName()
72 return AddPrefixToNodeName(name, kSimplifyAggregationConst, ""); in AggregationConstName()
77 return AddPrefixToNodeName(name, kSimplifyAggregationMul, ""); in AggregationMulName()
Dconstant_folding.cc237 ctrl_dep_name = AddPrefixToNodeName(ctrl_dep_name, kConstantFoldingCtrl); in AddControlDependency()
367 return AddPrefixToNodeName(strings::StrCat(node.name(), suffix), in OptimizedNodeName()
3523 string mul_new_name = AddPrefixToNodeName("merged_input", conv_node->name()); in MulConvPushDown()
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc113 string AddPrefixToNodeName(const string& name, const string& prefix, in AddPrefixToNodeName() function
123 string AddPrefixToNodeName(const string& name, const string& prefix) { in AddPrefixToNodeName() function
124 return AddPrefixToNodeName(name, prefix, "/"); in AddPrefixToNodeName()
Dutils.h298 string AddPrefixToNodeName(const string& name, const string& prefix,
302 string AddPrefixToNodeName(const string& name, const string& prefix);
Dutils_test.cc178 EXPECT_EQ(AddPrefixToNodeName("abc", "OPTIMIZED"), "OPTIMIZED/abc"); in TEST_F()
179 EXPECT_EQ(AddPrefixToNodeName("^abc", "OPTIMIZED"), "^OPTIMIZED/abc"); in TEST_F()
180 EXPECT_EQ(AddPrefixToNodeName("", "OPTIMIZED"), "OPTIMIZED/"); in TEST_F()
Dmutable_graph_view.cc332 return AddPrefixToNodeName( in GeneratedNameForIdentityConsumingSwitch()