| 78 | } |
| 79 | |
| 80 | StatusOr<HloInstruction*> MakeSliceHlo(HloInstruction* operand, |
| 81 | absl::Span<const int64> start_indices, |
| 82 | absl::Span<const int64> limit_indices, |
| 83 | absl::Span<const int64> strides) { |
| 84 | HloComputation* computation = operand->parent(); |
| 85 | TF_ASSIGN_OR_RETURN(Shape slice_shape, ShapeInference::InferSliceShape( |
| 86 | operand->shape(), start_indices, |
| 87 | limit_indices, strides)); |
| 88 | return computation->AddInstruction(HloInstruction::CreateSlice( |
| 89 | slice_shape, operand, start_indices, limit_indices, strides)); |
| 90 | } |
| 91 | |
| 92 | StatusOr<HloInstruction*> MakeConvolveHlo( |
| 93 | HloInstruction* lhs, HloInstruction* rhs, int64 feature_group_count, |
no test coverage detected