Searched refs:node_and_sharding (Results 1 – 1 of 1) sorted by relevance
1195 Status ParseAndValidateSharding(const NodeAndSharding& node_and_sharding, in ParseAndValidateSharding() argument1199 if (node_and_sharding.sharding.type() == xla::OpSharding::MAXIMAL) { in ParseAndValidateSharding()1201 node_and_sharding.sharding.tile_assignment_devices(0); in ParseAndValidateSharding()1206 result->emplace(node_and_sharding); in ParseAndValidateSharding()1209 if (node_and_sharding.sharding.type() == xla::OpSharding::OTHER) { in ParseAndValidateSharding()1210 for (int64 core : node_and_sharding.sharding.tile_assignment_devices()) { in ParseAndValidateSharding()1216 *result = node_and_sharding; in ParseAndValidateSharding()1224 xla::OpSharding sharding = node_and_sharding.sharding; in ParseAndValidateSharding()1233 DCHECK_NE(node_and_sharding.node, nullptr); in ParseAndValidateSharding()1235 CreateOpMetadataFromNode(*node_and_sharding.node); in ParseAndValidateSharding()[all …]