@@ -449,7 +449,7 @@ public extension Tensor where Scalar: BinaryFloatingPoint,
449
449
}
450
450
}
451
451
452
- fileprivate extension Tensor where Scalar: BinaryFloatingPoint {
452
+ fileprivate extension Tensor where Scalar: TensorFlowFloatingPoint {
453
453
private static func glorot(
454
454
fromStandardUniform randomUniform: __shared Tensor< Scalar > ,
455
455
shape: __shared TensorShape
@@ -459,12 +459,7 @@ fileprivate extension Tensor where Scalar: BinaryFloatingPoint {
459
459
let fanIn = shape [ shape. count - 2 ] * receptiveField
460
460
let fanOut = shape [ shape. count - 1 ] * receptiveField
461
461
let minusOneToOne = 2 * randomUniform - 1
462
- #if os(macOS) || os(iOS) || os(watchOS) || os(tvOS)
463
- let _sqrt = Darwin . sqrt as ( Scalar ) -> Scalar
464
- #else
465
- let _sqrt = Glibc . sqrt as ( Scalar ) -> Scalar
466
- #endif
467
- return _sqrt ( Scalar ( 6 ) / Scalar( fanIn + fanOut) ) * minusOneToOne
462
+ return Scalar . sqrt ( Scalar ( 6 ) / Scalar( fanIn + fanOut) ) * minusOneToOne
468
463
}
469
464
}
470
465
@@ -488,8 +483,7 @@ public extension Tensor where Scalar: TensorFlowFloatingPoint {
488
483
}
489
484
}
490
485
491
- public extension Tensor where Scalar: BinaryFloatingPoint ,
492
- Scalar. RawSignificand: FixedWidthInteger {
486
+ public extension Tensor where Scalar: TensorFlowFloatingPoint {
493
487
/// Performs Glorot uniform initialization for the specified shape, creating a tensor by
494
488
/// randomly sampling scalar values from a uniform distribution between `-limit` and `limit`,
495
489
/// where limit is `sqrt(6 / (fanIn + fanOut))` and `fanIn`/`fanOut` represent the number of
0 commit comments