| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 646 new_shape->set_name(AddPrefixToNodeName("NCHWShape", fused_node.name())); in AddBatchNormNodes()
661 AddPrefixToNodeName("NCHWShapedScale", fused_node.name())); in AddBatchNormNodes()
672 AddPrefixToNodeName("NCHWShapedOffset", fused_node.name())); in AddBatchNormNodes()
683 AddPrefixToNodeName("NCHWShapedMean", fused_node.name())); in AddBatchNormNodes()
694 AddPrefixToNodeName("NCHWShapedVariance", fused_node.name())); in AddBatchNormNodes()
713 AddPrefixToNodeName("Const", fused_node.name()), &value, in AddBatchNormNodes()
719 AddPrefixToNodeName("VarPlusEpsilon", fused_node.name())); in AddBatchNormNodes()
727 inv->set_name(AddPrefixToNodeName("Inv", fused_node.name())); in AddBatchNormNodes()
734 scaled->set_name(AddPrefixToNodeName("Scaled", fused_node.name())); in AddBatchNormNodes()
742 a->set_name(AddPrefixToNodeName("Mul", fused_node.name())); in AddBatchNormNodes()
[all …]
|
| D | auto_parallel.cc | 211 string new_name = AddPrefixToNodeName(new_node->input(i), prefix); in AddSharedNodes()
224 new_node->set_name(AddPrefixToNodeName(new_node->name(), prefix)); in AddOneReplica()
230 string new_name = AddPrefixToNodeName(new_node->input(i), prefix); in AddOneReplica()
247 string fetch = AddPrefixToNodeName(item_->fetch[i], prefix); in BuildGraph()
|
| D | function_optimizer.cc | 1080 input = AddPrefixToNodeName(input, /*prefix=*/func_node.name()); in InlineDirectFunctionCall()
1205 i, AddPrefixToNodeName(inlined_node.input(i), node.name())); in InlineSymbolicGradient()
1209 AddPrefixToNodeName(inlined_node.name(), node.name())); in InlineSymbolicGradient()
1216 AddPrefixToNodeName(inlined_node.name(), node.name())); in InlineSymbolicGradient()
1219 i, AddPrefixToNodeName(inlined_node.input(i), node.name())); in InlineSymbolicGradient()
1225 AddPrefixToNodeName("FunctionInputs", node.name())); in InlineSymbolicGradient()
1755 return AddPrefixToNodeName(name, /*prefix=*/func_node.name()); in InlineIndirectFunctionCall()
|
| D | memory_optimizer.cc | 336 new_trigger_node->set_name(AddPrefixToNodeName( in AddRecomputeControlDependencyNodes()
368 return AddPrefixToNodeName(original_node_name, kRecomputedNodePrefix); in RecomputedOrOriginalNodeName()
398 AddPrefixToNodeName(original_node->name(), kRecomputedNodePrefix)); in RecomputeSubgraph()
|
| D | loop_optimizer.cc | 133 AddPrefixToNodeName(node->name(), kLoopOptimizer); in HandleConst()
233 new_enter->set_name(AddPrefixToNodeName( in HandleInvariantNode()
|
| D | dependency_optimizer.cc | 625 group_name = AddPrefixToNodeName( in GroupCrossDeviceControlEdges()
|
| D | constant_folding_test.cc | 1341 EXPECT_NE(AddPrefixToNodeName("s-matshapes-0", kConstantFoldingConst), in TEST_F()
1343 EXPECT_NE(AddPrefixToNodeName("s-matshapes-1", kConstantFoldingConst), in TEST_F()
1355 EXPECT_EQ(AddPrefixToNodeName("s-matshapes-2", kConstantFoldingConst), in TEST_F()
1366 AddPrefixToNodeName("s-matshapes-2", kConstantFoldingConst)) { in TEST_F()
1414 EXPECT_NE(AddPrefixToNodeName("s-matshapes-0", kConstantFoldingConst), in TEST_F()
1426 EXPECT_EQ(AddPrefixToNodeName("s-matshapes-1", kConstantFoldingConst), in TEST_F()
|
| D | arithmetic_optimizer_test.cc | 51 return AddPrefixToNodeName(name, kHoistFactorOptimizerMul, ""); in HoistMulName()
56 return AddPrefixToNodeName(name, kHoistFactorOptimizerDiv, ""); in HoistDivName()
61 return AddPrefixToNodeName(name, kHoistFactorOptimizerAdd, ""); in HoistAddName()
66 return AddPrefixToNodeName(name, kSimplifyAggregationConst, ""); in AggregationConstName()
71 return AddPrefixToNodeName(name, kSimplifyAggregationMul, ""); in AggregationMulName()
|
| D | constant_folding.cc | 266 ctrl_dep_name = AddPrefixToNodeName(ctrl_dep_name, kConstantFoldingCtrl); in AddControlDependency()
332 return AddPrefixToNodeName(strings::StrCat(node.name(), suffix), in OptimizedNodeName()
2975 string mul_new_name = AddPrefixToNodeName("merged_input", conv_node->name()); in MulConvPushDown()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.h | 219 string AddPrefixToNodeName(const string& name, const string& prefix,
223 string AddPrefixToNodeName(const string& name, const string& prefix);
|
| D | utils.cc | 178 string AddPrefixToNodeName(const string& name, const string& prefix, in AddPrefixToNodeName() function
188 string AddPrefixToNodeName(const string& name, const string& prefix) { in AddPrefixToNodeName() function
189 return AddPrefixToNodeName(name, prefix, "/"); in AddPrefixToNodeName()
|
| D | utils_test.cc | 176 EXPECT_EQ(AddPrefixToNodeName("abc", "OPTIMIZED"), "OPTIMIZED/abc"); in TEST_F()
177 EXPECT_EQ(AddPrefixToNodeName("^abc", "OPTIMIZED"), "^OPTIMIZED/abc"); in TEST_F()
178 EXPECT_EQ(AddPrefixToNodeName("", "OPTIMIZED"), "OPTIMIZED/"); in TEST_F()
|
| D | mutable_graph_view.cc | 331 return AddPrefixToNodeName( in GeneratedNameForIdentityConsumingSwitch()
|