Searched refs:allow_xla_spmd_partition (Results 1 – 2 of 2) sorted by relevance
132 bool distribute_vars, bool allow_xla_spmd_partition,
4543 bool distribute_vars, bool allow_xla_spmd_partition, in SetDistributedTpuRewritePassOptions() argument4548 allow_xla_spmd_partition_ = allow_xla_spmd_partition; in SetDistributedTpuRewritePassOptions()