Searched refs:kShardingAttribute (Results 1 – 1 of 1) sorted by relevance
25 const char kShardingAttribute[] = "_XlaSharding"; variable31 if (!HasNodeAttr(node_def, kShardingAttribute)) { in GetShardingFromNodeDef()36 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, kShardingAttribute, &value)); in GetShardingFromNodeDef()105 if (const AttrValue* attr = src.attrs().Find(kShardingAttribute)) { in SetShardingDeviceAssignmentFromNode()106 dst->AddAttr(kShardingAttribute, *attr); in SetShardingDeviceAssignmentFromNode()