Searched defs:mask_invalid_region (Results 1 – 2 of 2) sorted by relevance
893 int64_t* next_channel_id, SpmdBuilder* b, bool mask_invalid_region) { in ExchangeHaloAndGetValidData()
654 bool mask_invalid_region) { in ReshardAsWindowedInput()