Searched refs:UnsupportedPartitionedShardingType (Results 1 – 1 of 1) sorted by relevance
203 bool UnsupportedPartitionedShardingType(xla::OpSharding::Type sharding) { in UnsupportedPartitionedShardingType() function260 if (UnsupportedPartitionedShardingType(input_sharding_type)) in ExtractInputsForLogicalDevices()566 if (UnsupportedPartitionedShardingType(output_sharding_type)) in RemapOutputsFromLogicalDevices()