Home
last modified time | relevance | path

Searched refs:fused_function (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils_test.cc61 auto *fused_function = FuseFunctions( in TEST() local
66 EXPECT_EQ(fused_function->signature().name(), "fused_maps"); in TEST()
67 EXPECT_EQ(fused_function->signature().input_arg_size(), 1); in TEST()
68 EXPECT_EQ(fused_function->signature().output_arg_size(), 1); in TEST()
69 EXPECT_EQ(fused_function->ret_size(), 1); in TEST()
70 std::cerr << fused_function->DebugString(); in TEST()
71 CheckUniqueNames(*fused_function); in TEST()
74 for (const auto &fused_node : fused_function->node_def()) { in TEST()
86 auto output_value = fused_function->ret().at( in TEST()
87 fused_function->signature().output_arg(0).name()); in TEST()
[all …]
Dfusion_utils.cc298 const FunctionDef& second_function, FunctionDef* fused_function, in MergeNodes() argument
301 fused_function->mutable_node_def()->CopyFrom(first_function.node_def()); in MergeNodes()
303 fused_function->mutable_node_def()->MergeFrom(second_function.node_def()); in MergeNodes()
378 FunctionDef* fused_function, in LazyConjunctionNodes() argument
380 fused_function->mutable_node_def()->CopyFrom(first_function.node_def()); in LazyConjunctionNodes()
408 auto* if_node = fused_function->add_node_def(); in LazyConjunctionNodes()
411 function_utils::SetUniqueFunctionNodeName("cond", fused_function, if_node); in LazyConjunctionNodes()
413 GetMutableOutputNode(fused_function, 0) = if_node->name() + ":output:0"; in LazyConjunctionNodes()
441 FunctionDef* fused_function = library->add_function(); in FuseFunctions() local
444 fused_function->mutable_signature()); in FuseFunctions()
[all …]
Dmap_fusion.cc40 const FunctionDef& fused_function, in MakeFusedNode() argument
48 *attr.mutable_func()->mutable_name() = fused_function.signature().name(); in MakeFusedNode()
131 const auto* fused_function = get_fused_function(parent_map_node, map_node); in OptimizeAndCollectStats() local
132 if (fused_function == nullptr) continue; in OptimizeAndCollectStats()
134 MakeFusedNode(*parent_map_node, *map_node, *fused_function, &graph)); in OptimizeAndCollectStats()
142 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
Dmap_and_filter_fusion.cc38 const FunctionDef& fused_function, in MakeFusedNode() argument
50 attr.mutable_func()->set_name(fused_function.signature().name()); in MakeFusedNode()
152 const auto* fused_function = make_fused_function(map_node, filter_node); in OptimizeAndCollectStats() local
153 if (fused_function == nullptr) continue; in OptimizeAndCollectStats()
156 graph.AddNode(MakeFusedNode(*map_node, *fused_function, &graph)); in OptimizeAndCollectStats()
163 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
Dfusion_utils.h60 FunctionDef* fused_function, FunctionDefLibrary* library)>;
63 const FunctionDef& second_function, FunctionDef* fused_function,
116 FunctionDef* fused_function,
Dfilter_fusion.cc38 const FunctionDef& fused_function, in MakeFusedFilterNode() argument
48 *attr.mutable_func()->mutable_name() = fused_function.signature().name(); in MakeFusedFilterNode()