Searched refs:kShardingAttribute (Results 1 – 2 of 2) sorted by relevance
25 const char kShardingAttribute[] = "_XlaSharding"; variable145 if (const AttrValue* attr = src.attrs().Find(kShardingAttribute)) { in SetShardingDeviceAssignmentFromNode()146 dst->AddAttr(kShardingAttribute, *attr); in SetShardingDeviceAssignmentFromNode()152 if (!HasNodeAttr(node_def, kShardingAttribute)) { in GetShardingFromNodeDef()157 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, kShardingAttribute, &value)); in GetShardingFromNodeDef()
90 const char kShardingAttribute[] = "_XlaSharding"; variable2109 args[i]->AddAttr(kShardingAttribute, in AssignArgsAndRetvalsToCores()2207 retvals[i]->AddAttr(kShardingAttribute, in AssignArgsAndRetvalsToCores()