@@ -34,32 +34,3 @@ for (TA, transA) in _BATCHED_MATRIX_LIST, (TB, transB) in _BATCHED_MATRIX_LIST
34
34
35
35
end
36
36
end
37
-
38
- function ∇batched_mul (Δ:: AbstractArray{T, 3} , A:: AbstractArray{T, 3} , B:: AbstractArray{T, 3} ) where T
39
- (batched_mul (Δ, batched_transpose (B)), batched_mul (batched_transpose (A), Δ))
40
- end
41
-
42
- function ∇batched_mul (Δ:: AbstractArray{T, 3} , A:: BatchedTranspose{T, <: AbstractArray{T, 3}} , B:: AbstractArray{T, 3} ) where T
43
- (batched_mul (B, batched_transpose (Δ)), batched_mul (batched_transpose (A), Δ))
44
- end
45
-
46
- function ∇batched_mul (Δ:: AbstractArray{T, 3} , A:: AbstractArray{T, 3} , B:: BatchedTranspose{T, <: AbstractArray{T, 3}} ) where T
47
- (batched_mul (Δ, batched_transpose (B)), batched_mul (batched_transpose (Δ), A))
48
- end
49
-
50
- function ∇batched_mul (Δ:: AbstractArray{T, 3} , A:: BatchedTranspose{T, <: AbstractArray{T, 3}} , B:: BatchedTranspose{T, <: AbstractArray{T, 3}} ) where T
51
- (batched_mul (batched_transpose (B), batched_transpose (Δ)), batched_mul (batched_transpose (Δ), batched_transpose (A)))
52
- end
53
-
54
- function ∇batched_mul (Δ:: AbstractArray{T, 3} , A:: BatchedAdjoint{T, <: AbstractArray{T, 3}} , B:: AbstractArray{T, 3} ) where T
55
- (batched_mul (B, batched_adjoint (Δ)), batched_mul (batched_adjoint (A), Δ))
56
- end
57
-
58
- function ∇batched_mul (Δ:: AbstractArray{T, 3} , A:: AbstractArray{T, 3} , B:: BatchedAdjoint{T, <: AbstractArray{T, 3}} ) where T
59
- (batched_mul (Δ, batched_adjoint (B)), batched_mul (batched_adjoint (Δ), A))
60
- end
61
-
62
- function ∇batched_mul (Δ:: AbstractArray{T, 3} , A:: BatchedAdjoint{T, <: AbstractArray{T, 3}} , B:: BatchedAdjoint{T, <: AbstractArray{T, 3}} ) where T
63
- (batched_mul (batched_adjoint (B), batched_adjoint (Δ)), batched_mul (batched_adjoint (Δ), batched_adjoint (A)))
64
- end
65
-
0 commit comments