Searched refs:NodeAndSharding (Results 1 – 1 of 1) sorted by relevance
1185 struct NodeAndSharding { struct1186 explicit NodeAndSharding(const Node* node, const xla::OpSharding& sharding) in NodeAndSharding() argument1195 Status ParseAndValidateSharding(const NodeAndSharding& node_and_sharding, in ParseAndValidateSharding()1198 absl::optional<NodeAndSharding>* result) { in ParseAndValidateSharding()1237 NodeAndSharding(node_and_sharding.node, core_zero_sharding)); in ParseAndValidateSharding()1260 xla::StatusOr<absl::optional<NodeAndSharding>>1269 return absl::optional<NodeAndSharding>(NodeAndSharding(&node, *sharding)); in ParseInputShardingFromAdjacentNode()1286 return absl::optional<NodeAndSharding>( in ParseInputShardingFromAdjacentNode()1287 NodeAndSharding(maybe_node_with_sharding_info, *sharding_config)); in ParseInputShardingFromAdjacentNode()1290 return absl::optional<NodeAndSharding>(); in ParseInputShardingFromAdjacentNode()[all …]