File tree Expand file tree Collapse file tree 5 files changed +114
-57
lines changed Expand file tree Collapse file tree 5 files changed +114
-57
lines changed Original file line number Diff line number Diff line change @@ -537,39 +537,44 @@ extension Tensor: Mergeable where Scalar: TensorFlowFloatingPoint {
537537}
538538
539539/// Concatenates two values.
540- @differentiable ( reverse) public func concatenate< T: Mergeable > (
540+ @differentiable ( reverse)
541+ public func concatenate< T: Mergeable > (
541542 _ first: T ,
542543 _ second: T
543544) -> T {
544545 T . concatenate ( first, second)
545546}
546547
547548/// Adds two values and produces their sum.
548- @differentiable ( reverse) public func sum< T: Mergeable > (
549+ @differentiable ( reverse)
550+ public func sum< T: Mergeable > (
549551 _ first: T ,
550552 _ second: T
551553) -> T {
552554 T . sum ( first, second)
553555}
554556
555557/// Averages two values.
556- @differentiable ( reverse) public func average< T: Mergeable > (
558+ @differentiable ( reverse)
559+ public func average< T: Mergeable > (
557560 _ first: T ,
558561 _ second: T
559562) -> T {
560563 T . average ( first, second)
561564}
562565
563566/// Multiplies two values.
564- @differentiable ( reverse) public func multiply< T: Mergeable > (
567+ @differentiable ( reverse)
568+ public func multiply< T: Mergeable > (
565569 _ first: T ,
566570 _ second: T
567571) -> T {
568572 T . multiply ( first, second)
569573}
570574
571575/// Stack two values.
572- @differentiable ( reverse) public func stack< T: Mergeable > (
576+ @differentiable ( reverse)
577+ public func stack< T: Mergeable > (
573578 _ first: T ,
574579 _ second: T
575580) -> T {
Original file line number Diff line number Diff line change @@ -357,7 +357,8 @@ public func huberLoss<Scalar: TensorFlowFloatingPoint>(
357357/// Workaround for TF-1030 so that we can use sum as a default argument for reductions.
358358/// `Tensor<Scalar>.sum()` is the preferred way to do this.
359359// TODO(TF-1030): Remove this and replace with `{ $0.sum() }`.
360- @differentiable ( reverse) public func _sum< Scalar: TensorFlowFloatingPoint > (
360+ @differentiable ( reverse)
361+ public func _sum< Scalar: TensorFlowFloatingPoint > (
361362 _ value: Tensor < Scalar >
362363) -> Tensor < Scalar > {
363364 return value. sum ( )
@@ -366,7 +367,8 @@ public func huberLoss<Scalar: TensorFlowFloatingPoint>(
366367/// Workaround for TF-1030 so that we can use mean as a default argument for reductions.
367368/// `Tensor<Scalar>.mean()` is the preferred way to do this.
368369// TODO(TF-1030): Remove this and replace with `{ $0.mean() }`.
369- @differentiable ( reverse) public func _mean< Scalar: TensorFlowFloatingPoint > (
370+ @differentiable ( reverse)
371+ public func _mean< Scalar: TensorFlowFloatingPoint > (
370372 _ value: Tensor < Scalar >
371373) -> Tensor < Scalar > {
372374 return value. mean ( )
Original file line number Diff line number Diff line change @@ -235,7 +235,8 @@ func logdet<T: TensorFlowFloatingPoint>(_ matrix: Tensor<T>) -> Tensor<T> {
235235///
236236/// - Parameter input: A tensor of shape `[..., M, M]`.
237237@inlinable
238- @differentiable ( reverse) public func cholesky< T: TensorFlowFloatingPoint > ( _ x: Tensor < T > ) -> Tensor < T > {
238+ @differentiable ( reverse)
239+ public func cholesky< T: TensorFlowFloatingPoint > ( _ x: Tensor < T > ) -> Tensor < T > {
239240 _Raw. cholesky ( x)
240241}
241242
@@ -325,7 +326,8 @@ extension Tensor where Scalar: TensorFlowFloatingPoint {
325326/// - Precondition: `matrix` must be a tensor with shape `[..., M, M]`.
326327/// - Precondition: `rhs` must be a tensor with shape `[..., M, K]`.
327328@inlinable
328- @differentiable ( reverse) public func triangularSolve< T: TensorFlowFloatingPoint > (
329+ @differentiable ( reverse)
330+ public func triangularSolve< T: TensorFlowFloatingPoint > (
329331 matrix: Tensor < T > ,
330332 rhs: Tensor < T > ,
331333 lower: Bool = true ,
You can’t perform that action at this time.
0 commit comments