Searched refs:device_ordinal_attr (Results 1 – 1 of 1) sorted by relevance
1015 AttrValue device_ordinal_attr; in ConstructHostGraph() local1016 device_ordinal_attr.set_i(0); in ConstructHostGraph()1018 attrs["_device_ordinal"] = device_ordinal_attr; in ConstructHostGraph()1128 AttrValue device_ordinal_attr; in ExpandHostGraphIntoMainGraph() local1129 device_ordinal_attr.set_i(0); in ExpandHostGraphIntoMainGraph()1131 attrs["_device_ordinal"] = device_ordinal_attr; in ExpandHostGraphIntoMainGraph()1206 AttrValue device_ordinal_attr; in RewriteShapeInferenceGraph() local1207 device_ordinal_attr.set_i(0); in RewriteShapeInferenceGraph()1209 attrs["_device_ordinal"] = device_ordinal_attr; in RewriteShapeInferenceGraph()1349 AttrValue device_ordinal_attr; in ReplaceKeyPlaceholderWithArgNode() local[all …]