Searched refs:per_partition_size_hlo (Results 1 – 1 of 1) sorted by relevance
2397 auto per_partition_size_hlo = add_hlo(HloInstruction::CreateConstant( in HandleDynamicUpdateSlice() local2399 const Shape& offset_shape = per_partition_size_hlo->shape(); in HandleDynamicUpdateSlice()2402 per_partition_size_hlo)); in HandleDynamicUpdateSlice()2416 per_partition_size_hlo)), in HandleDynamicUpdateSlice()