Searched refs:HloDynamicUpdateSliceInstruction (Results 1 – 6 of 6) sorted by relevance
1346 class HloDynamicUpdateSliceInstruction : public HloDynamicIndexInstruction {1348 explicit HloDynamicUpdateSliceInstruction(const Shape& shape,1352 explicit HloDynamicUpdateSliceInstruction(
2254 HloDynamicUpdateSliceInstruction::HloDynamicUpdateSliceInstruction( in HloDynamicUpdateSliceInstruction() function in xla::HloDynamicUpdateSliceInstruction2263 HloDynamicUpdateSliceInstruction::HloDynamicUpdateSliceInstruction( in HloDynamicUpdateSliceInstruction() function in xla::HloDynamicUpdateSliceInstruction
976 Cast<HloDynamicUpdateSliceInstruction>(fusion->fused_expression_root()); in CanDoInPlaceDynamicUpdateSlice()
591 Cast<HloDynamicUpdateSliceInstruction>(dynamic_update_slice) in HandleDynamicUpdateSlice()
1527 Cast<HloDynamicUpdateSliceInstruction>(dynamic_update_slice) in HandleDynamicUpdateSlice()
1004 return absl::make_unique<HloDynamicUpdateSliceInstruction>( in CreateDynamicUpdateSlice()