File tree Expand file tree Collapse file tree 3 files changed +5
-4
lines changed
branches/tensorflow/stdlib/public/TensorFlow Expand file tree Collapse file tree 3 files changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -816,7 +816,7 @@ refs/tags/swift-DEVELOPMENT-SNAPSHOT-2018-04-25-a: 22f738a831d43aff2b9c9773bcb65
816
816
refs/tags/swift-DEVELOPMENT-SNAPSHOT-2018-05-08-a: 7d98cc16689baba5c8a3b90a9329bdcc1a12b4e9
817
817
refs/heads/cherr42: a566ad54b073c2c56ac0a705d0a5bed9743135a5
818
818
"refs/heads/codable_test_comment_fix": fc8f6824f7f347e1e8db55bff62db385c5728b5a
819
- refs/heads/tensorflow: 878cdf9656e7db6ce9be20fe9719e80d3965d48e
819
+ refs/heads/tensorflow: 06be74be84a9cdf89c9f9f4cad78d60b0b842b7e
820
820
refs/tags/swift-4.1-DEVELOPMENT-SNAPSHOT-2018-05-11-a: 8126fd7a652e2f70ad6d76505239e34fb2ef3e1a
821
821
refs/tags/swift-4.1-DEVELOPMENT-SNAPSHOT-2018-05-12-a: b3fd3dd84df6717f2e2e9df58c6d7e99fed57086
822
822
refs/tags/swift-4.1-DEVELOPMENT-SNAPSHOT-2018-05-13-a: 71135119579039dc321c5f65d870050fe36efda2
Original file line number Diff line number Diff line change @@ -344,8 +344,9 @@ func _vjpPow<T : TensorFlowFloatingPoint>(
344
344
) -> ( Tensor < T > , ( Tensor < T > ) -> ( Tensor < T > , Tensor < T > ) ) {
345
345
let value = pow ( x, y)
346
346
return ( value, { v in
347
- ( ( v * y * pow( x, y- 1 ) ) . unbroadcast ( like: x) ,
348
- ( v * log( x) * value) . unbroadcast ( like: y) )
347
+ let dfdx = v * y * pow( x, y- 1 )
348
+ let dfdy = v * log( x) * value
349
+ return ( dfdx. unbroadcast ( like: x) , dfdy. unbroadcast ( like: y) )
349
350
} )
350
351
}
351
352
Original file line number Diff line number Diff line change @@ -1923,7 +1923,7 @@ internal extension Tensor.IndexPath {
1923
1923
@inlinable
1924
1924
init ( _ ranges: [ TensorRange ] ) {
1925
1925
precondition ( !ranges. isEmpty, " The tensor range collection cannot be empty. " )
1926
- precondition ( ranges. count { $0 == TensorRange . ellipsis } < 2 ,
1926
+ precondition ( ranges. count ( where : { $0 == TensorRange . ellipsis } ) < 2 ,
1927
1927
" Only one ellipsis is allowed per tensor range collection. " )
1928
1928
1929
1929
var begin = [ Int32] ( repeating: 0 , count: ranges. count)
You can’t perform that action at this time.
0 commit comments