Searched refs:mask_invalid_region (Results 1 – 4 of 4) sorted by relevance
261 int64* next_channel_id, SpmdBuilder* b, bool mask_invalid_region = true);
332 HloInstruction* pad_value, bool mask_invalid_region = true);
931 int64* next_channel_id, SpmdBuilder* b, bool mask_invalid_region) { in ExchangeHaloAndGetValidData() argument1012 if (!mask_invalid_region) { in ExchangeHaloAndGetValidData()
549 bool mask_invalid_region) { in ReshardAsWindowedInput() argument810 mask_invalid_region); in ReshardAsWindowedInput()