Home
last modified time | relevance | path

Searched refs:fused_node (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc633 const NodeDef& fused_node = *matched.fused_batch_norm; in AddBatchNormNodes() local
635 << SummarizeNodeDef(fused_node); in AddBatchNormNodes()
637 const string& x = fused_node.input(0); in AddBatchNormNodes()
638 string scale = fused_node.input(1); in AddBatchNormNodes()
639 string offset = fused_node.input(2); in AddBatchNormNodes()
640 string mean = fused_node.input(3); in AddBatchNormNodes()
641 string variance = fused_node.input(4); in AddBatchNormNodes()
643 if (fused_node.attr().at(kDataFormat).s() == "NCHW") { in AddBatchNormNodes()
646 new_shape->set_name(AddPrefixToNodeName("NCHWShape", fused_node.name())); in AddBatchNormNodes()
648 new_shape->set_device(fused_node.device()); in AddBatchNormNodes()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_fusion.cc42 NodeDef fused_node; in MakeFusedNode() local
43 graph_utils::SetUniqueGraphNodeName("fused_map", graph->graph(), &fused_node); in MakeFusedNode()
44 fused_node.set_op("MapDataset"); in MakeFusedNode()
45 fused_node.add_input(parent_map_node.input(0)); in MakeFusedNode()
49 (*fused_node.mutable_attr())["f"] = std::move(attr); in MakeFusedNode()
51 graph_utils::CopyAttribute("Targuments", parent_map_node, &fused_node); in MakeFusedNode()
53 graph_utils::CopyAttribute(key, map_node, &fused_node); in MakeFusedNode()
66 (*fused_node.mutable_attr())["use_inter_op_parallelism"].set_b( in MakeFusedNode()
73 (*fused_node.mutable_attr())["preserve_cardinality"].set_b( in MakeFusedNode()
76 return fused_node; in MakeFusedNode()
Dmap_and_filter_fusion.cc40 NodeDef fused_node; in MakeFusedNode() local
41 graph_utils::SetUniqueGraphNodeName("fused_map", graph->graph(), &fused_node); in MakeFusedNode()
42 fused_node.set_op(map_node.op()); in MakeFusedNode()
46 fused_node.add_input(map_node.input(i)); in MakeFusedNode()
51 (*fused_node.mutable_attr())["f"] = std::move(attr); in MakeFusedNode()
55 graph_utils::CopyAttribute(key, map_node, &fused_node); in MakeFusedNode()
62 graph_utils::CopyAttribute(key, map_node, &fused_node); in MakeFusedNode()
67 (*fused_node.mutable_attr())["output_types"] in MakeFusedNode()
71 (*fused_node.mutable_attr())["output_shapes"] in MakeFusedNode()
76 return fused_node; in MakeFusedNode()
Dfilter_fusion.cc40 NodeDef fused_node; in MakeFusedFilterNode() local
42 &fused_node); in MakeFusedFilterNode()
44 fused_node.set_op("FilterDataset"); in MakeFusedFilterNode()
45 fused_node.add_input(first_filter_node.input(0)); in MakeFusedFilterNode()
49 (*fused_node.mutable_attr())["predicate"] = std::move(attr); in MakeFusedFilterNode()
51 graph_utils::CopyAttribute("Targuments", first_filter_node, &fused_node); in MakeFusedFilterNode()
54 graph_utils::CopyAttribute(key, second_filter_node, &fused_node); in MakeFusedFilterNode()
56 return fused_node; in MakeFusedFilterNode()
Dfusion_utils_test.cc74 for (const auto &fused_node : fused_function->node_def()) { in TEST() local
75 if (fused_node.op() == "Mul") { in TEST()
76 if (fused_node.name() == "y") in TEST()
77 parent_mul = &fused_node; in TEST()
79 output_mul = &fused_node; in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/
Dmulti_output_fusion.cc167 FusionCandidate& fused_node = candidates_[get_candidate_id(fused)]; in Update() local
196 for (const auto& it : fused_node.fusibles) { in Update()
208 fused_node.fusibles.clear(); in Update()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc965 Node* fused_node; in FuseCluster() local
968 border_inputs, border_outputs, require_shape_type, &graph, &fused_node)); in FuseCluster()
987 graph.AddEdge(fused_node, j, dst_node, dst_input); in FuseCluster()