Searched refs:halo_region (Results 1 – 1 of 1) sorted by relevance
573 HloInstruction* halo_region = nullptr; in HaloDuplicateWithSlice() local582 TF_ASSIGN_OR_RETURN(halo_region, in HaloDuplicateWithSlice()585 VLOG(1) << "halo_region " << halo_region->ToString(); in HaloDuplicateWithSlice()587 MakeNoPaddingConfig(halo_region->shape().dimensions_size()); in HaloDuplicateWithSlice()591 halo_region, MakePadHlo(halo_region, padding, padding_config_halo)); in HaloDuplicateWithSlice()620 if (halo_region != nullptr) { in HaloDuplicateWithSlice()622 MakeConcatHlo({activations, halo_region}, in HaloDuplicateWithSlice()