Searched refs:valid_slice (Results 1 – 1 of 1) sorted by relevance
986 auto valid_slice = concat; in ExchangeHaloAndGetValidData() local998 valid_slice = b->AddInstruction( in ExchangeHaloAndGetValidData()1007 valid_slice = b->AddInstruction(HloInstruction::CreateDynamicSlice( in ExchangeHaloAndGetValidData()1013 return valid_slice; in ExchangeHaloAndGetValidData()1021 auto index_shape = ShapeUtil::ChangeElementType(valid_slice->shape(), S32); in ExchangeHaloAndGetValidData()1063 HloInstruction::CreateBroadcast(valid_slice->shape(), pad_value, {})); in ExchangeHaloAndGetValidData()1064 valid_slice = b->AddInstruction( in ExchangeHaloAndGetValidData()1065 HloInstruction::CreateTernary(valid_slice->shape(), HloOpcode::kSelect, in ExchangeHaloAndGetValidData()1066 is_valid, valid_slice, masking_value)); in ExchangeHaloAndGetValidData()1068 return valid_slice; in ExchangeHaloAndGetValidData()