Searched refs:fusion_utils (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils_test.cc | 30 namespace fusion_utils { namespace
62 *parent_function, *function, "fused_maps", fusion_utils::ComposeSignature, in TEST()
63 fusion_utils::ComposeInput, fusion_utils::ComposeOutput, in TEST()
64 fusion_utils::MergeNodes, graph.mutable_library()); in TEST()
101 fusion_utils::CombineSignature, fusion_utils::ComposeInput, in TEST()
102 fusion_utils::CombineOutput, fusion_utils::MergeNodes, in TEST()
140 *parent_function, *function, "fused_maps", fusion_utils::CombineSignature, in TEST()
141 fusion_utils::ComposeInput, fusion_utils::CombineOutput, in TEST()
142 fusion_utils::MergeNodes, graph.mutable_library()); in TEST()
174 fusion_utils::CombineOutput, fusion_utils::MergeNodes, in TEST()
|
| D | filter_fusion.cc | 88 if (!fusion_utils::HasSameSignature(first_func->signature(), in OptimizeAndCollectStats()
94 return fusion_utils::FuseFunctions( in OptimizeAndCollectStats()
96 fusion_utils::SameSignature, fusion_utils::SameInput, in OptimizeAndCollectStats()
97 fusion_utils::LazyConjunctionOutput, fusion_utils::LazyConjunctionNodes, in OptimizeAndCollectStats()
|
| D | map_fusion.cc | 110 if (!fusion_utils::CanCompose(parent_func->signature(), in OptimizeAndCollectStats()
117 return fusion_utils::FuseFunctions( in OptimizeAndCollectStats()
118 *parent_func, *func, "fused_map", fusion_utils::ComposeSignature, in OptimizeAndCollectStats()
119 fusion_utils::ComposeInput, fusion_utils::ComposeOutput, in OptimizeAndCollectStats()
120 fusion_utils::MergeNodes, output->mutable_library()); in OptimizeAndCollectStats()
|
| D | map_and_filter_fusion.cc | 130 if (!fusion_utils::CanCompose(map_func->signature(), in OptimizeAndCollectStats()
137 return fusion_utils::FuseFunctions( in OptimizeAndCollectStats()
139 fusion_utils::CombineSignature, fusion_utils::ComposeInput, in OptimizeAndCollectStats()
140 fusion_utils::CombineOutput, fusion_utils::MergeNodes, in OptimizeAndCollectStats()
|
| D | BUILD | 61 ":fusion_utils",
93 name = "fusion_utils",
94 srcs = ["fusion_utils.cc"],
96 "fusion_utils.h",
120 ":fusion_utils",
388 ":fusion_utils",
427 ":fusion_utils",
|
| D | fusion_utils.h | 29 namespace fusion_utils {
|
| D | fusion_utils.cc | 36 namespace fusion_utils { namespace
|