Searched refs:halo_region (Results 1 – 1 of 1) sorted by relevance
408 HloInstruction* halo_region = nullptr; in HaloDuplicateWithSlice() local417 TF_ASSIGN_OR_RETURN(halo_region, in HaloDuplicateWithSlice()420 VLOG(1) << "halo_region " << halo_region->ToString(); in HaloDuplicateWithSlice()422 MakeNoPaddingConfig(halo_region->shape().dimensions_size()); in HaloDuplicateWithSlice()425 TF_ASSIGN_OR_RETURN(halo_region, in HaloDuplicateWithSlice()426 MakePadHlo(halo_region, padding, padding_config_halo)); in HaloDuplicateWithSlice()454 if (halo_region != nullptr) { in HaloDuplicateWithSlice()455 TF_ASSIGN_OR_RETURN(activations, MakeConcatHlo({activations, halo_region}, in HaloDuplicateWithSlice()