Searched refs:sharding_config (Results 1 – 2 of 2) sorted by relevance
145 const auto sharding_config = new_metadata->args(i).enable_xla_sharding(); in ExtractExecuteNodeInfo() local146 if (sharding_config != TPUCompileMetadataProto::Arg::TENTATIVE && in ExtractExecuteNodeInfo()147 sharding_config != TPUCompileMetadataProto::Arg::ALLOWED) { in ExtractExecuteNodeInfo()
1282 absl::optional<xla::OpSharding> sharding_config, in ParseInputShardingFromAdjacentNode()1285 if (sharding_config.has_value()) { in ParseInputShardingFromAdjacentNode()1287 NodeAndSharding(maybe_node_with_sharding_info, *sharding_config)); in ParseInputShardingFromAdjacentNode()