| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | shuffle_and_repeat_fusion.cc | 49 NodeDef* fused_node) { in FuseShuffleV1AndRepeat() argument
50 fused_node->set_op(kShuffleAndRepeatDataset); in FuseShuffleV1AndRepeat()
52 fused_node); in FuseShuffleV1AndRepeat()
55 fused_node->add_input(shuffle_node.input(0)); in FuseShuffleV1AndRepeat()
58 fused_node->add_input(shuffle_node.input(1)); in FuseShuffleV1AndRepeat()
61 fused_node->add_input(shuffle_node.input(2)); in FuseShuffleV1AndRepeat()
64 fused_node->add_input(shuffle_node.input(3)); in FuseShuffleV1AndRepeat()
67 fused_node->add_input(repeat_node.input(1)); in FuseShuffleV1AndRepeat()
72 graph_utils::CopyAttribute(key, shuffle_node, fused_node); in FuseShuffleV1AndRepeat()
81 NodeDef* fused_node) { in FuseShuffleV2AndRepeat() argument
[all …]
|
| D | map_fusion.cc | 42 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()
|
| D | filter_fusion.cc | 40 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()
|
| D | filter_with_random_uniform_fusion.cc | 38 NodeDef fused_node; in MakeFusedNode() local
40 &fused_node); in MakeFusedNode()
41 fused_node.set_op(kFusedOpName); in MakeFusedNode()
45 fused_node.add_input(filter_node.input(i)); in MakeFusedNode()
50 graph_utils::CopyAttribute(key, filter_node, &fused_node); in MakeFusedNode()
56 graph_utils::CopyAttribute(key, filter_node, &fused_node); in MakeFusedNode()
61 fused_node.add_input(tmp_rate->name()); in MakeFusedNode()
63 fused_node.add_input(tmp_seed->name()); in MakeFusedNode()
65 fused_node.add_input(tmp_seed2->name()); in MakeFusedNode()
67 return fused_node; in MakeFusedNode()
|
| D | map_and_filter_fusion.cc | 48 NodeDef fused_node; in MakeFusedNode() local
49 graph_utils::SetUniqueGraphNodeName("fused_map", graph->graph(), &fused_node); in MakeFusedNode()
50 fused_node.set_op(map_node.op()); in MakeFusedNode()
54 fused_node.add_input(map_node.input(i)); in MakeFusedNode()
59 (*fused_node.mutable_attr())["f"] = std::move(attr); in MakeFusedNode()
63 graph_utils::CopyAttribute(key, map_node, &fused_node); in MakeFusedNode()
70 graph_utils::CopyAttribute(key, map_node, &fused_node); in MakeFusedNode()
75 (*fused_node.mutable_attr())["output_types"] in MakeFusedNode()
79 (*fused_node.mutable_attr())["output_shapes"] in MakeFusedNode()
84 return fused_node; in MakeFusedNode()
|
| D | fusion_utils_test.cc | 74 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/core/grappler/optimizers/ |
| D | remapper.cc | 1446 const NodeDef& fused_node = graph->node(matched.fused_batch_norm); in IsCpuCompatibleDataType() local
1448 << SummarizeNodeDef(fused_node); in IsCpuCompatibleDataType()
1450 const string& x = fused_node.input(0); in IsCpuCompatibleDataType()
1451 string scale = fused_node.input(1); in IsCpuCompatibleDataType()
1452 string offset = fused_node.input(2); in IsCpuCompatibleDataType()
1453 string mean = fused_node.input(3); in IsCpuCompatibleDataType()
1454 string variance = fused_node.input(4); in IsCpuCompatibleDataType()
1459 string x_format = fused_node.attr().at(kDataFormat).s(); in IsCpuCompatibleDataType()
1464 AddPrefixToNodeName(x_format + "Shape", fused_node.name()); in IsCpuCompatibleDataType()
1467 new_shape.set_device(fused_node.device()); in IsCpuCompatibleDataType()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | pattern_utils_test.cc | 439 NodeDef fused_node; in TEST_F() local
440 fused_node.set_name("gelu"); in TEST_F()
441 fused_node.set_op("_FusedMatMul"); in TEST_F()
442 fused_node.add_input(graph_view.GetNode("matmul")->node()->input(0)); in TEST_F()
443 fused_node.add_input(graph_view.GetNode("matmul")->node()->input(1)); in TEST_F()
444 fused_node.add_input(graph_view.GetNode("bias_add")->node()->input(1)); in TEST_F()
445 mutation->AddNode(std::move(fused_node), &status); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | multi_output_fusion.cc | 209 FusionCandidate& fused_node = candidates_[get_candidate_id(fused)]; in GetNewFusibles() local
229 for (const auto& it : fused_node.fusibles) { in GetNewFusibles()
241 fused_node.fusibles.clear(); in GetNewFusibles()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 967 Node* fused_node; in FuseCluster() local
970 border_inputs, border_outputs, require_shape_type, &graph, &fused_node)); in FuseCluster()
989 graph.AddEdge(fused_node, j, dst_node, dst_input); in FuseCluster()
|