Home
last modified time | relevance | path

Searched refs:CopyAttribute (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_and_batch_fusion.cc83 graph_utils::CopyAttribute(key, map_node, &new_node); in MakeMapAndBatchNode()
86 graph_utils::CopyAttribute(key, batch_node, &new_node); in MakeMapAndBatchNode()
93 graph_utils::CopyAttribute(key, map_node, &new_node); in MakeMapAndBatchNode()
Dfilter_fusion.cc51 graph_utils::CopyAttribute("Targuments", first_filter_node, &fused_node); in MakeFusedFilterNode()
54 graph_utils::CopyAttribute(key, second_filter_node, &fused_node); in MakeFusedFilterNode()
Dmap_fusion.cc51 graph_utils::CopyAttribute("Targuments", parent_map_node, &fused_node); in MakeFusedNode()
53 graph_utils::CopyAttribute(key, map_node, &fused_node); in MakeFusedNode()
Dmap_vectorization.cc79 graph_utils::CopyAttribute(k, map_node, map_defun_node); in CreateMapDefunWrapper()
280 graph_utils::CopyAttribute("Targuments", old_map_node, &map_node); in AddNewMapNode()
285 graph_utils::CopyAttribute(key, old_batch_node, &map_node); in AddNewMapNode()
313 graph_utils::CopyAttribute(key, new_map_node, &prefetch_node); in AddNewPrefetchNode()
434 graph_utils::CopyAttribute(key, in AddNewChooseFastestNode()
Dmap_and_filter_fusion.cc55 graph_utils::CopyAttribute(key, map_node, &fused_node); in MakeFusedNode()
62 graph_utils::CopyAttribute(key, map_node, &fused_node); in MakeFusedNode()
Dauto_shard.cc125 graph_utils::CopyAttribute("output_shapes", *add_after, &new_node); in AddShardNode()
131 graph_utils::CopyAttribute("output_types", *add_after, &new_node); in AddShardNode()
Dhoist_random_uniform.cc57 graph_utils::CopyAttribute("Targuments", map_node, &stateless_map); in MakeStatelessMap()
59 graph_utils::CopyAttribute(key, map_node, &stateless_map); in MakeStatelessMap()
Dshuffle_and_repeat_fusion.cc67 graph_utils::CopyAttribute(key, repeat_node, &new_node); in OptimizeAndCollectStats()
Dgraph_utils.h131 void CopyAttribute(const string& attribute_name, const NodeDef& from,
Dgraph_utils.cc269 void CopyAttribute(const string& attribute_name, const NodeDef& from, in CopyAttribute() function
276 CopyAttribute(attribute_name, first, to_node); in ConcatAttributeList()