Safe Haskell | None |
---|---|
Language | Haskell2010 |
- abort :: forall m'. MonadBuild m' => m' ControlNode
- abort' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode
- abs :: forall v'1 t. OneOf '[Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- abs' :: forall v'1 t. OneOf '[Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- accumulateNV2 :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Shape -> [Tensor v'1 t] -> Tensor Build t
- accumulateNV2' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Shape -> [Tensor v'1 t] -> Tensor Build t
- accumulatorApplyGradient :: forall v'2 v'3 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => Tensor Ref ByteString -> Tensor v'2 Int64 -> Tensor v'3 dtype -> m' ControlNode
- accumulatorApplyGradient' :: forall v'2 v'3 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int64 -> Tensor v'3 dtype -> m' ControlNode
- accumulatorNumAccumulated :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int32)
- accumulatorNumAccumulated' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
- accumulatorSetGlobalStep :: forall v'2 m'. MonadBuild m' => Tensor Ref ByteString -> Tensor v'2 Int64 -> m' ControlNode
- accumulatorSetGlobalStep' :: forall v'2 m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int64 -> m' ControlNode
- accumulatorTakeGradient :: forall v'2 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (Tensor Value dtype)
- accumulatorTakeGradient' :: forall v'2 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (Tensor Value dtype)
- acos :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- acos' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- acosh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- acosh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- add :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- add' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- addManySparseToTensorsMap :: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorType t) => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> m' (Tensor Value Int64)
- addManySparseToTensorsMap' :: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> m' (Tensor Value Int64)
- addN :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float, Variant] t => [Tensor v'1 t] -> Tensor Build t
- addN' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float, Variant] t => OpParams -> [Tensor v'1 t] -> Tensor Build t
- addSparseToTensorsMap :: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorType t) => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> m' (Tensor Value Int64)
- addSparseToTensorsMap' :: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> m' (Tensor Value Int64)
- addV2 :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- addV2' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- adjustContrast :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor Build Float
- adjustContrast' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor Build Float
- adjustContrastv2 :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
- adjustContrastv2' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
- adjustHue :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
- adjustHue' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
- adjustSaturation :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
- adjustSaturation' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
- all :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
- all' :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => OpParams -> Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
- allCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- allCandidateSampler' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- angle :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => Tensor v'1 t -> Tensor Build tout
- angle' :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => OpParams -> Tensor v'1 t -> Tensor Build tout
- anonymousIterator :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle)
- anonymousIterator' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
- any :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
- any' :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => OpParams -> Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
- applyAdaMax :: forall v'4 v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' (Tensor Ref t)
- applyAdaMax' :: forall v'4 v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' (Tensor Ref t)
- applyAdadelta :: forall v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' (Tensor Ref t)
- applyAdadelta' :: forall v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' (Tensor Ref t)
- applyAdagrad :: forall v'3 v'4 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> m' (Tensor Ref t)
- applyAdagrad' :: forall v'3 v'4 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> m' (Tensor Ref t)
- applyAdagradDA :: forall v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 Int64 -> m' (Tensor Ref t)
- applyAdagradDA' :: forall v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 Int64 -> m' (Tensor Ref t)
- applyAdam :: forall v'4 v'5 v'6 v'7 v'8 v'9 v'10 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> m' (Tensor Ref t)
- applyAdam' :: forall v'4 v'5 v'6 v'7 v'8 v'9 v'10 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> m' (Tensor Ref t)
- applyAddSign :: forall v'3 v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' (Tensor Ref t)
- applyAddSign' :: forall v'3 v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' (Tensor Ref t)
- applyCenteredRMSProp :: forall v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' (Tensor Ref t)
- applyCenteredRMSProp' :: forall v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' (Tensor Ref t)
- applyFtrl :: forall v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> m' (Tensor Ref t)
- applyFtrl' :: forall v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> m' (Tensor Ref t)
- applyFtrlV2 :: forall v'4 v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' (Tensor Ref t)
- applyFtrlV2' :: forall v'4 v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' (Tensor Ref t)
- applyGradientDescent :: forall v'2 v'3 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor v'2 t -> Tensor v'3 t -> m' (Tensor Ref t)
- applyGradientDescent' :: forall v'2 v'3 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor v'2 t -> Tensor v'3 t -> m' (Tensor Ref t)
- applyMomentum :: forall v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' (Tensor Ref t)
- applyMomentum' :: forall v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' (Tensor Ref t)
- applyPowerSign :: forall v'3 v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' (Tensor Ref t)
- applyPowerSign' :: forall v'3 v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' (Tensor Ref t)
- applyProximalAdagrad :: forall v'3 v'4 v'5 v'6 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> m' (Tensor Ref t)
- applyProximalAdagrad' :: forall v'3 v'4 v'5 v'6 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> m' (Tensor Ref t)
- applyProximalGradientDescent :: forall v'2 v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' (Tensor Ref t)
- applyProximalGradientDescent' :: forall v'2 v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' (Tensor Ref t)
- applyRMSProp :: forall v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> m' (Tensor Ref t)
- applyRMSProp' :: forall v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> m' (Tensor Ref t)
- approximateEqual :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- approximateEqual' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- argMax :: forall v'1 v'2 t tidx output_type. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] output_type) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
- argMax' :: forall v'1 v'2 t tidx output_type. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] output_type) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
- argMin :: forall v'1 v'2 t tidx output_type. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] output_type) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
- argMin' :: forall v'1 v'2 t tidx output_type. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] output_type) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
- asString :: forall v'1 t. OneOf '[Complex Float, Bool, Int32, Int64, Int8, Double, Float] t => Tensor v'1 t -> Tensor Build ByteString
- asString' :: forall v'1 t. OneOf '[Complex Float, Bool, Int32, Int64, Int8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build ByteString
- asin :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- asin' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- asinh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- asinh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- assert :: forall v'1 v'2 t m'. (MonadBuild m', TensorTypes t) => Tensor v'1 Bool -> TensorList v'2 t -> m' ControlNode
- assert' :: forall v'1 v'2 t m'. (MonadBuild m', TensorTypes t) => OpParams -> Tensor v'1 Bool -> TensorList v'2 t -> m' ControlNode
- assign :: forall v'2 t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
- assign' :: forall v'2 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
- assignAdd :: forall v'2 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
- assignAdd' :: forall v'2 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
- assignAddVariableOp :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
- assignAddVariableOp' :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
- assignSub :: forall v'2 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
- assignSub' :: forall v'2 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
- assignSubVariableOp :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
- assignSubVariableOp' :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
- assignVariableOp :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
- assignVariableOp' :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
- atan :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- atan' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- atan2 :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- atan2' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- atanh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- atanh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- audioSpectrogram :: Int64 -> Int64 -> Tensor v'1 Float -> Tensor Build Float
- audioSpectrogram' :: OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor Build Float
- audioSummary :: Float -> Tensor v'1 ByteString -> Tensor v'2 Float -> Tensor Build ByteString
- audioSummary' :: OpParams -> Float -> Tensor v'1 ByteString -> Tensor v'2 Float -> Tensor Build ByteString
- audioSummaryV2 :: Tensor v'1 ByteString -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build ByteString
- audioSummaryV2' :: OpParams -> Tensor v'1 ByteString -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build ByteString
- avgPool :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- avgPool' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- avgPool3D :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- avgPool3D' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- avgPool3DGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
- avgPool3DGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
- avgPoolGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
- avgPoolGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
- barrier :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Ref ByteString)
- barrier' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Ref ByteString)
- barrierClose :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' ControlNode
- barrierClose' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' ControlNode
- barrierIncompleteSize :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int32)
- barrierIncompleteSize' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
- barrierInsertMany :: forall v'2 v'3 t m'. (MonadBuild m', TensorType t) => Int64 -> Tensor Ref ByteString -> Tensor v'2 ByteString -> Tensor v'3 t -> m' ControlNode
- barrierInsertMany' :: forall v'2 v'3 t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Tensor Ref ByteString -> Tensor v'2 ByteString -> Tensor v'3 t -> m' ControlNode
- barrierReadySize :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int32)
- barrierReadySize' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
- barrierTakeMany :: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (Tensor Value Int64, Tensor Value ByteString, TensorList Value component_types)
- barrierTakeMany' :: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (Tensor Value Int64, Tensor Value ByteString, TensorList Value component_types)
- batch :: forall v'1 t. TensorTypes t => Int64 -> Int64 -> Int64 -> Int64 -> TensorList v'1 t -> (TensorList Build t, Tensor Build Int64, Tensor Build Int64)
- batch' :: forall v'1 t. TensorTypes t => OpParams -> Int64 -> Int64 -> Int64 -> Int64 -> TensorList v'1 t -> (TensorList Build t, Tensor Build Int64, Tensor Build Int64)
- batchCholesky :: forall v'1 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor Build t
- batchCholesky' :: forall v'1 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- batchCholeskyGrad :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- batchCholeskyGrad' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- batchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
- batchDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
- batchFFT :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchFFT' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchFFT2D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchFFT2D' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchFFT3D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchFFT3D' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchIFFT :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchIFFT' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchIFFT2D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchIFFT2D' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchIFFT3D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchIFFT3D' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
- batchMatMul :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- batchMatMul' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- batchMatrixBandPart :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build t
- batchMatrixBandPart' :: forall v'1 v'2 v'3 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build t
- batchMatrixDeterminant :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor Build t
- batchMatrixDeterminant' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- batchMatrixDiag :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- batchMatrixDiag' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- batchMatrixDiagPart :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- batchMatrixDiagPart' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- batchMatrixInverse :: forall v'1 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor Build t
- batchMatrixInverse' :: forall v'1 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- batchMatrixSetDiag :: forall v'1 v'2 t. TensorType t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- batchMatrixSetDiag' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- batchMatrixSolve :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- batchMatrixSolve' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- batchMatrixSolveLs :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Double -> Tensor Build t
- batchMatrixSolveLs' :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Double -> Tensor Build t
- batchMatrixTriangularSolve :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- batchMatrixTriangularSolve' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- batchNormWithGlobalNormalization :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Bool -> Float -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor Build t
- batchNormWithGlobalNormalization' :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Bool -> Float -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor Build t
- batchNormWithGlobalNormalizationGrad :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Bool -> Float -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
- batchNormWithGlobalNormalizationGrad' :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Bool -> Float -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
- batchSelfAdjointEig :: forall v'1 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor Build t
- batchSelfAdjointEig' :: forall v'1 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- batchSelfAdjointEigV2 :: forall v'1 t. OneOf '[Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t)
- batchSelfAdjointEigV2' :: forall v'1 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
- batchSvd :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
- batchSvd' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
- batchToSpace :: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx) => Int64 -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- batchToSpace' :: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx) => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- batchToSpaceND :: forall v'1 v'2 v'3 t tblock_shape tcrops. (TensorType t, OneOf '[Int32, Int64] tblock_shape, OneOf '[Int32, Int64] tcrops) => Tensor v'1 t -> Tensor v'2 tblock_shape -> Tensor v'3 tcrops -> Tensor Build t
- batchToSpaceND' :: forall v'1 v'2 v'3 t tblock_shape tcrops. (TensorType t, OneOf '[Int32, Int64] tblock_shape, OneOf '[Int32, Int64] tcrops) => OpParams -> Tensor v'1 t -> Tensor v'2 tblock_shape -> Tensor v'3 tcrops -> Tensor Build t
- betainc :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- betainc' :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- biasAdd :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- biasAdd' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- biasAddGrad :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
- biasAddGrad' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- biasAddV1 :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- biasAddV1' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- bigQueryReader :: forall m'. MonadBuild m' => Int64 -> m' (Tensor Ref ByteString)
- bigQueryReader' :: forall m'. MonadBuild m' => OpParams -> Int64 -> m' (Tensor Ref ByteString)
- bincount :: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- bincount' :: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64, Double, Float] t => OpParams -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- bitcast :: forall v'1 t type'. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t, OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] type') => Tensor v'1 t -> Tensor Build type'
- bitcast' :: forall v'1 t type'. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t, OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] type') => OpParams -> Tensor v'1 t -> Tensor Build type'
- bitwiseAnd :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- bitwiseAnd' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- bitwiseOr :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- bitwiseOr' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- bitwiseXor :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- bitwiseXor' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- boostedTreesCalculateBestGainsPerFeature :: Int64 -> Tensor v'1 Int32 -> [Tensor v'2 Float] -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> ([Tensor Build Int32], [Tensor Build Float], [Tensor Build Int32], [Tensor Build Float], [Tensor Build Float])
- boostedTreesCalculateBestGainsPerFeature' :: OpParams -> Int64 -> Tensor v'1 Int32 -> [Tensor v'2 Float] -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> ([Tensor Build Int32], [Tensor Build Float], [Tensor Build Int32], [Tensor Build Float], [Tensor Build Float])
- boostedTreesCreateEnsemble :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
- boostedTreesCreateEnsemble' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
- boostedTreesDeserializeEnsemble :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
- boostedTreesDeserializeEnsemble' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
- boostedTreesEnsembleResourceHandleOp :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
- boostedTreesEnsembleResourceHandleOp' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
- boostedTreesGetEnsembleStates :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32, Tensor Value Int32, Tensor Value Int32)
- boostedTreesGetEnsembleStates' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32, Tensor Value Int32, Tensor Value Int32)
- boostedTreesMakeStatsSummary :: Int64 -> Int64 -> Tensor v'1 Int32 -> Tensor v'2 Float -> Tensor v'3 Float -> [Tensor v'4 Int32] -> Tensor Build Float
- boostedTreesMakeStatsSummary' :: OpParams -> Int64 -> Int64 -> Tensor v'1 Int32 -> Tensor v'2 Float -> Tensor v'3 Float -> [Tensor v'4 Int32] -> Tensor Build Float
- boostedTreesPredict :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> [Tensor v'2 Int32] -> m' (Tensor Value Float)
- boostedTreesPredict' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> [Tensor v'2 Int32] -> m' (Tensor Value Float)
- boostedTreesSerializeEnsemble :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int64, Tensor Value ByteString)
- boostedTreesSerializeEnsemble' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64, Tensor Value ByteString)
- boostedTreesTrainingPredict :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> [Tensor v'4 Int32] -> m' (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
- boostedTreesTrainingPredict' :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> [Tensor v'4 Int32] -> m' (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
- boostedTreesUpdateEnsemble :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> [Tensor v'3 Int32] -> [Tensor v'4 Float] -> [Tensor v'5 Int32] -> [Tensor v'6 Float] -> [Tensor v'7 Float] -> Tensor v'8 Int32 -> Tensor v'9 Float -> m' ControlNode
- boostedTreesUpdateEnsemble' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> [Tensor v'3 Int32] -> [Tensor v'4 Float] -> [Tensor v'5 Int32] -> [Tensor v'6 Float] -> [Tensor v'7 Float] -> Tensor v'8 Int32 -> Tensor v'9 Float -> m' ControlNode
- broadcastArgs :: forall v'1 v'2 t. OneOf '[Int32, Int64] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- broadcastArgs' :: forall v'1 v'2 t. OneOf '[Int32, Int64] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- broadcastGradientArgs :: forall v'1 v'2 t. OneOf '[Int32, Int64] t => Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
- broadcastGradientArgs' :: forall v'1 v'2 t. OneOf '[Int32, Int64] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
- broadcastTo :: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- broadcastTo' :: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- bucketize :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 t -> Tensor Build Int32
- bucketize' :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build Int32
- bytesProducedStatsDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
- bytesProducedStatsDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
- cTCBeamSearchDecoder :: Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Int32 -> ([Tensor Build Int64], [Tensor Build Int64], [Tensor Build Int64], Tensor Build Float)
- cTCBeamSearchDecoder' :: OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Int32 -> ([Tensor Build Int64], [Tensor Build Int64], [Tensor Build Int64], Tensor Build Float)
- cTCGreedyDecoder :: Tensor v'1 Float -> Tensor v'2 Int32 -> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64, Tensor Build Float)
- cTCGreedyDecoder' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Int32 -> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64, Tensor Build Float)
- cTCLoss :: Tensor v'1 Float -> Tensor v'2 Int64 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> (Tensor Build Float, Tensor Build Float)
- cTCLoss' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Int64 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> (Tensor Build Float, Tensor Build Float)
- cacheDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
- cacheDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
- cast :: forall v'1 srcT dstT. (TensorType srcT, TensorType dstT) => Tensor v'1 srcT -> Tensor Build dstT
- cast' :: forall v'1 srcT dstT. (TensorType srcT, TensorType dstT) => OpParams -> Tensor v'1 srcT -> Tensor Build dstT
- ceil :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- ceil' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- checkNumerics :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- checkNumerics' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- cholesky :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor Build t
- cholesky' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- choleskyGrad :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- choleskyGrad' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- clipByValue :: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- clipByValue' :: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- closeSummaryWriter :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
- closeSummaryWriter' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
- collectiveBcastRecv :: forall t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> Int64 -> Int64 -> Shape -> m' (Tensor Value t)
- collectiveBcastRecv' :: forall t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> Int64 -> Int64 -> Shape -> m' (Tensor Value t)
- collectiveBcastSend :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> Int64 -> Int64 -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
- collectiveBcastSend' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> Int64 -> Int64 -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
- collectiveReduce :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> Int64 -> Int64 -> Tensor v'1 t -> m' (Tensor Value t)
- collectiveReduce' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> Int64 -> Int64 -> Tensor v'1 t -> m' (Tensor Value t)
- compareAndBitpack :: forall v'1 v'2 t. OneOf '[Bool, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Word8
- compareAndBitpack' :: forall v'1 v'2 t. OneOf '[Bool, Int16, Int32, Int64, Int8, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Word8
- complex :: forall v'1 v'2 t tout. (OneOf '[Double, Float] t, OneOf '[Complex Double, Complex Float] tout) => Tensor v'1 t -> Tensor v'2 t -> Tensor Build tout
- complex' :: forall v'1 v'2 t tout. (OneOf '[Double, Float] t, OneOf '[Complex Double, Complex Float] tout) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build tout
- complexAbs :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => Tensor v'1 t -> Tensor Build tout
- complexAbs' :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => OpParams -> Tensor v'1 t -> Tensor Build tout
- computeAccidentalHits :: Int64 -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> (Tensor Build Int32, Tensor Build Int64, Tensor Build Float)
- computeAccidentalHits' :: OpParams -> Int64 -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> (Tensor Build Int32, Tensor Build Int64, Tensor Build Float)
- concat :: forall v'1 v'2 t. TensorType t => Tensor v'1 Int32 -> [Tensor v'2 t] -> Tensor Build t
- concat' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 Int32 -> [Tensor v'2 t] -> Tensor Build t
- concatOffset :: Tensor v'1 Int32 -> [Tensor v'2 Int32] -> [Tensor Build Int32]
- concatOffset' :: OpParams -> Tensor v'1 Int32 -> [Tensor v'2 Int32] -> [Tensor Build Int32]
- concatV2 :: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx) => [Tensor v'1 t] -> Tensor v'2 tidx -> Tensor Build t
- concatV2' :: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx) => OpParams -> [Tensor v'1 t] -> Tensor v'2 tidx -> Tensor Build t
- concatenateDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
- concatenateDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
- conditionalAccumulator :: forall m'. MonadBuild m' => DataType -> Shape -> m' (Tensor Ref ByteString)
- conditionalAccumulator' :: forall m'. MonadBuild m' => OpParams -> DataType -> Shape -> m' (Tensor Ref ByteString)
- configureDistributedTPU :: forall m'. MonadBuild m' => m' (Tensor Value ByteString)
- configureDistributedTPU' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ByteString)
- conj :: forall v'1 t. OneOf '[Complex Double, Complex Float, Variant] t => Tensor v'1 t -> Tensor Build t
- conj' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Variant] t => OpParams -> Tensor v'1 t -> Tensor Build t
- conjugateTranspose :: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm) => Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
- conjugateTranspose' :: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm) => OpParams -> Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
- const :: forall dtype. TensorType dtype => Tensor Build dtype
- const' :: forall dtype. TensorType dtype => OpParams -> Tensor Build dtype
- consumeMutexLock :: forall v'1 m'. MonadBuild m' => Tensor v'1 Variant -> m' ControlNode
- consumeMutexLock' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 Variant -> m' ControlNode
- controlTrigger :: forall m'. MonadBuild m' => m' ControlNode
- controlTrigger' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode
- conv2D :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- conv2D' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- conv2DBackpropFilter :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- conv2DBackpropFilter' :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- conv2DBackpropInput :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => Tensor v'1 Int32 -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- conv2DBackpropInput' :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- conv3D :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- conv3D' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- conv3DBackpropFilter :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- conv3DBackpropFilter' :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- conv3DBackpropFilterV2 :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- conv3DBackpropFilterV2' :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- conv3DBackpropInput :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- conv3DBackpropInput' :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- conv3DBackpropInputV2 :: forall v'1 v'2 v'3 t tshape. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tshape) => Tensor v'1 tshape -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- conv3DBackpropInputV2' :: forall v'1 v'2 v'3 t tshape. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tshape) => OpParams -> Tensor v'1 tshape -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- copy :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- copy' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- copyHost :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- copyHost' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- cos :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- cos' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- cosh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- cosh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- countUpTo :: forall t m'. (MonadBuild m', OneOf '[Int32, Int64] t) => Int64 -> Tensor Ref t -> m' (Tensor Value t)
- countUpTo' :: forall t m'. (MonadBuild m', OneOf '[Int32, Int64] t) => OpParams -> Int64 -> Tensor Ref t -> m' (Tensor Value t)
- createSummaryDbWriter :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> Tensor v'5 ByteString -> m' ControlNode
- createSummaryDbWriter' :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> Tensor v'5 ByteString -> m' ControlNode
- createSummaryFileWriter :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 ByteString -> m' ControlNode
- createSummaryFileWriter' :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 ByteString -> m' ControlNode
- cropAndResize :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor Build Float
- cropAndResize' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor Build Float
- cropAndResizeGradBoxes :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 Float -> Tensor v'2 t -> Tensor v'3 Float -> Tensor v'4 Int32 -> Tensor Build Float
- cropAndResizeGradBoxes' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor v'3 Float -> Tensor v'4 Int32 -> Tensor Build Float
- cropAndResizeGradImage :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor Build t
- cropAndResizeGradImage' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor Build t
- cross :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- cross' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- crossReplicaSum :: forall v'1 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor Build t
- crossReplicaSum' :: forall v'1 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- cudnnRNN :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
- cudnnRNN' :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
- cudnnRNNBackprop :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> Tensor v'11 t -> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
- cudnnRNNBackprop' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> Tensor v'11 t -> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
- cudnnRNNBackpropV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> Tensor v'11 t -> Tensor v'12 Int8 -> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
- cudnnRNNBackpropV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> Tensor v'11 t -> Tensor v'12 Int8 -> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
- cudnnRNNCanonicalToParams :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Word16, Double, Float] t => Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> [Tensor v'4 t] -> [Tensor v'5 t] -> Tensor Build t
- cudnnRNNCanonicalToParams' :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> [Tensor v'4 t] -> [Tensor v'5 t] -> Tensor Build t
- cudnnRNNParamsSize :: forall v'1 v'2 v'3 s. OneOf '[Int32, Int64] s => DataType -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build s
- cudnnRNNParamsSize' :: forall v'1 v'2 v'3 s. OneOf '[Int32, Int64] s => OpParams -> DataType -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build s
- cudnnRNNParamsToCanonical :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => Int64 -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 t -> ([Tensor Build t], [Tensor Build t])
- cudnnRNNParamsToCanonical' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => OpParams -> Int64 -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 t -> ([Tensor Build t], [Tensor Build t])
- cudnnRNNV2 :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value Int8)
- cudnnRNNV2' :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value Int8)
- cumprod :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- cumprod' :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- cumsum :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- cumsum' :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- dataFormatDimMap :: forall v'1 t. OneOf '[Int32, Int64] t => Tensor v'1 t -> Tensor Build t
- dataFormatDimMap' :: forall v'1 t. OneOf '[Int32, Int64] t => OpParams -> Tensor v'1 t -> Tensor Build t
- dataFormatVecPermute :: forall v'1 t. OneOf '[Int32, Int64] t => Tensor v'1 t -> Tensor Build t
- dataFormatVecPermute' :: forall v'1 t. OneOf '[Int32, Int64] t => OpParams -> Tensor v'1 t -> Tensor Build t
- datasetToSingleElement :: forall v'1 output_types. TensorTypes output_types => Tensor v'1 Variant -> TensorList Build output_types
- datasetToSingleElement' :: forall v'1 output_types. TensorTypes output_types => OpParams -> Tensor v'1 Variant -> TensorList Build output_types
- datasetToTFRecord :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' ControlNode
- datasetToTFRecord' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' ControlNode
- debugGradientIdentity :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- debugGradientIdentity' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- debugGradientRefIdentity :: forall t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> m' (Tensor Ref t)
- debugGradientRefIdentity' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> m' (Tensor Ref t)
- debugIdentity :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- debugIdentity' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- debugNanCount :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build Int64
- debugNanCount' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build Int64
- debugNumericSummary :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build Double
- debugNumericSummary' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build Double
- decodeAndCropJpeg :: Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build Word8
- decodeAndCropJpeg' :: OpParams -> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build Word8
- decodeBase64 :: Tensor v'1 ByteString -> Tensor Build ByteString
- decodeBase64' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
- decodeBmp :: Tensor v'1 ByteString -> Tensor Build Word8
- decodeBmp' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
- decodeCSV :: forall v'1 v'2 oUT_TYPE. OneOfs '[ByteString, Int32, Int64, Double, Float] oUT_TYPE => Tensor v'1 ByteString -> TensorList v'2 oUT_TYPE -> TensorList Build oUT_TYPE
- decodeCSV' :: forall v'1 v'2 oUT_TYPE. OneOfs '[ByteString, Int32, Int64, Double, Float] oUT_TYPE => OpParams -> Tensor v'1 ByteString -> TensorList v'2 oUT_TYPE -> TensorList Build oUT_TYPE
- decodeCompressed :: Tensor v'1 ByteString -> Tensor Build ByteString
- decodeCompressed' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
- decodeGif :: Tensor v'1 ByteString -> Tensor Build Word8
- decodeGif' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
- decodeJSONExample :: Tensor v'1 ByteString -> Tensor Build ByteString
- decodeJSONExample' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
- decodeJpeg :: Tensor v'1 ByteString -> Tensor Build Word8
- decodeJpeg' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
- decodePng :: forall v'1 dtype. OneOf '[Word16, Word8] dtype => Tensor v'1 ByteString -> Tensor Build dtype
- decodePng' :: forall v'1 dtype. OneOf '[Word16, Word8] dtype => OpParams -> Tensor v'1 ByteString -> Tensor Build dtype
- decodeProtoV2 :: forall v'1 output_types. TensorTypes output_types => Tensor v'1 ByteString -> (Tensor Build Int32, TensorList Build output_types)
- decodeProtoV2' :: forall v'1 output_types. TensorTypes output_types => OpParams -> Tensor v'1 ByteString -> (Tensor Build Int32, TensorList Build output_types)
- decodeRaw :: forall v'1 out_type. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] out_type => Tensor v'1 ByteString -> Tensor Build out_type
- decodeRaw' :: forall v'1 out_type. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] out_type => OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
- decodeWav :: Tensor v'1 ByteString -> (Tensor Build Float, Tensor Build Int32)
- decodeWav' :: OpParams -> Tensor v'1 ByteString -> (Tensor Build Float, Tensor Build Int32)
- deepCopy :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value t)
- deepCopy' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value t)
- deleteSessionTensor :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' ControlNode
- deleteSessionTensor' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' ControlNode
- denseToDenseSetOperation :: forall v'1 v'2 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => Tensor v'1 t -> Tensor v'2 t -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- denseToDenseSetOperation' :: forall v'1 v'2 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- denseToSparseBatchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
- denseToSparseBatchDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
- denseToSparseSetOperation :: forall v'1 v'2 v'3 v'4 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 t -> Tensor v'4 Int64 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- denseToSparseSetOperation' :: forall v'1 v'2 v'3 v'4 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 t -> Tensor v'4 Int64 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- depthToSpace :: forall v'1 t. TensorType t => Int64 -> Tensor v'1 t -> Tensor Build t
- depthToSpace' :: forall v'1 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> Tensor Build t
- depthwiseConv2dNative :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- depthwiseConv2dNative' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- depthwiseConv2dNativeBackpropFilter :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- depthwiseConv2dNativeBackpropFilter' :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- depthwiseConv2dNativeBackpropInput :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => Tensor v'1 Int32 -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- depthwiseConv2dNativeBackpropInput' :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- dequantize :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build Float
- dequantize' :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build Float
- deserializeIterator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
- deserializeIterator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
- deserializeManySparse :: forall v'1 dtype. TensorType dtype => Tensor v'1 ByteString -> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
- deserializeManySparse' :: forall v'1 dtype. TensorType dtype => OpParams -> Tensor v'1 ByteString -> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
- deserializeSparse :: forall v'1 dtype tserialized. (TensorType dtype, OneOf '[ByteString, Variant] tserialized) => Tensor v'1 tserialized -> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
- deserializeSparse' :: forall v'1 dtype tserialized. (TensorType dtype, OneOf '[ByteString, Variant] tserialized) => OpParams -> Tensor v'1 tserialized -> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
- destroyResourceOp :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
- destroyResourceOp' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
- destroyTemporaryVariable :: forall t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> m' (Tensor Value t)
- destroyTemporaryVariable' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> m' (Tensor Value t)
- diag :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- diag' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- diagPart :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- diagPart' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- digamma :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- digamma' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- dilation2D :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- dilation2D' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- dilation2DBackpropFilter :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- dilation2DBackpropFilter' :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- dilation2DBackpropInput :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- dilation2DBackpropInput' :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- div :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- div' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- drawBoundingBoxes :: forall v'1 v'2 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
- drawBoundingBoxes' :: forall v'1 v'2 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
- dynamicPartition :: forall v'1 v'2 t. TensorType t => Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
- dynamicPartition' :: forall v'1 v'2 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
- dynamicStitch :: forall v'1 v'2 t. TensorType t => [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
- dynamicStitch' :: forall v'1 v'2 t. TensorType t => OpParams -> [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
- editDistance :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. TensorType t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 t -> Tensor v'6 Int64 -> Tensor Build Float
- editDistance' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. TensorType t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 t -> Tensor v'6 Int64 -> Tensor Build Float
- elu :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- elu' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- eluGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- eluGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- empty :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 Int32 -> m' (Tensor Value dtype)
- empty' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 Int32 -> m' (Tensor Value dtype)
- emptyTensorList :: forall v'1 shape_type. OneOf '[Int32, Int64] shape_type => DataType -> Tensor v'1 shape_type -> Tensor Build Variant
- emptyTensorList' :: forall v'1 shape_type. OneOf '[Int32, Int64] shape_type => OpParams -> DataType -> Tensor v'1 shape_type -> Tensor Build Variant
- encodeBase64 :: Tensor v'1 ByteString -> Tensor Build ByteString
- encodeBase64' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
- encodeJpeg :: Tensor v'1 Word8 -> Tensor Build ByteString
- encodeJpeg' :: OpParams -> Tensor v'1 Word8 -> Tensor Build ByteString
- encodePng :: forall v'1 t. OneOf '[Word16, Word8] t => Tensor v'1 t -> Tensor Build ByteString
- encodePng' :: forall v'1 t. OneOf '[Word16, Word8] t => OpParams -> Tensor v'1 t -> Tensor Build ByteString
- encodeProto :: forall v'1 v'2 tinput_types. TensorTypes tinput_types => Tensor v'1 Int32 -> TensorList v'2 tinput_types -> Tensor Build ByteString
- encodeProto' :: forall v'1 v'2 tinput_types. TensorTypes tinput_types => OpParams -> Tensor v'1 Int32 -> TensorList v'2 tinput_types -> Tensor Build ByteString
- encodeWav :: Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build ByteString
- encodeWav' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build ByteString
- enqueueInQueueDataset :: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => Tensor v'1 Variant -> TensorList v'2 tcomponents -> m' ControlNode
- enqueueInQueueDataset' :: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => OpParams -> Tensor v'1 Variant -> TensorList v'2 tcomponents -> m' ControlNode
- enter :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- enter' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- equal :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- equal' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- erf :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- erf' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- erfc :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- erfc' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- exit :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- exit' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- exp :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- exp' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- expandDims :: forall v'1 v'2 t tdim. (TensorType t, OneOf '[Int32, Int64] tdim) => Tensor v'1 t -> Tensor v'2 tdim -> Tensor Build t
- expandDims' :: forall v'1 v'2 t tdim. (TensorType t, OneOf '[Int32, Int64] tdim) => OpParams -> Tensor v'1 t -> Tensor v'2 tdim -> Tensor Build t
- expm1 :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- expm1' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- extractGlimpse :: Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build Float
- extractGlimpse' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build Float
- extractImagePatches :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
- extractImagePatches' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- extractJpegShape :: forall v'1 output_type. OneOf '[Int32, Int64] output_type => Tensor v'1 ByteString -> Tensor Build output_type
- extractJpegShape' :: forall v'1 output_type. OneOf '[Int32, Int64] output_type => OpParams -> Tensor v'1 ByteString -> Tensor Build output_type
- fFT :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
- fFT' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
- fFT2D :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
- fFT2D' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
- fFT3D :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
- fFT3D' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
- fIFOQueue :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Ref ByteString)
- fIFOQueue' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Ref ByteString)
- fIFOQueueV2 :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle)
- fIFOQueueV2' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
- fact :: Tensor Build ByteString
- fact' :: OpParams -> Tensor Build ByteString
- fakeQuantWithMinMaxArgs :: Tensor v'1 Float -> Tensor Build Float
- fakeQuantWithMinMaxArgs' :: OpParams -> Tensor v'1 Float -> Tensor Build Float
- fakeQuantWithMinMaxArgsGradient :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
- fakeQuantWithMinMaxArgsGradient' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
- fakeQuantWithMinMaxVars :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build Float
- fakeQuantWithMinMaxVars' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build Float
- fakeQuantWithMinMaxVarsGradient :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
- fakeQuantWithMinMaxVarsGradient' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
- fakeQuantWithMinMaxVarsPerChannel :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build Float
- fakeQuantWithMinMaxVarsPerChannel' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build Float
- fakeQuantWithMinMaxVarsPerChannelGradient :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
- fakeQuantWithMinMaxVarsPerChannelGradient' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
- fakeQueue :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Ref ByteString)
- fakeQueue' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Ref ByteString)
- fill :: forall v'1 v'2 t index_type. (TensorType t, OneOf '[Int32, Int64] index_type) => Tensor v'1 index_type -> Tensor v'2 t -> Tensor Build t
- fill' :: forall v'1 v'2 t index_type. (TensorType t, OneOf '[Int32, Int64] index_type) => OpParams -> Tensor v'1 index_type -> Tensor v'2 t -> Tensor Build t
- fixedLengthRecordDataset :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> m' (Tensor Value Variant)
- fixedLengthRecordDataset' :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> m' (Tensor Value Variant)
- fixedLengthRecordReader :: forall m'. MonadBuild m' => Int64 -> m' (Tensor Ref ByteString)
- fixedLengthRecordReader' :: forall m'. MonadBuild m' => OpParams -> Int64 -> m' (Tensor Ref ByteString)
- fixedLengthRecordReaderV2 :: forall m'. MonadBuild m' => Int64 -> m' (Tensor Value ResourceHandle)
- fixedLengthRecordReaderV2' :: forall m'. MonadBuild m' => OpParams -> Int64 -> m' (Tensor Value ResourceHandle)
- fixedUnigramCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- fixedUnigramCandidateSampler' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- floor :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- floor' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- floorDiv :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- floorDiv' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- floorMod :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- floorMod' :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- flushSummaryWriter :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
- flushSummaryWriter' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
- fractionalAvgPool :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
- fractionalAvgPool' :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
- fractionalAvgPoolGrad :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build t
- fractionalAvgPoolGrad' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Int64, Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build t
- fractionalMaxPool :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
- fractionalMaxPool' :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
- fractionalMaxPoolGrad :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> Tensor Build t
- fractionalMaxPoolGrad' :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int32, Int64, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> Tensor Build t
- fusedBatchNorm :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
- fusedBatchNorm' :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
- fusedBatchNormGrad :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
- fusedBatchNormGrad' :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
- fusedBatchNormGradV2 :: forall v'1 v'2 v'3 v'4 v'5 t u. (OneOf '[Word16, Float] t, OneOf '[Float] u) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Float -> Tensor v'4 u -> Tensor v'5 u -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u)
- fusedBatchNormGradV2' :: forall v'1 v'2 v'3 v'4 v'5 t u. (OneOf '[Word16, Float] t, OneOf '[Float] u) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Float -> Tensor v'4 u -> Tensor v'5 u -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u)
- fusedBatchNormV2 :: forall v'1 v'2 v'3 v'4 v'5 t u. (OneOf '[Word16, Float] t, OneOf '[Float] u) => Tensor v'1 t -> Tensor v'2 u -> Tensor v'3 u -> Tensor v'4 u -> Tensor v'5 u -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u)
- fusedBatchNormV2' :: forall v'1 v'2 v'3 v'4 v'5 t u. (OneOf '[Word16, Float] t, OneOf '[Float] u) => OpParams -> Tensor v'1 t -> Tensor v'2 u -> Tensor v'3 u -> Tensor v'4 u -> Tensor v'5 u -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u)
- fusedPadConv2D :: forall v'1 v'2 v'3 t. OneOf '[Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- fusedPadConv2D' :: forall v'1 v'2 v'3 t. OneOf '[Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- fusedResizeAndPadConv2D :: forall v'1 v'2 v'3 v'4 t. OneOf '[Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 t -> Tensor Build t
- fusedResizeAndPadConv2D' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 t -> Tensor Build t
- gather :: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices) => Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor Build tparams
- gather' :: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor Build tparams
- gatherNd :: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices) => Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor Build tparams
- gatherNd' :: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor Build tparams
- gatherV2 :: forall v'1 v'2 v'3 tparams tindices taxis. (TensorType tparams, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] taxis) => Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor v'3 taxis -> Tensor Build tparams
- gatherV2' :: forall v'1 v'2 v'3 tparams tindices taxis. (TensorType tparams, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] taxis) => OpParams -> Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor v'3 taxis -> Tensor Build tparams
- gcsConfigureBlockCache :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 Word64 -> Tensor v'2 Word64 -> Tensor v'3 Word64 -> m' ControlNode
- gcsConfigureBlockCache' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 Word64 -> Tensor v'2 Word64 -> Tensor v'3 Word64 -> m' ControlNode
- gcsConfigureCredentials :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' ControlNode
- gcsConfigureCredentials' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' ControlNode
- generateBigQueryReaderPartitions :: Int64 -> Int64 -> Tensor Build ByteString
- generateBigQueryReaderPartitions' :: OpParams -> Int64 -> Int64 -> Tensor Build ByteString
- generateVocabRemapping :: Int64 -> Int64 -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> (Tensor Build Int64, Tensor Build Int32)
- generateVocabRemapping' :: OpParams -> Int64 -> Int64 -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> (Tensor Build Int64, Tensor Build Int32)
- getSessionHandle :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value ByteString)
- getSessionHandle' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value ByteString)
- getSessionHandleV2 :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value ResourceHandle)
- getSessionHandleV2' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value ResourceHandle)
- getSessionTensor :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ByteString -> m' (Tensor Value dtype)
- getSessionTensor' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value dtype)
- greater :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- greater' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- greaterEqual :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- greaterEqual' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- guaranteeConst :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value t)
- guaranteeConst' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value t)
- hSVToRGB :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- hSVToRGB' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- hashTable :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Ref ByteString)
- hashTable' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
- hashTableV2 :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Value ResourceHandle)
- hashTableV2' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Value ResourceHandle)
- histogramFixedWidth :: forall v'1 v'2 v'3 t dtype. (OneOf '[Int32, Int64, Double, Float] t, OneOf '[Int32, Int64] dtype) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build dtype
- histogramFixedWidth' :: forall v'1 v'2 v'3 t dtype. (OneOf '[Int32, Int64, Double, Float] t, OneOf '[Int32, Int64] dtype) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build dtype
- histogramSummary :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
- histogramSummary' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
- iFFT :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
- iFFT' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
- iFFT2D :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
- iFFT2D' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
- iFFT3D :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
- iFFT3D' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
- iRFFT :: Tensor v'1 (Complex Float) -> Tensor v'2 Int32 -> Tensor Build Float
- iRFFT' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor v'2 Int32 -> Tensor Build Float
- iRFFT2D :: Tensor v'1 (Complex Float) -> Tensor v'2 Int32 -> Tensor Build Float
- iRFFT2D' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor v'2 Int32 -> Tensor Build Float
- iRFFT3D :: Tensor v'1 (Complex Float) -> Tensor v'2 Int32 -> Tensor Build Float
- iRFFT3D' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor v'2 Int32 -> Tensor Build Float
- identity :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- identity' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- identityN :: forall v'1 t. TensorTypes t => TensorList v'1 t -> TensorList Build t
- identityN' :: forall v'1 t. TensorTypes t => OpParams -> TensorList v'1 t -> TensorList Build t
- identityReader :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
- identityReader' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
- identityReaderV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
- identityReaderV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
- igamma :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- igamma' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- igammac :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- igammac' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- imag :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => Tensor v'1 t -> Tensor Build tout
- imag' :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => OpParams -> Tensor v'1 t -> Tensor Build tout
- imageSummary :: forall v'1 v'2 t. OneOf '[Word16, Word8, Double, Float] t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
- imageSummary' :: forall v'1 v'2 t. OneOf '[Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
- immutableConst :: forall dtype. TensorType dtype => Shape -> Tensor Build dtype
- immutableConst' :: forall dtype. TensorType dtype => OpParams -> Shape -> Tensor Build dtype
- importEvent :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
- importEvent' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
- inTopK :: forall v'1 v'2 t. OneOf '[Int32, Int64] t => Int64 -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build Bool
- inTopK' :: forall v'1 v'2 t. OneOf '[Int32, Int64] t => OpParams -> Int64 -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build Bool
- inTopKV2 :: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64] t => Tensor v'1 Float -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build Bool
- inTopKV2' :: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64] t => OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build Bool
- infeedDequeue :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Value dtype)
- infeedDequeue' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Value dtype)
- infeedDequeueTuple :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => m' (TensorList Value dtypes)
- infeedDequeueTuple' :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> m' (TensorList Value dtypes)
- infeedEnqueue :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 dtype -> m' ControlNode
- infeedEnqueue' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 dtype -> m' ControlNode
- infeedEnqueueTuple :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => TensorList v'1 dtypes -> m' ControlNode
- infeedEnqueueTuple' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> TensorList v'1 dtypes -> m' ControlNode
- initializeTable :: forall v'2 v'3 tkey tval m'. (MonadBuild m', TensorType tkey, TensorType tval) => Tensor Ref ByteString -> Tensor v'2 tkey -> Tensor v'3 tval -> m' ControlNode
- initializeTable' :: forall v'2 v'3 tkey tval m'. (MonadBuild m', TensorType tkey, TensorType tval) => OpParams -> Tensor Ref ByteString -> Tensor v'2 tkey -> Tensor v'3 tval -> m' ControlNode
- initializeTableFromTextFile :: forall v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
- initializeTableFromTextFile' :: forall v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
- initializeTableFromTextFileV2 :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
- initializeTableFromTextFileV2' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
- initializeTableV2 :: forall v'1 v'2 v'3 tkey tval m'. (MonadBuild m', TensorType tkey, TensorType tval) => Tensor v'1 ResourceHandle -> Tensor v'2 tkey -> Tensor v'3 tval -> m' ControlNode
- initializeTableV2' :: forall v'1 v'2 v'3 tkey tval m'. (MonadBuild m', TensorType tkey, TensorType tval) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tkey -> Tensor v'3 tval -> m' ControlNode
- inplaceAdd :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- inplaceAdd' :: forall v'1 v'2 v'3 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- inplaceSub :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- inplaceSub' :: forall v'1 v'2 v'3 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- inplaceUpdate :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- inplaceUpdate' :: forall v'1 v'2 v'3 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
- inv :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- inv' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- invGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- invGrad' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- invert :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor Build t
- invert' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => OpParams -> Tensor v'1 t -> Tensor Build t
- invertPermutation :: forall v'1 t. OneOf '[Int32, Int64] t => Tensor v'1 t -> Tensor Build t
- invertPermutation' :: forall v'1 t. OneOf '[Int32, Int64] t => OpParams -> Tensor v'1 t -> Tensor Build t
- isBoostedTreesEnsembleInitialized :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
- isBoostedTreesEnsembleInitialized' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
- isFinite :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build Bool
- isFinite' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build Bool
- isInf :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build Bool
- isInf' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build Bool
- isNan :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build Bool
- isNan' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build Bool
- isVariableInitialized :: forall dtype m'. (MonadBuild m', TensorType dtype) => Tensor Ref dtype -> m' (Tensor Value Bool)
- isVariableInitialized' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor Ref dtype -> m' (Tensor Value Bool)
- iterator :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle)
- iterator' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
- iteratorFromStringHandle :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
- iteratorFromStringHandle' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
- iteratorGetNext :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
- iteratorGetNext' :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => OpParams -> Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
- iteratorGetNextSync :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
- iteratorGetNextSync' :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => OpParams -> Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
- iteratorToStringHandle :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
- iteratorToStringHandle' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
- l2Loss :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- l2Loss' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- lMDBReader :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
- lMDBReader' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
- lRN :: forall v'1 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor Build t
- lRN' :: forall v'1 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- lRNGrad :: forall v'1 v'2 v'3 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- lRNGrad' :: forall v'1 v'2 v'3 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- latencyStatsDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
- latencyStatsDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
- learnedUnigramCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- learnedUnigramCandidateSampler' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- leftShift :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- leftShift' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- less :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- less' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- lessEqual :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- lessEqual' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- lgamma :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- lgamma' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- linSpace :: forall v'1 v'2 v'3 t tidx. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tidx -> Tensor Build t
- linSpace' :: forall v'1 v'2 v'3 t tidx. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tidx -> Tensor Build t
- listDiff :: forall v'1 v'2 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx) => Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build out_idx)
- listDiff' :: forall v'1 v'2 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build out_idx)
- loadAndRemapMatrix :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Float -> m' (Tensor Value Float)
- loadAndRemapMatrix' :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Float -> m' (Tensor Value Float)
- log :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- log' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- log1p :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- log1p' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- logMatrixDeterminant :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t)
- logMatrixDeterminant' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
- logSoftmax :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- logSoftmax' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- logUniformCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- logUniformCandidateSampler' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- logicalAnd :: Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
- logicalAnd' :: OpParams -> Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
- logicalNot :: Tensor v'1 Bool -> Tensor Build Bool
- logicalNot' :: OpParams -> Tensor v'1 Bool -> Tensor Build Bool
- logicalOr :: Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
- logicalOr' :: OpParams -> Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
- lookupTableExport :: forall tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues) => Tensor Ref ByteString -> m' (Tensor Value tkeys, Tensor Value tvalues)
- lookupTableExport' :: forall tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues) => OpParams -> Tensor Ref ByteString -> m' (Tensor Value tkeys, Tensor Value tvalues)
- lookupTableExportV2 :: forall v'1 tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues) => Tensor v'1 ResourceHandle -> m' (Tensor Value tkeys, Tensor Value tvalues)
- lookupTableExportV2' :: forall v'1 tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues) => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value tkeys, Tensor Value tvalues)
- lookupTableFind :: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor Ref ByteString -> Tensor v'2 tin -> Tensor v'3 tout -> m' (Tensor Value tout)
- lookupTableFind' :: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => OpParams -> Tensor Ref ByteString -> Tensor v'2 tin -> Tensor v'3 tout -> m' (Tensor Value tout)
- lookupTableFindV2 :: forall v'1 v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor v'1 ResourceHandle -> Tensor v'2 tin -> Tensor v'3 tout -> m' (Tensor Value tout)
- lookupTableFindV2' :: forall v'1 v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tin -> Tensor v'3 tout -> m' (Tensor Value tout)
- lookupTableImport :: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor Ref ByteString -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
- lookupTableImport' :: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => OpParams -> Tensor Ref ByteString -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
- lookupTableImportV2 :: forall v'1 v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor v'1 ResourceHandle -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
- lookupTableImportV2' :: forall v'1 v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
- lookupTableInsert :: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor Ref ByteString -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
- lookupTableInsert' :: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => OpParams -> Tensor Ref ByteString -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
- lookupTableInsertV2 :: forall v'1 v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor v'1 ResourceHandle -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
- lookupTableInsertV2' :: forall v'1 v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
- lookupTableSize :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int64)
- lookupTableSize' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
- lookupTableSizeV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
- lookupTableSizeV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
- loopCond :: Tensor v'1 Bool -> Tensor Build Bool
- loopCond' :: OpParams -> Tensor v'1 Bool -> Tensor Build Bool
- makeIterator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' ControlNode
- makeIterator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' ControlNode
- mapClear :: forall m'. MonadBuild m' => [DataType] -> m' ControlNode
- mapClear' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' ControlNode
- mapIncompleteSize :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value Int32)
- mapIncompleteSize' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value Int32)
- mapPeek :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
- mapPeek' :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
- mapSize :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value Int32)
- mapSize' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value Int32)
- mapStage :: forall v'1 v'2 v'3 fake_dtypes m'. (MonadBuild m', TensorTypes fake_dtypes) => [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> TensorList v'3 fake_dtypes -> m' ControlNode
- mapStage' :: forall v'1 v'2 v'3 fake_dtypes m'. (MonadBuild m', TensorTypes fake_dtypes) => OpParams -> [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> TensorList v'3 fake_dtypes -> m' ControlNode
- mapUnstage :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
- mapUnstage' :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
- mapUnstageNoKey :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int32 -> m' (Tensor Value Int64, TensorList Value dtypes)
- mapUnstageNoKey' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int32 -> m' (Tensor Value Int64, TensorList Value dtypes)
- matMul :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- matMul' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- matchingFiles :: Tensor v'1 ByteString -> Tensor Build ByteString
- matchingFiles' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
- matrixBandPart :: forall v'1 v'2 v'3 t tindex. (TensorType t, OneOf '[Int32, Int64] tindex) => Tensor v'1 t -> Tensor v'2 tindex -> Tensor v'3 tindex -> Tensor Build t
- matrixBandPart' :: forall v'1 v'2 v'3 t tindex. (TensorType t, OneOf '[Int32, Int64] tindex) => OpParams -> Tensor v'1 t -> Tensor v'2 tindex -> Tensor v'3 tindex -> Tensor Build t
- matrixDeterminant :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor Build t
- matrixDeterminant' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- matrixDiag :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- matrixDiag' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- matrixDiagPart :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- matrixDiagPart' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- matrixExponential :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor Build t
- matrixExponential' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- matrixInverse :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor Build t
- matrixInverse' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- matrixLogarithm :: forall v'1 t. OneOf '[Complex Double, Complex Float] t => Tensor v'1 t -> Tensor Build t
- matrixLogarithm' :: forall v'1 t. OneOf '[Complex Double, Complex Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- matrixSetDiag :: forall v'1 v'2 t. TensorType t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- matrixSetDiag' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- matrixSolve :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- matrixSolve' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- matrixSolveLs :: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Double -> Tensor Build t
- matrixSolveLs' :: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Double -> Tensor Build t
- matrixTriangularSolve :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- matrixTriangularSolve' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- max :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- max' :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- maxPool :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
- maxPool' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- maxPool3D :: forall v'1 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor Build t
- maxPool3D' :: forall v'1 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- maxPool3DGrad :: forall v'1 v'2 v'3 t tInput. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] tInput) => Tensor v'1 tInput -> Tensor v'2 tInput -> Tensor v'3 t -> Tensor Build t
- maxPool3DGrad' :: forall v'1 v'2 v'3 t tInput. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] tInput) => OpParams -> Tensor v'1 tInput -> Tensor v'2 tInput -> Tensor v'3 t -> Tensor Build t
- maxPool3DGradGrad :: forall v'1 v'2 v'3 t. OneOf '[Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- maxPool3DGradGrad' :: forall v'1 v'2 v'3 t. OneOf '[Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- maxPoolGrad :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- maxPoolGrad' :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- maxPoolGradGrad :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- maxPoolGradGrad' :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- maxPoolGradGradV2 :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 Int32 -> Tensor v'5 Int32 -> Tensor Build t
- maxPoolGradGradV2' :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 Int32 -> Tensor v'5 Int32 -> Tensor Build t
- maxPoolGradGradWithArgmax :: forall v'1 v'2 v'3 targmax t. (OneOf '[Int32, Int64] targmax, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 targmax -> Tensor Build t
- maxPoolGradGradWithArgmax' :: forall v'1 v'2 v'3 targmax t. (OneOf '[Int32, Int64] targmax, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 targmax -> Tensor Build t
- maxPoolGradV2 :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 Int32 -> Tensor v'5 Int32 -> Tensor Build t
- maxPoolGradV2' :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 Int32 -> Tensor v'5 Int32 -> Tensor Build t
- maxPoolGradWithArgmax :: forall v'1 v'2 v'3 targmax t. (OneOf '[Int32, Int64] targmax, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 targmax -> Tensor Build t
- maxPoolGradWithArgmax' :: forall v'1 v'2 v'3 targmax t. (OneOf '[Int32, Int64] targmax, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 targmax -> Tensor Build t
- maxPoolV2 :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build t
- maxPoolV2' :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build t
- maxPoolWithArgmax :: forall v'1 targmax t. (OneOf '[Int32, Int64] targmax, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 t -> (Tensor Build t, Tensor Build targmax)
- maxPoolWithArgmax' :: forall v'1 targmax t. (OneOf '[Int32, Int64] targmax, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build targmax)
- maximum :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- maximum' :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- mean :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- mean' :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- merge :: forall v'1 t. TensorType t => [Tensor v'1 t] -> (Tensor Build t, Tensor Build Int32)
- merge' :: forall v'1 t. TensorType t => OpParams -> [Tensor v'1 t] -> (Tensor Build t, Tensor Build Int32)
- mergeSummary :: [Tensor v'1 ByteString] -> Tensor Build ByteString
- mergeSummary' :: OpParams -> [Tensor v'1 ByteString] -> Tensor Build ByteString
- mergeV2Checkpoints :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
- mergeV2Checkpoints' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
- mfcc :: Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build Float
- mfcc' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build Float
- min :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- min' :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- minimum :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- minimum' :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- mirrorPad :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
- mirrorPad' :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => OpParams -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
- mirrorPadGrad :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
- mirrorPadGrad' :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => OpParams -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
- mod :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- mod' :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- mul :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- mul' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- multinomial :: forall v'1 v'2 t output_dtype m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] output_dtype) => Tensor v'1 t -> Tensor v'2 Int32 -> m' (Tensor Value output_dtype)
- multinomial' :: forall v'1 v'2 t output_dtype m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] output_dtype) => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> m' (Tensor Value output_dtype)
- mutableDenseHashTable :: forall v'1 key_dtype m'. (MonadBuild m', TensorType key_dtype) => DataType -> Tensor v'1 key_dtype -> m' (Tensor Ref ByteString)
- mutableDenseHashTable' :: forall v'1 key_dtype m'. (MonadBuild m', TensorType key_dtype) => OpParams -> DataType -> Tensor v'1 key_dtype -> m' (Tensor Ref ByteString)
- mutableDenseHashTableV2 :: forall v'1 key_dtype m'. (MonadBuild m', TensorType key_dtype) => DataType -> Tensor v'1 key_dtype -> m' (Tensor Value ResourceHandle)
- mutableDenseHashTableV2' :: forall v'1 key_dtype m'. (MonadBuild m', TensorType key_dtype) => OpParams -> DataType -> Tensor v'1 key_dtype -> m' (Tensor Value ResourceHandle)
- mutableHashTable :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Ref ByteString)
- mutableHashTable' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
- mutableHashTableOfTensors :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Ref ByteString)
- mutableHashTableOfTensors' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
- mutableHashTableOfTensorsV2 :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Value ResourceHandle)
- mutableHashTableOfTensorsV2' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Value ResourceHandle)
- mutableHashTableV2 :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Value ResourceHandle)
- mutableHashTableV2' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Value ResourceHandle)
- mutexLock :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
- mutexLock' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
- mutexV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
- mutexV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
- neg :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- neg' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- negTrain :: forall v'3 v'4 v'5 m'. MonadBuild m' => Int64 -> Tensor Ref Float -> Tensor Ref Float -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 Float -> m' ControlNode
- negTrain' :: forall v'3 v'4 v'5 m'. MonadBuild m' => OpParams -> Int64 -> Tensor Ref Float -> Tensor Ref Float -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 Float -> m' ControlNode
- nextIteration :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- nextIteration' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- noOp :: forall m'. MonadBuild m' => m' ControlNode
- noOp' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode
- nonMaxSuppression :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor Build Int32
- nonMaxSuppression' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor Build Int32
- nonMaxSuppressionV2 :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Float -> Tensor Build Int32
- nonMaxSuppressionV2' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Float -> Tensor Build Int32
- nonMaxSuppressionV3 :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor Build Int32
- nonMaxSuppressionV3' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor Build Int32
- notEqual :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- notEqual' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
- nthElement :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
- nthElement' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
- oneHot :: forall v'1 v'2 v'3 v'4 t tI. (TensorType t, OneOf '[Int32, Int64, Word8] tI) => Tensor v'1 tI -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build t
- oneHot' :: forall v'1 v'2 v'3 v'4 t tI. (TensorType t, OneOf '[Int32, Int64, Word8] tI) => OpParams -> Tensor v'1 tI -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build t
- onesLike :: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
- onesLike' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- orderedMapClear :: forall m'. MonadBuild m' => [DataType] -> m' ControlNode
- orderedMapClear' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' ControlNode
- orderedMapIncompleteSize :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value Int32)
- orderedMapIncompleteSize' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value Int32)
- orderedMapPeek :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
- orderedMapPeek' :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
- orderedMapSize :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value Int32)
- orderedMapSize' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value Int32)
- orderedMapStage :: forall v'1 v'2 v'3 fake_dtypes m'. (MonadBuild m', TensorTypes fake_dtypes) => [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> TensorList v'3 fake_dtypes -> m' ControlNode
- orderedMapStage' :: forall v'1 v'2 v'3 fake_dtypes m'. (MonadBuild m', TensorTypes fake_dtypes) => OpParams -> [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> TensorList v'3 fake_dtypes -> m' ControlNode
- orderedMapUnstage :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
- orderedMapUnstage' :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
- orderedMapUnstageNoKey :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int32 -> m' (Tensor Value Int64, TensorList Value dtypes)
- orderedMapUnstageNoKey' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int32 -> m' (Tensor Value Int64, TensorList Value dtypes)
- outfeedDequeue :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Value dtype)
- outfeedDequeue' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Value dtype)
- outfeedDequeueTuple :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => m' (TensorList Value dtypes)
- outfeedDequeueTuple' :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> m' (TensorList Value dtypes)
- outfeedEnqueue :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 dtype -> m' ControlNode
- outfeedEnqueue' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 dtype -> m' ControlNode
- outfeedEnqueueTuple :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => TensorList v'1 dtypes -> m' ControlNode
- outfeedEnqueueTuple' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> TensorList v'1 dtypes -> m' ControlNode
- pack :: forall v'1 t. TensorType t => [Tensor v'1 t] -> Tensor Build t
- pack' :: forall v'1 t. TensorType t => OpParams -> [Tensor v'1 t] -> Tensor Build t
- pad :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
- pad' :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => OpParams -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
- padV2 :: forall v'1 v'2 v'3 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor v'3 t -> Tensor Build t
- padV2' :: forall v'1 v'2 v'3 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => OpParams -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor v'3 t -> Tensor Build t
- paddedBatchDataset :: forall v'1 v'2 v'3 v'4 toutput_types. TensorTypes toutput_types => Tensor v'1 Variant -> Tensor v'2 Int64 -> [Tensor v'3 Int64] -> TensorList v'4 toutput_types -> Tensor Build Variant
- paddedBatchDataset' :: forall v'1 v'2 v'3 v'4 toutput_types. TensorTypes toutput_types => OpParams -> Tensor v'1 Variant -> Tensor v'2 Int64 -> [Tensor v'3 Int64] -> TensorList v'4 toutput_types -> Tensor Build Variant
- paddingFIFOQueue :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Ref ByteString)
- paddingFIFOQueue' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Ref ByteString)
- paddingFIFOQueueV2 :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle)
- paddingFIFOQueueV2' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
- parallelConcat :: forall v'1 t. TensorType t => Shape -> [Tensor v'1 t] -> Tensor Build t
- parallelConcat' :: forall v'1 t. TensorType t => OpParams -> Shape -> [Tensor v'1 t] -> Tensor Build t
- parallelDynamicStitch :: forall v'1 v'2 t. TensorType t => [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
- parallelDynamicStitch' :: forall v'1 v'2 t. TensorType t => OpParams -> [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
- parameterizedTruncatedNormal :: forall v'1 v'2 v'3 v'4 v'5 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t) => Tensor v'1 t -> Tensor v'2 dtype -> Tensor v'3 dtype -> Tensor v'4 dtype -> Tensor v'5 dtype -> m' (Tensor Value dtype)
- parameterizedTruncatedNormal' :: forall v'1 v'2 v'3 v'4 v'5 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t) => OpParams -> Tensor v'1 t -> Tensor v'2 dtype -> Tensor v'3 dtype -> Tensor v'4 dtype -> Tensor v'5 dtype -> m' (Tensor Value dtype)
- parseExample :: forall v'1 v'2 v'3 v'4 v'5 sparse_types tdense. (OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] tdense) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> [Tensor v'3 ByteString] -> [Tensor v'4 ByteString] -> TensorList v'5 tdense -> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense)
- parseExample' :: forall v'1 v'2 v'3 v'4 v'5 sparse_types tdense. (OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] tdense) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> [Tensor v'3 ByteString] -> [Tensor v'4 ByteString] -> TensorList v'5 tdense -> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense)
- parseSingleExample :: forall v'1 v'2 sparse_types tdense. (OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] tdense) => Int64 -> Tensor v'1 ByteString -> TensorList v'2 tdense -> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense)
- parseSingleExample' :: forall v'1 v'2 sparse_types tdense. (OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] tdense) => OpParams -> Int64 -> Tensor v'1 ByteString -> TensorList v'2 tdense -> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense)
- parseSingleSequenceExample :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 context_sparse_types tcontext_dense feature_list_dense_types feature_list_sparse_types. (OneOfs '[ByteString, Int64, Float] context_sparse_types, OneOfs '[ByteString, Int64, Float] tcontext_dense, OneOfs '[ByteString, Int64, Float] feature_list_dense_types, OneOfs '[ByteString, Int64, Float] feature_list_sparse_types) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> [Tensor v'3 ByteString] -> [Tensor v'4 ByteString] -> [Tensor v'5 ByteString] -> [Tensor v'6 ByteString] -> TensorList v'7 tcontext_dense -> Tensor v'8 ByteString -> ([Tensor Build Int64], TensorList Build context_sparse_types, [Tensor Build Int64], TensorList Build tcontext_dense, [Tensor Build Int64], TensorList Build feature_list_sparse_types, [Tensor Build Int64], TensorList Build feature_list_dense_types)
- parseSingleSequenceExample' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 context_sparse_types tcontext_dense feature_list_dense_types feature_list_sparse_types. (OneOfs '[ByteString, Int64, Float] context_sparse_types, OneOfs '[ByteString, Int64, Float] tcontext_dense, OneOfs '[ByteString, Int64, Float] feature_list_dense_types, OneOfs '[ByteString, Int64, Float] feature_list_sparse_types) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> [Tensor v'3 ByteString] -> [Tensor v'4 ByteString] -> [Tensor v'5 ByteString] -> [Tensor v'6 ByteString] -> TensorList v'7 tcontext_dense -> Tensor v'8 ByteString -> ([Tensor Build Int64], TensorList Build context_sparse_types, [Tensor Build Int64], TensorList Build tcontext_dense, [Tensor Build Int64], TensorList Build feature_list_sparse_types, [Tensor Build Int64], TensorList Build feature_list_dense_types)
- parseTensor :: forall v'1 out_type. TensorType out_type => Tensor v'1 ByteString -> Tensor Build out_type
- parseTensor' :: forall v'1 out_type. TensorType out_type => OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
- placeholder :: forall dtype. TensorType dtype => Tensor Build dtype
- placeholder' :: forall dtype. TensorType dtype => OpParams -> Tensor Build dtype
- placeholderV2 :: forall dtype. TensorType dtype => Shape -> Tensor Build dtype
- placeholderV2' :: forall dtype. TensorType dtype => OpParams -> Shape -> Tensor Build dtype
- placeholderWithDefault :: forall v'1 dtype. TensorType dtype => Shape -> Tensor v'1 dtype -> Tensor Build dtype
- placeholderWithDefault' :: forall v'1 dtype. TensorType dtype => OpParams -> Shape -> Tensor v'1 dtype -> Tensor Build dtype
- polygamma :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- polygamma' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- populationCount :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor Build Word8
- populationCount' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => OpParams -> Tensor v'1 t -> Tensor Build Word8
- pow :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- pow' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- prefetchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
- prefetchDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
- prependFromQueueAndPaddedBatchDataset :: forall v'1 v'2 v'3 v'4 toutput_types. TensorTypes toutput_types => Tensor v'1 Variant -> Tensor v'2 Int64 -> [Tensor v'3 Int64] -> TensorList v'4 toutput_types -> Tensor Build Variant
- prependFromQueueAndPaddedBatchDataset' :: forall v'1 v'2 v'3 v'4 toutput_types. TensorTypes toutput_types => OpParams -> Tensor v'1 Variant -> Tensor v'2 Int64 -> [Tensor v'3 Int64] -> TensorList v'4 toutput_types -> Tensor Build Variant
- preventGradient :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- preventGradient' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- print :: forall v'1 v'2 t u m'. (MonadBuild m', TensorType t, TensorTypes u) => Tensor v'1 t -> TensorList v'2 u -> m' (Tensor Value t)
- print' :: forall v'1 v'2 t u m'. (MonadBuild m', TensorType t, TensorTypes u) => OpParams -> Tensor v'1 t -> TensorList v'2 u -> m' (Tensor Value t)
- priorityQueue :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
- priorityQueue' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
- priorityQueueV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
- priorityQueueV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
- prod :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- prod' :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- qr :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t)
- qr' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
- quantizeAndDequantize :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- quantizeAndDequantize' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- quantizeAndDequantizeV2 :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- quantizeAndDequantizeV2' :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- quantizeAndDequantizeV3 :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 Int32 -> Tensor Build t
- quantizeAndDequantizeV3' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 Int32 -> Tensor Build t
- quantizeDownAndShrinkRange :: forall v'1 v'2 v'3 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizeDownAndShrinkRange' :: forall v'1 v'2 v'3 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizeV2 :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizeV2' :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedAdd :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2 toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) => Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
- quantizedAdd' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2 toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) => OpParams -> Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
- quantizedAvgPool :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedAvgPool' :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedBatchNormWithGlobalNormalization :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12 v'13 v'14 v'15 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => Bool -> Float -> Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 tinput -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 tinput -> Tensor v'8 Float -> Tensor v'9 Float -> Tensor v'10 tinput -> Tensor v'11 Float -> Tensor v'12 Float -> Tensor v'13 tinput -> Tensor v'14 Float -> Tensor v'15 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedBatchNormWithGlobalNormalization' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12 v'13 v'14 v'15 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> Bool -> Float -> Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 tinput -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 tinput -> Tensor v'8 Float -> Tensor v'9 Float -> Tensor v'10 tinput -> Tensor v'11 Float -> Tensor v'12 Float -> Tensor v'13 tinput -> Tensor v'14 Float -> Tensor v'15 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedBiasAdd :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2 out_type. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] out_type) => Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedBiasAdd' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2 out_type. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedConcat :: forall v'1 v'2 v'3 v'4 t. TensorType t => Tensor v'1 Int32 -> [Tensor v'2 t] -> [Tensor v'3 Float] -> [Tensor v'4 Float] -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedConcat' :: forall v'1 v'2 v'3 v'4 t. TensorType t => OpParams -> Tensor v'1 Int32 -> [Tensor v'2 t] -> [Tensor v'3 Float] -> [Tensor v'4 Float] -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedConv2D :: forall v'1 v'2 v'3 v'4 v'5 v'6 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedConv2D' :: forall v'1 v'2 v'3 v'4 v'5 v'6 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedInstanceNorm :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedInstanceNorm' :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedMatMul :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2 toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) => Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
- quantizedMatMul' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2 toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) => OpParams -> Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
- quantizedMaxPool :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedMaxPool' :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedMul :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2 toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) => Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
- quantizedMul' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2 toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) => OpParams -> Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
- quantizedRelu :: forall v'1 v'2 v'3 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedRelu' :: forall v'1 v'2 v'3 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedRelu6 :: forall v'1 v'2 v'3 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedRelu6' :: forall v'1 v'2 v'3 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedReluX :: forall v'1 v'2 v'3 v'4 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedReluX' :: forall v'1 v'2 v'3 v'4 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- quantizedReshape :: forall v'1 v'2 v'3 v'4 t tshape. (TensorType t, OneOf '[Int32, Int64] tshape) => Tensor v'1 t -> Tensor v'2 tshape -> Tensor v'3 Float -> Tensor v'4 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedReshape' :: forall v'1 v'2 v'3 v'4 t tshape. (TensorType t, OneOf '[Int32, Int64] tshape) => OpParams -> Tensor v'1 t -> Tensor v'2 tshape -> Tensor v'3 Float -> Tensor v'4 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedResizeBilinear :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Word8, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor v'4 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- quantizedResizeBilinear' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Word8, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor v'4 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
- queueClose :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' ControlNode
- queueClose' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' ControlNode
- queueCloseV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
- queueCloseV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
- queueDequeue :: forall component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor Ref ByteString -> m' (TensorList Value component_types)
- queueDequeue' :: forall component_types m'. (MonadBuild m', TensorTypes component_types) => OpParams -> Tensor Ref ByteString -> m' (TensorList Value component_types)
- queueDequeueMany :: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
- queueDequeueMany' :: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
- queueDequeueManyV2 :: forall v'1 v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
- queueDequeueManyV2' :: forall v'1 v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
- queueDequeueUpTo :: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
- queueDequeueUpTo' :: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
- queueDequeueUpToV2 :: forall v'1 v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
- queueDequeueUpToV2' :: forall v'1 v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
- queueDequeueV2 :: forall v'1 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor v'1 ResourceHandle -> m' (TensorList Value component_types)
- queueDequeueV2' :: forall v'1 component_types m'. (MonadBuild m', TensorTypes component_types) => OpParams -> Tensor v'1 ResourceHandle -> m' (TensorList Value component_types)
- queueEnqueue :: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => Tensor Ref ByteString -> TensorList v'2 tcomponents -> m' ControlNode
- queueEnqueue' :: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => OpParams -> Tensor Ref ByteString -> TensorList v'2 tcomponents -> m' ControlNode
- queueEnqueueMany :: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => Tensor Ref ByteString -> TensorList v'2 tcomponents -> m' ControlNode
- queueEnqueueMany' :: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => OpParams -> Tensor Ref ByteString -> TensorList v'2 tcomponents -> m' ControlNode
- queueEnqueueManyV2 :: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => Tensor v'1 ResourceHandle -> TensorList v'2 tcomponents -> m' ControlNode
- queueEnqueueManyV2' :: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => OpParams -> Tensor v'1 ResourceHandle -> TensorList v'2 tcomponents -> m' ControlNode
- queueEnqueueV2 :: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => Tensor v'1 ResourceHandle -> TensorList v'2 tcomponents -> m' ControlNode
- queueEnqueueV2' :: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => OpParams -> Tensor v'1 ResourceHandle -> TensorList v'2 tcomponents -> m' ControlNode
- queueIsClosed :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Bool)
- queueIsClosed' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Bool)
- queueIsClosedV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
- queueIsClosedV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
- queueSize :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int32)
- queueSize' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
- queueSizeV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
- queueSizeV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
- rFFT :: Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build (Complex Float)
- rFFT' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build (Complex Float)
- rFFT2D :: Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build (Complex Float)
- rFFT2D' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build (Complex Float)
- rFFT3D :: Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build (Complex Float)
- rFFT3D' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build (Complex Float)
- rGBToHSV :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- rGBToHSV' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- randomCrop :: forall v'1 v'2 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t) => Tensor v'1 t -> Tensor v'2 Int64 -> m' (Tensor Value t)
- randomCrop' :: forall v'1 v'2 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t) => OpParams -> Tensor v'1 t -> Tensor v'2 Int64 -> m' (Tensor Value t)
- randomDataset :: forall v'1 v'2 m'. MonadBuild m' => [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value Variant)
- randomDataset' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value Variant)
- randomGamma :: forall v'1 v'2 s t m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Word16, Double, Float] t) => Tensor v'1 s -> Tensor v'2 t -> m' (Tensor Value t)
- randomGamma' :: forall v'1 v'2 s t m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Word16, Double, Float] t) => OpParams -> Tensor v'1 s -> Tensor v'2 t -> m' (Tensor Value t)
- randomPoisson :: forall v'1 v'2 s dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Word16, Double, Float] dtype) => Tensor v'1 s -> Tensor v'2 dtype -> m' (Tensor Value dtype)
- randomPoisson' :: forall v'1 v'2 s dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Word16, Double, Float] dtype) => OpParams -> Tensor v'1 s -> Tensor v'2 dtype -> m' (Tensor Value dtype)
- randomPoissonV2 :: forall v'1 v'2 s r dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Int32, Int64, Word16, Double, Float] r, OneOf '[Int32, Int64, Word16, Double, Float] dtype) => Tensor v'1 s -> Tensor v'2 r -> m' (Tensor Value dtype)
- randomPoissonV2' :: forall v'1 v'2 s r dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Int32, Int64, Word16, Double, Float] r, OneOf '[Int32, Int64, Word16, Double, Float] dtype) => OpParams -> Tensor v'1 s -> Tensor v'2 r -> m' (Tensor Value dtype)
- randomShuffle :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value t)
- randomShuffle' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value t)
- randomShuffleQueue :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Ref ByteString)
- randomShuffleQueue' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Ref ByteString)
- randomShuffleQueueV2 :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle)
- randomShuffleQueueV2' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
- randomStandardNormal :: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t) => Tensor v'1 t -> m' (Tensor Value dtype)
- randomStandardNormal' :: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t) => OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
- randomUniform :: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t) => Tensor v'1 t -> m' (Tensor Value dtype)
- randomUniform' :: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t) => OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
- randomUniformInt :: forall v'1 v'2 v'3 tout t m'. (MonadBuild m', OneOf '[Int32, Int64] tout, OneOf '[Int32, Int64] t) => Tensor v'1 t -> Tensor v'2 tout -> Tensor v'3 tout -> m' (Tensor Value tout)
- randomUniformInt' :: forall v'1 v'2 v'3 tout t m'. (MonadBuild m', OneOf '[Int32, Int64] tout, OneOf '[Int32, Int64] t) => OpParams -> Tensor v'1 t -> Tensor v'2 tout -> Tensor v'3 tout -> m' (Tensor Value tout)
- range :: forall v'1 v'2 v'3 tidx. OneOf '[Int32, Int64, Word16, Double, Float] tidx => Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor v'3 tidx -> Tensor Build tidx
- range' :: forall v'1 v'2 v'3 tidx. OneOf '[Int32, Int64, Word16, Double, Float] tidx => OpParams -> Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor v'3 tidx -> Tensor Build tidx
- rangeDataset :: forall v'1 v'2 v'3 m'. MonadBuild m' => [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
- rangeDataset' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
- rank :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build Int32
- rank' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build Int32
- readFile :: Tensor v'1 ByteString -> Tensor Build ByteString
- readFile' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
- readVariableOp :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> m' (Tensor Value dtype)
- readVariableOp' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value dtype)
- readerNumRecordsProduced :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int64)
- readerNumRecordsProduced' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
- readerNumRecordsProducedV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
- readerNumRecordsProducedV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
- readerNumWorkUnitsCompleted :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int64)
- readerNumWorkUnitsCompleted' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
- readerNumWorkUnitsCompletedV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
- readerNumWorkUnitsCompletedV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
- readerRead :: forall m'. MonadBuild m' => Tensor Ref ByteString -> Tensor Ref ByteString -> m' (Tensor Value ByteString, Tensor Value ByteString)
- readerRead' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> Tensor Ref ByteString -> m' (Tensor Value ByteString, Tensor Value ByteString)
- readerReadUpTo :: forall v'3 m'. MonadBuild m' => Tensor Ref ByteString -> Tensor Ref ByteString -> Tensor v'3 Int64 -> m' (Tensor Value ByteString, Tensor Value ByteString)
- readerReadUpTo' :: forall v'3 m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> Tensor Ref ByteString -> Tensor v'3 Int64 -> m' (Tensor Value ByteString, Tensor Value ByteString)
- readerReadUpToV2 :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 Int64 -> m' (Tensor Value ByteString, Tensor Value ByteString)
- readerReadUpToV2' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 Int64 -> m' (Tensor Value ByteString, Tensor Value ByteString)
- readerReadV2 :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> m' (Tensor Value ByteString, Tensor Value ByteString)
- readerReadV2' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> m' (Tensor Value ByteString, Tensor Value ByteString)
- readerReset :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' ControlNode
- readerReset' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' ControlNode
- readerResetV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
- readerResetV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
- readerRestoreState :: forall v'2 m'. MonadBuild m' => Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
- readerRestoreState' :: forall v'2 m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
- readerRestoreStateV2 :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
- readerRestoreStateV2' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
- readerSerializeState :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value ByteString)
- readerSerializeState' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value ByteString)
- readerSerializeStateV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
- readerSerializeStateV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
- real :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => Tensor v'1 t -> Tensor Build tout
- real' :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => OpParams -> Tensor v'1 t -> Tensor Build tout
- realDiv :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- realDiv' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- reciprocal :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- reciprocal' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- reciprocalGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- reciprocalGrad' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- recordInput :: forall m'. MonadBuild m' => m' (Tensor Value ByteString)
- recordInput' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ByteString)
- reduceJoin :: Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build ByteString
- reduceJoin' :: OpParams -> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build ByteString
- refEnter :: forall t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> m' (Tensor Ref t)
- refEnter' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> m' (Tensor Ref t)
- refExit :: forall t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> m' (Tensor Ref t)
- refExit' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> m' (Tensor Ref t)
- refIdentity :: forall t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> m' (Tensor Ref t)
- refIdentity' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> m' (Tensor Ref t)
- refMerge :: forall t m'. (MonadBuild m', TensorType t) => [Tensor Ref t] -> m' (Tensor Ref t, Tensor Value Int32)
- refMerge' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> [Tensor Ref t] -> m' (Tensor Ref t, Tensor Value Int32)
- refNextIteration :: forall t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> m' (Tensor Ref t)
- refNextIteration' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> m' (Tensor Ref t)
- refSelect :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 Int32 -> [Tensor Ref t] -> m' (Tensor Ref t)
- refSelect' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 Int32 -> [Tensor Ref t] -> m' (Tensor Ref t)
- refSwitch :: forall v'2 t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> Tensor v'2 Bool -> m' (Tensor Ref t, Tensor Ref t)
- refSwitch' :: forall v'2 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> Tensor v'2 Bool -> m' (Tensor Ref t, Tensor Ref t)
- regexFullMatch :: Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor Build Bool
- regexFullMatch' :: OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor Build Bool
- regexReplace :: Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> Tensor Build ByteString
- regexReplace' :: OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> Tensor Build ByteString
- relu :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
- relu' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- relu6 :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
- relu6' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- relu6Grad :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- relu6Grad' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- reluGrad :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- reluGrad' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- remoteFusedGraphExecute :: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs) => TensorList v'1 tinputs -> TensorList Build toutputs
- remoteFusedGraphExecute' :: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs) => OpParams -> TensorList v'1 tinputs -> TensorList Build toutputs
- repeatDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
- repeatDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
- requantizationRange :: forall v'1 v'2 v'3 tinput. OneOf '[Int16, Int32, Word16, Word8] tinput => Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build Float, Tensor Build Float)
- requantizationRange' :: forall v'1 v'2 v'3 tinput. OneOf '[Int16, Int32, Word16, Word8] tinput => OpParams -> Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build Float, Tensor Build Float)
- requantize :: forall v'1 v'2 v'3 v'4 v'5 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- requantize' :: forall v'1 v'2 v'3 v'4 v'5 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
- reshape :: forall v'1 v'2 t tshape. (TensorType t, OneOf '[Int32, Int64] tshape) => Tensor v'1 t -> Tensor v'2 tshape -> Tensor Build t
- reshape' :: forall v'1 v'2 t tshape. (TensorType t, OneOf '[Int32, Int64] tshape) => OpParams -> Tensor v'1 t -> Tensor v'2 tshape -> Tensor Build t
- resizeArea :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
- resizeArea' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
- resizeBicubic :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
- resizeBicubic' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
- resizeBicubicGrad :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
- resizeBicubicGrad' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
- resizeBilinear :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
- resizeBilinear' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
- resizeBilinearGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
- resizeBilinearGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
- resizeNearestNeighbor :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
- resizeNearestNeighbor' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
- resizeNearestNeighborGrad :: forall v'1 v'2 t. OneOf '[Int32, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
- resizeNearestNeighborGrad' :: forall v'1 v'2 t. OneOf '[Int32, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
- resourceApplyAdaMax :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' ControlNode
- resourceApplyAdaMax' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' ControlNode
- resourceApplyAdadelta :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' ControlNode
- resourceApplyAdadelta' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' ControlNode
- resourceApplyAdagrad :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> m' ControlNode
- resourceApplyAdagrad' :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> m' ControlNode
- resourceApplyAdagradDA :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 Int64 -> m' ControlNode
- resourceApplyAdagradDA' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 Int64 -> m' ControlNode
- resourceApplyAdam :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> m' ControlNode
- resourceApplyAdam' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> m' ControlNode
- resourceApplyAddSign :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' ControlNode
- resourceApplyAddSign' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' ControlNode
- resourceApplyCenteredRMSProp :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 ResourceHandle -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' ControlNode
- resourceApplyCenteredRMSProp' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 ResourceHandle -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' ControlNode
- resourceApplyFtrl :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> m' ControlNode
- resourceApplyFtrl' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> m' ControlNode
- resourceApplyFtrlV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' ControlNode
- resourceApplyFtrlV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' ControlNode
- resourceApplyGradientDescent :: forall v'1 v'2 v'3 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 t -> Tensor v'3 t -> m' ControlNode
- resourceApplyGradientDescent' :: forall v'1 v'2 v'3 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 t -> Tensor v'3 t -> m' ControlNode
- resourceApplyMomentum :: forall v'1 v'2 v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' ControlNode
- resourceApplyMomentum' :: forall v'1 v'2 v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' ControlNode
- resourceApplyPowerSign :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' ControlNode
- resourceApplyPowerSign' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> m' ControlNode
- resourceApplyProximalAdagrad :: forall v'1 v'2 v'3 v'4 v'5 v'6 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> m' ControlNode
- resourceApplyProximalAdagrad' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> m' ControlNode
- resourceApplyProximalGradientDescent :: forall v'1 v'2 v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' ControlNode
- resourceApplyProximalGradientDescent' :: forall v'1 v'2 v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' ControlNode
- resourceApplyRMSProp :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> m' ControlNode
- resourceApplyRMSProp' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> m' ControlNode
- resourceCountUpTo :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64] t) => Int64 -> Tensor v'1 ResourceHandle -> m' (Tensor Value t)
- resourceCountUpTo' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64] t) => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> m' (Tensor Value t)
- resourceGather :: forall v'1 v'2 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> m' (Tensor Value dtype)
- resourceGather' :: forall v'1 v'2 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> m' (Tensor Value dtype)
- resourceScatterAdd :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterAdd' :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterDiv :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterDiv' :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterMax :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterMax' :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterMin :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterMin' :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterMul :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterMul' :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterNdUpdate :: forall v'1 v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
- resourceScatterNdUpdate' :: forall v'1 v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
- resourceScatterSub :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterSub' :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterUpdate :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceScatterUpdate' :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
- resourceSparseApplyAdadelta :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 tindices -> m' ControlNode
- resourceSparseApplyAdadelta' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 tindices -> m' ControlNode
- resourceSparseApplyAdagrad :: forall v'1 v'2 v'3 v'4 v'5 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 tindices -> m' ControlNode
- resourceSparseApplyAdagrad' :: forall v'1 v'2 v'3 v'4 v'5 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 tindices -> m' ControlNode
- resourceSparseApplyAdagradDA :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 Int64 -> m' ControlNode
- resourceSparseApplyAdagradDA' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 Int64 -> m' ControlNode
- resourceSparseApplyCenteredRMSProp :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 ResourceHandle -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 tindices -> m' ControlNode
- resourceSparseApplyCenteredRMSProp' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 ResourceHandle -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 tindices -> m' ControlNode
- resourceSparseApplyFtrl :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' ControlNode
- resourceSparseApplyFtrl' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' ControlNode
- resourceSparseApplyFtrlV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> m' ControlNode
- resourceSparseApplyFtrlV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> m' ControlNode
- resourceSparseApplyMomentum :: forall v'1 v'2 v'3 v'4 v'5 v'6 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> m' ControlNode
- resourceSparseApplyMomentum' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> m' ControlNode
- resourceSparseApplyProximalAdagrad :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 tindices -> m' ControlNode
- resourceSparseApplyProximalAdagrad' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 tindices -> m' ControlNode
- resourceSparseApplyProximalGradientDescent :: forall v'1 v'2 v'3 v'4 v'5 v'6 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 tindices -> m' ControlNode
- resourceSparseApplyProximalGradientDescent' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 tindices -> m' ControlNode
- resourceSparseApplyRMSProp :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 tindices -> m' ControlNode
- resourceSparseApplyRMSProp' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 ResourceHandle -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 tindices -> m' ControlNode
- resourceStridedSliceAssign :: forall v'1 v'2 v'3 v'4 v'5 t index m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] index) => Tensor v'1 ResourceHandle -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor v'5 t -> m' ControlNode
- resourceStridedSliceAssign' :: forall v'1 v'2 v'3 v'4 v'5 t index m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] index) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor v'5 t -> m' ControlNode
- restore :: forall v'1 v'2 dt m'. (MonadBuild m', TensorType dt) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' (Tensor Value dt)
- restore' :: forall v'1 v'2 dt m'. (MonadBuild m', TensorType dt) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' (Tensor Value dt)
- restoreSlice :: forall v'1 v'2 v'3 dt m'. (MonadBuild m', TensorType dt) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value dt)
- restoreSlice' :: forall v'1 v'2 v'3 dt m'. (MonadBuild m', TensorType dt) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value dt)
- restoreV2 :: forall v'1 v'2 v'3 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (TensorList Value dtypes)
- restoreV2' :: forall v'1 v'2 v'3 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (TensorList Value dtypes)
- reverse :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Bool -> Tensor Build t
- reverse' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Bool -> Tensor Build t
- reverseSequence :: forall v'1 v'2 t tlen. (TensorType t, OneOf '[Int32, Int64] tlen) => Int64 -> Tensor v'1 t -> Tensor v'2 tlen -> Tensor Build t
- reverseSequence' :: forall v'1 v'2 t tlen. (TensorType t, OneOf '[Int32, Int64] tlen) => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 tlen -> Tensor Build t
- reverseV2 :: forall v'1 v'2 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- reverseV2' :: forall v'1 v'2 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- rightShift :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- rightShift' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- rint :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- rint' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- roll :: forall v'1 v'2 v'3 t tshift taxis. (TensorType t, OneOf '[Int32, Int64] tshift, OneOf '[Int32, Int64] taxis) => Tensor v'1 t -> Tensor v'2 tshift -> Tensor v'3 taxis -> Tensor Build t
- roll' :: forall v'1 v'2 v'3 t tshift taxis. (TensorType t, OneOf '[Int32, Int64] tshift, OneOf '[Int32, Int64] taxis) => OpParams -> Tensor v'1 t -> Tensor v'2 tshift -> Tensor v'3 taxis -> Tensor Build t
- round :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- round' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- rpc :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value ByteString)
- rpc' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value ByteString)
- rsqrt :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- rsqrt' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- rsqrtGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- rsqrtGrad' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- sampleDistortedBoundingBox :: forall v'1 v'2 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t) => Tensor v'1 t -> Tensor v'2 Float -> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
- sampleDistortedBoundingBox' :: forall v'1 v'2 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t) => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
- sampleDistortedBoundingBoxV2 :: forall v'1 v'2 v'3 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t) => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
- sampleDistortedBoundingBoxV2' :: forall v'1 v'2 v'3 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t) => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
- save :: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorTypes t) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> TensorList v'3 t -> m' ControlNode
- save' :: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorTypes t) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> TensorList v'3 t -> m' ControlNode
- saveSlices :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorTypes t) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> TensorList v'4 t -> m' ControlNode
- saveSlices' :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorTypes t) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> TensorList v'4 t -> m' ControlNode
- saveV2 :: forall v'1 v'2 v'3 v'4 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> TensorList v'4 dtypes -> m' ControlNode
- saveV2' :: forall v'1 v'2 v'3 v'4 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> TensorList v'4 dtypes -> m' ControlNode
- scalarSummary :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
- scalarSummary' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
- scatterAdd :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterAdd' :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterDiv :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterDiv' :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterMax :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterMax' :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterMin :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterMin' :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterMul :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterMul' :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterNd :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 tindices -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor Build t
- scatterNd' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 tindices -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor Build t
- scatterNdAdd :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterNdAdd' :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterNdNonAliasingAdd :: forall v'1 v'2 v'3 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
- scatterNdNonAliasingAdd' :: forall v'1 v'2 v'3 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
- scatterNdSub :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterNdSub' :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterNdUpdate :: forall v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterNdUpdate' :: forall v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterSub :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterSub' :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterUpdate :: forall v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- scatterUpdate' :: forall v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
- sdcaFprint :: Tensor v'1 ByteString -> Tensor Build Int64
- sdcaFprint' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Int64
- sdcaOptimizer :: Float -> Float -> Int64 -> Int64 -> [Tensor v'1 Int64] -> [Tensor v'2 Int64] -> [Tensor v'3 Float] -> [Tensor v'4 Float] -> Tensor v'5 Float -> Tensor v'6 Float -> [Tensor v'7 Int64] -> [Tensor v'8 Float] -> [Tensor v'9 Float] -> Tensor v'10 Float -> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
- sdcaOptimizer' :: OpParams -> Float -> Float -> Int64 -> Int64 -> [Tensor v'1 Int64] -> [Tensor v'2 Int64] -> [Tensor v'3 Float] -> [Tensor v'4 Float] -> Tensor v'5 Float -> Tensor v'6 Float -> [Tensor v'7 Int64] -> [Tensor v'8 Float] -> [Tensor v'9 Float] -> Tensor v'10 Float -> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
- sdcaShrinkL1 :: forall m'. MonadBuild m' => Float -> Float -> [Tensor Ref Float] -> m' ControlNode
- sdcaShrinkL1' :: forall m'. MonadBuild m' => OpParams -> Float -> Float -> [Tensor Ref Float] -> m' ControlNode
- segmentMax :: forall v'1 v'2 t tindices. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
- segmentMax' :: forall v'1 v'2 t tindices. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
- segmentMean :: forall v'1 v'2 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
- segmentMean' :: forall v'1 v'2 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
- segmentMin :: forall v'1 v'2 t tindices. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
- segmentMin' :: forall v'1 v'2 t tindices. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
- segmentProd :: forall v'1 v'2 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
- segmentProd' :: forall v'1 v'2 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
- segmentSum :: forall v'1 v'2 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
- segmentSum' :: forall v'1 v'2 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
- select :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 Bool -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- select' :: forall v'1 v'2 v'3 t. TensorType t => OpParams -> Tensor v'1 Bool -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
- selfAdjointEig :: forall v'1 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor Build t
- selfAdjointEig' :: forall v'1 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- selfAdjointEigV2 :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t)
- selfAdjointEigV2' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
- selu :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- selu' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- seluGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- seluGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- serializeIterator :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
- serializeIterator' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
- serializeManySparse :: forall v'1 v'2 v'3 t out_type. (TensorType t, OneOf '[ByteString, Variant] out_type) => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build out_type
- serializeManySparse' :: forall v'1 v'2 v'3 t out_type. (TensorType t, OneOf '[ByteString, Variant] out_type) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build out_type
- serializeSparse :: forall v'1 v'2 v'3 t out_type. (TensorType t, OneOf '[ByteString, Variant] out_type) => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build out_type
- serializeSparse' :: forall v'1 v'2 v'3 t out_type. (TensorType t, OneOf '[ByteString, Variant] out_type) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build out_type
- serializeTensor :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build ByteString
- serializeTensor' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build ByteString
- setSize :: forall v'1 v'2 v'3 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build Int32
- setSize' :: forall v'1 v'2 v'3 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build Int32
- setStatsAggregatorDataset :: forall v'1 v'2 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' (Tensor Value Variant)
- setStatsAggregatorDataset' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' (Tensor Value Variant)
- shape :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => Tensor v'1 t -> Tensor Build out_type
- shape' :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => OpParams -> Tensor v'1 t -> Tensor Build out_type
- shapeN :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => [Tensor v'1 t] -> [Tensor Build out_type]
- shapeN' :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => OpParams -> [Tensor v'1 t] -> [Tensor Build out_type]
- shardedFilename :: Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build ByteString
- shardedFilename' :: OpParams -> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build ByteString
- shardedFilespec :: Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build ByteString
- shardedFilespec' :: OpParams -> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build ByteString
- shuffleAndRepeatDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> Tensor Build Variant
- shuffleAndRepeatDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> Tensor Build Variant
- shuffleDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build Variant
- shuffleDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build Variant
- shutdownDistributedTPU :: forall m'. MonadBuild m' => m' ControlNode
- shutdownDistributedTPU' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode
- sigmoid :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- sigmoid' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- sigmoidGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- sigmoidGrad' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- sign :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- sign' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- sin :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- sin' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- sinh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- sinh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- size :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => Tensor v'1 t -> Tensor Build out_type
- size' :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => OpParams -> Tensor v'1 t -> Tensor Build out_type
- skipDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
- skipDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
- skipgram :: forall m'. MonadBuild m' => Int64 -> m' (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64, Tensor Value Int32, Tensor Value Int64, Tensor Value Int32, Tensor Value Int32)
- skipgram' :: forall m'. MonadBuild m' => OpParams -> Int64 -> m' (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64, Tensor Value Int32, Tensor Value Int64, Tensor Value Int32, Tensor Value Int32)
- slice :: forall v'1 v'2 v'3 t index. (TensorType t, OneOf '[Int32, Int64] index) => Tensor v'1 t -> Tensor v'2 index -> Tensor v'3 index -> Tensor Build t
- slice' :: forall v'1 v'2 v'3 t index. (TensorType t, OneOf '[Int32, Int64] index) => OpParams -> Tensor v'1 t -> Tensor v'2 index -> Tensor v'3 index -> Tensor Build t
- slideDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
- slideDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
- snapshot :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- snapshot' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- softmax :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- softmax' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- softmaxCrossEntropyWithLogits :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
- softmaxCrossEntropyWithLogits' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
- softplus :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
- softplus' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- softplusGrad :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- softplusGrad' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- softsign :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
- softsign' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- softsignGrad :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- softsignGrad' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- spaceToBatch :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => Int64 -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
- spaceToBatch' :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
- spaceToBatchND :: forall v'1 v'2 v'3 t tblock_shape tpaddings. (TensorType t, OneOf '[Int32, Int64] tblock_shape, OneOf '[Int32, Int64] tpaddings) => Tensor v'1 t -> Tensor v'2 tblock_shape -> Tensor v'3 tpaddings -> Tensor Build t
- spaceToBatchND' :: forall v'1 v'2 v'3 t tblock_shape tpaddings. (TensorType t, OneOf '[Int32, Int64] tblock_shape, OneOf '[Int32, Int64] tpaddings) => OpParams -> Tensor v'1 t -> Tensor v'2 tblock_shape -> Tensor v'3 tpaddings -> Tensor Build t
- spaceToDepth :: forall v'1 t. TensorType t => Int64 -> Tensor v'1 t -> Tensor Build t
- spaceToDepth' :: forall v'1 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> Tensor Build t
- sparseAccumulatorApplyGradient :: forall v'2 v'3 v'4 v'5 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => Bool -> Tensor Ref ByteString -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 dtype -> Tensor v'5 Int64 -> m' ControlNode
- sparseAccumulatorApplyGradient' :: forall v'2 v'3 v'4 v'5 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => OpParams -> Bool -> Tensor Ref ByteString -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 dtype -> Tensor v'5 Int64 -> m' ControlNode
- sparseAccumulatorTakeGradient :: forall v'2 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
- sparseAccumulatorTakeGradient' :: forall v'2 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
- sparseAdd :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t treal. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] treal) => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 t -> Tensor v'6 Int64 -> Tensor v'7 treal -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- sparseAdd' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t treal. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] treal) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 t -> Tensor v'6 Int64 -> Tensor v'7 treal -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- sparseAddGrad :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> (Tensor Build t, Tensor Build t)
- sparseAddGrad' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> (Tensor Build t, Tensor Build t)
- sparseApplyAdadelta :: forall v'4 v'5 v'6 v'7 v'8 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 tindices -> m' (Tensor Ref t)
- sparseApplyAdadelta' :: forall v'4 v'5 v'6 v'7 v'8 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 tindices -> m' (Tensor Ref t)
- sparseApplyAdagrad :: forall v'3 v'4 v'5 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 tindices -> m' (Tensor Ref t)
- sparseApplyAdagrad' :: forall v'3 v'4 v'5 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 tindices -> m' (Tensor Ref t)
- sparseApplyAdagradDA :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 Int64 -> m' (Tensor Ref t)
- sparseApplyAdagradDA' :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 Int64 -> m' (Tensor Ref t)
- sparseApplyCenteredRMSProp :: forall v'5 v'6 v'7 v'8 v'9 v'10 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 tindices -> m' (Tensor Ref t)
- sparseApplyCenteredRMSProp' :: forall v'5 v'6 v'7 v'8 v'9 v'10 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 tindices -> m' (Tensor Ref t)
- sparseApplyFtrl :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' (Tensor Ref t)
- sparseApplyFtrl' :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> m' (Tensor Ref t)
- sparseApplyFtrlV2 :: forall v'4 v'5 v'6 v'7 v'8 v'9 v'10 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> m' (Tensor Ref t)
- sparseApplyFtrlV2' :: forall v'4 v'5 v'6 v'7 v'8 v'9 v'10 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> m' (Tensor Ref t)
- sparseApplyMomentum :: forall v'3 v'4 v'5 v'6 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> m' (Tensor Ref t)
- sparseApplyMomentum' :: forall v'3 v'4 v'5 v'6 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 tindices -> Tensor v'6 t -> m' (Tensor Ref t)
- sparseApplyProximalAdagrad :: forall v'3 v'4 v'5 v'6 v'7 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 tindices -> m' (Tensor Ref t)
- sparseApplyProximalAdagrad' :: forall v'3 v'4 v'5 v'6 v'7 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 tindices -> m' (Tensor Ref t)
- sparseApplyProximalGradientDescent :: forall v'2 v'3 v'4 v'5 v'6 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 tindices -> m' (Tensor Ref t)
- sparseApplyProximalGradientDescent' :: forall v'2 v'3 v'4 v'5 v'6 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 tindices -> m' (Tensor Ref t)
- sparseApplyRMSProp :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 tindices -> m' (Tensor Ref t)
- sparseApplyRMSProp' :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor Ref t -> Tensor Ref t -> Tensor Ref t -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 tindices -> m' (Tensor Ref t)
- sparseConcat :: forall v'1 v'2 v'3 t. TensorType t => Int64 -> [Tensor v'1 Int64] -> [Tensor v'2 t] -> [Tensor v'3 Int64] -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- sparseConcat' :: forall v'1 v'2 v'3 t. TensorType t => OpParams -> Int64 -> [Tensor v'1 Int64] -> [Tensor v'2 t] -> [Tensor v'3 Int64] -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- sparseConditionalAccumulator :: forall m'. MonadBuild m' => DataType -> Shape -> m' (Tensor Ref ByteString)
- sparseConditionalAccumulator' :: forall m'. MonadBuild m' => OpParams -> DataType -> Shape -> m' (Tensor Ref ByteString)
- sparseCross :: forall v'1 v'2 v'3 v'4 sparse_types dense_types out_type. (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types, OneOf '[ByteString, Int64] out_type) => Int64 -> Bool -> DataType -> Int64 -> [Tensor v'1 Int64] -> TensorList v'2 sparse_types -> [Tensor v'3 Int64] -> TensorList v'4 dense_types -> (Tensor Build Int64, Tensor Build out_type, Tensor Build Int64)
- sparseCross' :: forall v'1 v'2 v'3 v'4 sparse_types dense_types out_type. (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types, OneOf '[ByteString, Int64] out_type) => OpParams -> Int64 -> Bool -> DataType -> Int64 -> [Tensor v'1 Int64] -> TensorList v'2 sparse_types -> [Tensor v'3 Int64] -> TensorList v'4 dense_types -> (Tensor Build Int64, Tensor Build out_type, Tensor Build Int64)
- sparseDenseCwiseAdd :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
- sparseDenseCwiseAdd' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
- sparseDenseCwiseDiv :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
- sparseDenseCwiseDiv' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
- sparseDenseCwiseMul :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
- sparseDenseCwiseMul' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
- sparseFillEmptyRows :: forall v'1 v'2 v'3 v'4 t. TensorType t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> (Tensor Build Int64, Tensor Build t, Tensor Build Bool, Tensor Build Int64)
- sparseFillEmptyRows' :: forall v'1 v'2 v'3 v'4 t. TensorType t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> (Tensor Build Int64, Tensor Build t, Tensor Build Bool, Tensor Build Int64)
- sparseFillEmptyRowsGrad :: forall v'1 v'2 t. TensorType t => Tensor v'1 Int64 -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
- sparseFillEmptyRowsGrad' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
- sparseMatMul :: forall v'1 v'2 ta tb. (OneOf '[Word16, Float] ta, OneOf '[Word16, Float] tb) => Tensor v'1 ta -> Tensor v'2 tb -> Tensor Build Float
- sparseMatMul' :: forall v'1 v'2 ta tb. (OneOf '[Word16, Float] ta, OneOf '[Word16, Float] tb) => OpParams -> Tensor v'1 ta -> Tensor v'2 tb -> Tensor Build Float
- sparseReduceMax :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int32 -> Tensor Build t
- sparseReduceMax' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int32 -> Tensor Build t
- sparseReduceMaxSparse :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int32 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- sparseReduceMaxSparse' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int32 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- sparseReduceSum :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int32 -> Tensor Build t
- sparseReduceSum' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int32 -> Tensor Build t
- sparseReduceSumSparse :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int32 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- sparseReduceSumSparse' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int32 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- sparseReorder :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> (Tensor Build Int64, Tensor Build t)
- sparseReorder' :: forall v'1 v'2 v'3 t. TensorType t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> (Tensor Build Int64, Tensor Build t)
- sparseReshape :: Tensor v'1 Int64 -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> (Tensor Build Int64, Tensor Build Int64)
- sparseReshape' :: OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> (Tensor Build Int64, Tensor Build Int64)
- sparseSegmentMean :: forall v'1 v'2 v'3 t tidx. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor Build t
- sparseSegmentMean' :: forall v'1 v'2 v'3 t tidx. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor Build t
- sparseSegmentMeanGrad :: forall v'1 v'2 v'3 v'4 t tidx. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor Build t
- sparseSegmentMeanGrad' :: forall v'1 v'2 v'3 v'4 t tidx. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor Build t
- sparseSegmentMeanWithNumSegments :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor v'4 tnumsegments -> Tensor Build t
- sparseSegmentMeanWithNumSegments' :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor v'4 tnumsegments -> Tensor Build t
- sparseSegmentSqrtN :: forall v'1 v'2 v'3 t tidx. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor Build t
- sparseSegmentSqrtN' :: forall v'1 v'2 v'3 t tidx. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor Build t
- sparseSegmentSqrtNGrad :: forall v'1 v'2 v'3 v'4 t tidx. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor Build t
- sparseSegmentSqrtNGrad' :: forall v'1 v'2 v'3 v'4 t tidx. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor Build t
- sparseSegmentSqrtNWithNumSegments :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor v'4 tnumsegments -> Tensor Build t
- sparseSegmentSqrtNWithNumSegments' :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor v'4 tnumsegments -> Tensor Build t
- sparseSegmentSum :: forall v'1 v'2 v'3 t tidx. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor Build t
- sparseSegmentSum' :: forall v'1 v'2 v'3 t tidx. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor Build t
- sparseSegmentSumWithNumSegments :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor v'4 tnumsegments -> Tensor Build t
- sparseSegmentSumWithNumSegments' :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 Int32 -> Tensor v'4 tnumsegments -> Tensor Build t
- sparseSlice :: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- sparseSlice' :: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- sparseSoftmax :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build t
- sparseSoftmax' :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build t
- sparseSoftmaxCrossEntropyWithLogits :: forall v'1 v'2 t tlabels. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tlabels) => Tensor v'1 t -> Tensor v'2 tlabels -> (Tensor Build t, Tensor Build t)
- sparseSoftmaxCrossEntropyWithLogits' :: forall v'1 v'2 t tlabels. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tlabels) => OpParams -> Tensor v'1 t -> Tensor v'2 tlabels -> (Tensor Build t, Tensor Build t)
- sparseSparseMaximum :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 t -> Tensor v'6 Int64 -> (Tensor Build Int64, Tensor Build t)
- sparseSparseMaximum' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 t -> Tensor v'6 Int64 -> (Tensor Build Int64, Tensor Build t)
- sparseSparseMinimum :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 t -> Tensor v'6 Int64 -> (Tensor Build Int64, Tensor Build t)
- sparseSparseMinimum' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 t -> Tensor v'6 Int64 -> (Tensor Build Int64, Tensor Build t)
- sparseSplit :: forall v'1 v'2 v'3 v'4 t. TensorType t => Int64 -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> Tensor v'3 t -> Tensor v'4 Int64 -> ([Tensor Build Int64], [Tensor Build t], [Tensor Build Int64])
- sparseSplit' :: forall v'1 v'2 v'3 v'4 t. TensorType t => OpParams -> Int64 -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> Tensor v'3 t -> Tensor v'4 Int64 -> ([Tensor Build Int64], [Tensor Build t], [Tensor Build Int64])
- sparseTensorDenseAdd :: forall v'1 v'2 v'3 v'4 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 tindices -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor v'4 t -> Tensor Build t
- sparseTensorDenseAdd' :: forall v'1 v'2 v'3 v'4 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 tindices -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor v'4 t -> Tensor Build t
- sparseTensorDenseMatMul :: forall v'1 v'2 v'3 v'4 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 tindices -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
- sparseTensorDenseMatMul' :: forall v'1 v'2 v'3 v'4 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 tindices -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
- sparseTensorSliceDataset :: forall v'1 v'2 v'3 tvalues m'. (MonadBuild m', TensorType tvalues) => Tensor v'1 Int64 -> Tensor v'2 tvalues -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
- sparseTensorSliceDataset' :: forall v'1 v'2 v'3 tvalues m'. (MonadBuild m', TensorType tvalues) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 tvalues -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
- sparseToDense :: forall v'1 v'2 v'3 v'4 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 tindices -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build t
- sparseToDense' :: forall v'1 v'2 v'3 v'4 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 tindices -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build t
- sparseToSparseSetOperation :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 t -> Tensor v'6 Int64 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- sparseToSparseSetOperation' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 t -> Tensor v'6 Int64 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
- split :: forall v'1 v'2 t. TensorType t => Int64 -> Tensor v'1 Int32 -> Tensor v'2 t -> [Tensor Build t]
- split' :: forall v'1 v'2 t. TensorType t => OpParams -> Int64 -> Tensor v'1 Int32 -> Tensor v'2 t -> [Tensor Build t]
- splitV :: forall v'1 v'2 v'3 t tlen. (TensorType t, OneOf '[Int32, Int64] tlen) => Int64 -> Tensor v'1 t -> Tensor v'2 tlen -> Tensor v'3 Int32 -> [Tensor Build t]
- splitV' :: forall v'1 v'2 v'3 t tlen. (TensorType t, OneOf '[Int32, Int64] tlen) => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 tlen -> Tensor v'3 Int32 -> [Tensor Build t]
- sqlDataset :: forall v'1 v'2 v'3 m'. MonadBuild m' => [DataType] -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value Variant)
- sqlDataset' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value Variant)
- sqrt :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- sqrt' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- sqrtGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- sqrtGrad' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- square :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- square' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- squaredDifference :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- squaredDifference' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- squeeze :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- squeeze' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- stack :: forall m'. MonadBuild m' => DataType -> m' (Tensor Ref ByteString)
- stack' :: forall m'. MonadBuild m' => OpParams -> DataType -> m' (Tensor Ref ByteString)
- stackClose :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' ControlNode
- stackClose' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' ControlNode
- stackCloseV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
- stackCloseV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
- stackPop :: forall elem_type m'. (MonadBuild m', TensorType elem_type) => Tensor Ref ByteString -> m' (Tensor Value elem_type)
- stackPop' :: forall elem_type m'. (MonadBuild m', TensorType elem_type) => OpParams -> Tensor Ref ByteString -> m' (Tensor Value elem_type)
- stackPopV2 :: forall v'1 elem_type m'. (MonadBuild m', TensorType elem_type) => Tensor v'1 ResourceHandle -> m' (Tensor Value elem_type)
- stackPopV2' :: forall v'1 elem_type m'. (MonadBuild m', TensorType elem_type) => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value elem_type)
- stackPush :: forall v'2 t m'. (MonadBuild m', TensorType t) => Tensor Ref ByteString -> Tensor v'2 t -> m' (Tensor Value t)
- stackPush' :: forall v'2 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref ByteString -> Tensor v'2 t -> m' (Tensor Value t)
- stackPushV2 :: forall v'1 v'2 t m'. (MonadBuild m', TensorType t) => Tensor v'1 ResourceHandle -> Tensor v'2 t -> m' (Tensor Value t)
- stackPushV2' :: forall v'1 v'2 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 t -> m' (Tensor Value t)
- stackV2 :: forall v'1 m'. MonadBuild m' => DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle)
- stackV2' :: forall v'1 m'. MonadBuild m' => OpParams -> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle)
- stage :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => TensorList v'1 dtypes -> m' ControlNode
- stage' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> TensorList v'1 dtypes -> m' ControlNode
- stageClear :: forall m'. MonadBuild m' => [DataType] -> m' ControlNode
- stageClear' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' ControlNode
- stagePeek :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int32 -> m' (TensorList Value dtypes)
- stagePeek' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int32 -> m' (TensorList Value dtypes)
- stageSize :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value Int32)
- stageSize' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value Int32)
- statelessMultinomial :: forall v'1 v'2 v'3 t tseed output_dtype. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tseed, OneOf '[Int32, Int64] output_dtype) => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 tseed -> Tensor Build output_dtype
- statelessMultinomial' :: forall v'1 v'2 v'3 t tseed output_dtype. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tseed, OneOf '[Int32, Int64] output_dtype) => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 tseed -> Tensor Build output_dtype
- statelessRandomNormal :: forall v'1 v'2 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
- statelessRandomNormal' :: forall v'1 v'2 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
- statelessRandomUniform :: forall v'1 v'2 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
- statelessRandomUniform' :: forall v'1 v'2 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
- statelessTruncatedNormal :: forall v'1 v'2 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
- statelessTruncatedNormal' :: forall v'1 v'2 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
- statsAggregatorHandle :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
- statsAggregatorHandle' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
- statsAggregatorSummary :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
- statsAggregatorSummary' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
- stopGradient :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- stopGradient' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- stridedSlice :: forall v'1 v'2 v'3 v'4 t index. (TensorType t, OneOf '[Int32, Int64] index) => Tensor v'1 t -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor Build t
- stridedSlice' :: forall v'1 v'2 v'3 v'4 t index. (TensorType t, OneOf '[Int32, Int64] index) => OpParams -> Tensor v'1 t -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor Build t
- stridedSliceAssign :: forall v'2 v'3 v'4 v'5 t index m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] index) => Tensor Ref t -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor v'5 t -> m' (Tensor Ref t)
- stridedSliceAssign' :: forall v'2 v'3 v'4 v'5 t index m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] index) => OpParams -> Tensor Ref t -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor v'5 t -> m' (Tensor Ref t)
- stridedSliceGrad :: forall v'1 v'2 v'3 v'4 v'5 t index. (TensorType t, OneOf '[Int32, Int64] index) => Tensor v'1 index -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor v'5 t -> Tensor Build t
- stridedSliceGrad' :: forall v'1 v'2 v'3 v'4 v'5 t index. (TensorType t, OneOf '[Int32, Int64] index) => OpParams -> Tensor v'1 index -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor v'5 t -> Tensor Build t
- stringJoin :: [Tensor v'1 ByteString] -> Tensor Build ByteString
- stringJoin' :: OpParams -> [Tensor v'1 ByteString] -> Tensor Build ByteString
- stringSplit :: Tensor v'1 ByteString -> Tensor v'2 ByteString -> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64)
- stringSplit' :: OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64)
- stringStrip :: Tensor v'1 ByteString -> Tensor Build ByteString
- stringStrip' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
- stringToHashBucket :: Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
- stringToHashBucket' :: OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
- stringToHashBucketFast :: Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
- stringToHashBucketFast' :: OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
- stringToHashBucketStrong :: Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
- stringToHashBucketStrong' :: OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
- stringToNumber :: forall v'1 out_type. OneOf '[Int32, Int64, Double, Float] out_type => Tensor v'1 ByteString -> Tensor Build out_type
- stringToNumber' :: forall v'1 out_type. OneOf '[Int32, Int64, Double, Float] out_type => OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
- sub :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- sub' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- substr :: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64] t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build ByteString
- substr' :: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64] t => OpParams -> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build ByteString
- sum :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- sum' :: forall v'1 v'2 t tidx. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
- summaryWriter :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
- summaryWriter' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
- svd :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
- svd' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
- switch :: forall v'1 v'2 t. TensorType t => Tensor v'1 t -> Tensor v'2 Bool -> (Tensor Build t, Tensor Build t)
- switch' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 Bool -> (Tensor Build t, Tensor Build t)
- tFRecordDataset :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
- tFRecordDataset' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
- tFRecordReader :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
- tFRecordReader' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
- tFRecordReaderV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
- tFRecordReaderV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
- tPUCompilationResult :: Tensor Build ByteString
- tPUCompilationResult' :: OpParams -> Tensor Build ByteString
- tPUEmbeddingActivations :: Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
- tPUEmbeddingActivations' :: OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
- tPUEmbeddingEnqueueSparseBatch :: forall v'1 v'2 v'3 m'. MonadBuild m' => [Tensor v'1 Int32] -> [Tensor v'2 Int32] -> [Tensor v'3 Float] -> m' ControlNode
- tPUEmbeddingEnqueueSparseBatch' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> [Tensor v'1 Int32] -> [Tensor v'2 Int32] -> [Tensor v'3 Float] -> m' ControlNode
- tPUEmbeddingLoadAdagradParameters :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
- tPUEmbeddingLoadAdagradParameters' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
- tPUEmbeddingLoadGradientDescentParameters :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Tensor v'1 Float -> m' ControlNode
- tPUEmbeddingLoadGradientDescentParameters' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Int64 -> Tensor v'1 Float -> m' ControlNode
- tPUEmbeddingReceiveActivations :: forall m'. MonadBuild m' => Int64 -> m' [Tensor Value Float]
- tPUEmbeddingReceiveActivations' :: forall m'. MonadBuild m' => OpParams -> Int64 -> m' [Tensor Value Float]
- tPUEmbeddingRetrieveAdagradParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
- tPUEmbeddingRetrieveAdagradParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
- tPUEmbeddingRetrieveGradientDescentParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> m' (Tensor Value Float)
- tPUEmbeddingRetrieveGradientDescentParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Int64 -> m' (Tensor Value Float)
- tPUEmbeddingSendGradients :: forall v'1 m'. MonadBuild m' => [Tensor v'1 Float] -> m' ControlNode
- tPUEmbeddingSendGradients' :: forall v'1 m'. MonadBuild m' => OpParams -> [Tensor v'1 Float] -> m' ControlNode
- tPUReplicateMetadata :: forall m'. MonadBuild m' => Int64 -> m' ControlNode
- tPUReplicateMetadata' :: forall m'. MonadBuild m' => OpParams -> Int64 -> m' ControlNode
- tPUReplicatedInput :: forall v'1 t. TensorType t => [Tensor v'1 t] -> Tensor Build t
- tPUReplicatedInput' :: forall v'1 t. TensorType t => OpParams -> [Tensor v'1 t] -> Tensor Build t
- tPUReplicatedOutput :: forall v'1 t. TensorType t => Int64 -> Tensor v'1 t -> [Tensor Build t]
- tPUReplicatedOutput' :: forall v'1 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> [Tensor Build t]
- takeDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
- takeDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
- takeManySparseFromTensorsMap :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
- takeManySparseFromTensorsMap' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
- tan :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- tan' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- tanh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
- tanh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
- tanhGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- tanhGrad' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- temporaryVariable :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Ref dtype)
- temporaryVariable' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Ref dtype)
- tensorArray :: forall v'1 m'. MonadBuild m' => DataType -> Tensor v'1 Int32 -> m' (Tensor Ref ByteString)
- tensorArray' :: forall v'1 m'. MonadBuild m' => OpParams -> DataType -> Tensor v'1 Int32 -> m' (Tensor Ref ByteString)
- tensorArrayClose :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' ControlNode
- tensorArrayClose' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' ControlNode
- tensorArrayCloseV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' ControlNode
- tensorArrayCloseV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' ControlNode
- tensorArrayCloseV3 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
- tensorArrayCloseV3' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
- tensorArrayConcat :: forall v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value dtype, Tensor Value Int64)
- tensorArrayConcat' :: forall v'2 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value dtype, Tensor Value Int64)
- tensorArrayConcatV2 :: forall v'1 v'2 dtype. TensorType dtype => Tensor v'1 ByteString -> Tensor v'2 Float -> (Tensor Build dtype, Tensor Build Int64)
- tensorArrayConcatV2' :: forall v'1 v'2 dtype. TensorType dtype => OpParams -> Tensor v'1 ByteString -> Tensor v'2 Float -> (Tensor Build dtype, Tensor Build Int64)
- tensorArrayConcatV3 :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value dtype, Tensor Value Int64)
- tensorArrayConcatV3' :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value dtype, Tensor Value Int64)
- tensorArrayGather :: forall v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
- tensorArrayGather' :: forall v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
- tensorArrayGatherV2 :: forall v'1 v'2 v'3 dtype. TensorType dtype => Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build dtype
- tensorArrayGatherV2' :: forall v'1 v'2 v'3 dtype. TensorType dtype => OpParams -> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build dtype
- tensorArrayGatherV3 :: forall v'1 v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
- tensorArrayGatherV3' :: forall v'1 v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
- tensorArrayGrad :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Ref ByteString)
- tensorArrayGrad' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Ref ByteString)
- tensorArrayGradV2 :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Value ByteString)
- tensorArrayGradV2' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Value ByteString)
- tensorArrayGradV3 :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value ResourceHandle, Tensor Value Float)
- tensorArrayGradV3' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value ResourceHandle, Tensor Value Float)
- tensorArrayPack :: forall v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value dtype)
- tensorArrayPack' :: forall v'2 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value dtype)
- tensorArrayRead :: forall v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
- tensorArrayRead' :: forall v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
- tensorArrayReadV2 :: forall v'1 v'2 v'3 dtype. TensorType dtype => Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build dtype
- tensorArrayReadV2' :: forall v'1 v'2 v'3 dtype. TensorType dtype => OpParams -> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build dtype
- tensorArrayReadV3 :: forall v'1 v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
- tensorArrayReadV3' :: forall v'1 v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
- tensorArrayScatter :: forall v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorArrayScatter' :: forall v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorArrayScatterV2 :: forall v'1 v'2 v'3 v'4 t. TensorType t => Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> Tensor Build Float
- tensorArrayScatterV2' :: forall v'1 v'2 v'3 v'4 t. TensorType t => OpParams -> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> Tensor Build Float
- tensorArrayScatterV3 :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorArrayScatterV3' :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorArraySize :: forall v'2 m'. MonadBuild m' => Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value Int32)
- tensorArraySize' :: forall v'2 m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value Int32)
- tensorArraySizeV2 :: Tensor v'1 ByteString -> Tensor v'2 Float -> Tensor Build Int32
- tensorArraySizeV2' :: OpParams -> Tensor v'1 ByteString -> Tensor v'2 Float -> Tensor Build Int32
- tensorArraySizeV3 :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value Int32)
- tensorArraySizeV3' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value Int32)
- tensorArraySplit :: forall v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor Ref ByteString -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorArraySplit' :: forall v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref ByteString -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorArraySplitV2 :: forall v'1 v'2 v'3 v'4 t. TensorType t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Float -> Tensor Build Float
- tensorArraySplitV2' :: forall v'1 v'2 v'3 v'4 t. TensorType t => OpParams -> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Float -> Tensor Build Float
- tensorArraySplitV3 :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor v'1 ResourceHandle -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorArraySplitV3' :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorArrayUnpack :: forall v'2 v'3 t m'. (MonadBuild m', TensorType t) => Tensor Ref ByteString -> Tensor v'2 t -> Tensor v'3 Float -> m' (Tensor Value Float)
- tensorArrayUnpack' :: forall v'2 v'3 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref ByteString -> Tensor v'2 t -> Tensor v'3 Float -> m' (Tensor Value Float)
- tensorArrayV2 :: forall v'1 m'. MonadBuild m' => DataType -> Tensor v'1 Int32 -> m' (Tensor Value ByteString)
- tensorArrayV2' :: forall v'1 m'. MonadBuild m' => OpParams -> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ByteString)
- tensorArrayV3 :: forall v'1 m'. MonadBuild m' => DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle, Tensor Value Float)
- tensorArrayV3' :: forall v'1 m'. MonadBuild m' => OpParams -> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle, Tensor Value Float)
- tensorArrayWrite :: forall v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorArrayWrite' :: forall v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorArrayWriteV2 :: forall v'1 v'2 v'3 v'4 t. TensorType t => Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> Tensor Build Float
- tensorArrayWriteV2' :: forall v'1 v'2 v'3 v'4 t. TensorType t => OpParams -> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> Tensor Build Float
- tensorArrayWriteV3 :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorArrayWriteV3' :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
- tensorDataset :: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types) => TensorList v'1 toutput_types -> m' (Tensor Value Variant)
- tensorDataset' :: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types) => OpParams -> TensorList v'1 toutput_types -> m' (Tensor Value Variant)
- tensorListConcatLists :: DataType -> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
- tensorListConcatLists' :: OpParams -> DataType -> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
- tensorListElementShape :: forall v'1 shape_type. OneOf '[Int32, Int64] shape_type => Tensor v'1 Variant -> Tensor Build shape_type
- tensorListElementShape' :: forall v'1 shape_type. OneOf '[Int32, Int64] shape_type => OpParams -> Tensor v'1 Variant -> Tensor Build shape_type
- tensorListFromTensor :: forall v'1 v'2 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => Tensor v'1 element_dtype -> Tensor v'2 shape_type -> Tensor Build Variant
- tensorListFromTensor' :: forall v'1 v'2 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => OpParams -> Tensor v'1 element_dtype -> Tensor v'2 shape_type -> Tensor Build Variant
- tensorListGetItem :: forall v'1 v'2 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build element_dtype
- tensorListGetItem' :: forall v'1 v'2 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build element_dtype
- tensorListLength :: Tensor v'1 Variant -> Tensor Build Int32
- tensorListLength' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int32
- tensorListPopBack :: forall v'1 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> (Tensor Build Variant, Tensor Build element_dtype)
- tensorListPopBack' :: forall v'1 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> (Tensor Build Variant, Tensor Build element_dtype)
- tensorListPushBack :: forall v'1 v'2 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 element_dtype -> Tensor Build Variant
- tensorListPushBack' :: forall v'1 v'2 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 element_dtype -> Tensor Build Variant
- tensorListPushBackBatch :: forall v'1 v'2 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 element_dtype -> Tensor Build Variant
- tensorListPushBackBatch' :: forall v'1 v'2 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 element_dtype -> Tensor Build Variant
- tensorListReserve :: forall v'1 v'2 shape_type. OneOf '[Int32, Int64] shape_type => DataType -> Tensor v'1 shape_type -> Tensor v'2 Int32 -> Tensor Build Variant
- tensorListReserve' :: forall v'1 v'2 shape_type. OneOf '[Int32, Int64] shape_type => OpParams -> DataType -> Tensor v'1 shape_type -> Tensor v'2 Int32 -> Tensor Build Variant
- tensorListSetItem :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor v'3 element_dtype -> Tensor Build Variant
- tensorListSetItem' :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor v'3 element_dtype -> Tensor Build Variant
- tensorListStack :: forall v'1 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor Build element_dtype
- tensorListStack' :: forall v'1 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor Build element_dtype
- tensorSliceDataset :: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types) => TensorList v'1 toutput_types -> m' (Tensor Value Variant)
- tensorSliceDataset' :: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types) => OpParams -> TensorList v'1 toutput_types -> m' (Tensor Value Variant)
- tensorSummary :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build ByteString
- tensorSummary' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build ByteString
- tensorSummaryV2 :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor v'3 ByteString -> Tensor Build ByteString
- tensorSummaryV2' :: forall v'1 v'2 v'3 t. TensorType t => OpParams -> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor v'3 ByteString -> Tensor Build ByteString
- textLineDataset :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
- textLineDataset' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
- textLineReader :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
- textLineReader' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
- textLineReaderV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
- textLineReaderV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
- threadUnsafeUnigramCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- threadUnsafeUnigramCandidateSampler' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- tile :: forall v'1 v'2 t tmultiples. (TensorType t, OneOf '[Int32, Int64] tmultiples) => Tensor v'1 t -> Tensor v'2 tmultiples -> Tensor Build t
- tile' :: forall v'1 v'2 t tmultiples. (TensorType t, OneOf '[Int32, Int64] tmultiples) => OpParams -> Tensor v'1 t -> Tensor v'2 tmultiples -> Tensor Build t
- tileGrad :: forall v'1 v'2 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
- tileGrad' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
- timestamp :: forall m'. MonadBuild m' => m' (Tensor Value Double)
- timestamp' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value Double)
- topK :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Int64 -> Tensor v'1 t -> (Tensor Build t, Tensor Build Int32)
- topK' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Int64 -> Tensor v'1 t -> (Tensor Build t, Tensor Build Int32)
- topKV2 :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> (Tensor Build t, Tensor Build Int32)
- topKV2' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> (Tensor Build t, Tensor Build Int32)
- transpose :: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm) => Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
- transpose' :: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm) => OpParams -> Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
- truncateDiv :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- truncateDiv' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- truncateMod :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- truncateMod' :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- truncatedNormal :: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t) => Tensor v'1 t -> m' (Tensor Value dtype)
- truncatedNormal' :: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t) => OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
- tryRpc :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value ByteString, Tensor Value Int32, Tensor Value ByteString)
- tryRpc' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value ByteString, Tensor Value Int32, Tensor Value ByteString)
- unbatch :: forall v'1 v'2 v'3 t. TensorType t => Int64 -> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build t
- unbatch' :: forall v'1 v'2 v'3 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build t
- unbatchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
- unbatchDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
- unbatchGrad :: forall v'1 v'2 v'3 v'4 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 t -> Tensor v'4 Int64 -> Tensor Build t
- unbatchGrad' :: forall v'1 v'2 v'3 v'4 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 t -> Tensor v'4 Int64 -> Tensor Build t
- uniformCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- uniformCandidateSampler' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
- unique :: forall v'1 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx) => Tensor v'1 t -> (Tensor Build t, Tensor Build out_idx)
- unique' :: forall v'1 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx) => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build out_idx)
- uniqueV2 :: forall v'1 v'2 t taxis out_idx. (TensorType t, OneOf '[Int32, Int64] taxis, OneOf '[Int32, Int64] out_idx) => Tensor v'1 t -> Tensor v'2 taxis -> (Tensor Build t, Tensor Build out_idx)
- uniqueV2' :: forall v'1 v'2 t taxis out_idx. (TensorType t, OneOf '[Int32, Int64] taxis, OneOf '[Int32, Int64] out_idx) => OpParams -> Tensor v'1 t -> Tensor v'2 taxis -> (Tensor Build t, Tensor Build out_idx)
- uniqueWithCounts :: forall v'1 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx) => Tensor v'1 t -> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
- uniqueWithCounts' :: forall v'1 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx) => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
- uniqueWithCountsV2 :: forall v'1 v'2 t taxis out_idx. (TensorType t, OneOf '[Int32, Int64] taxis, OneOf '[Int32, Int64] out_idx) => Tensor v'1 t -> Tensor v'2 taxis -> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
- uniqueWithCountsV2' :: forall v'1 v'2 t taxis out_idx. (TensorType t, OneOf '[Int32, Int64] taxis, OneOf '[Int32, Int64] out_idx) => OpParams -> Tensor v'1 t -> Tensor v'2 taxis -> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
- unpack :: forall v'1 t. TensorType t => Int64 -> Tensor v'1 t -> [Tensor Build t]
- unpack' :: forall v'1 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> [Tensor Build t]
- unravelIndex :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor Build tidx
- unravelIndex' :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => OpParams -> Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor Build tidx
- unsortedSegmentMax :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
- unsortedSegmentMax' :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
- unsortedSegmentMin :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
- unsortedSegmentMin' :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
- unsortedSegmentProd :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
- unsortedSegmentProd' :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
- unsortedSegmentSum :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
- unsortedSegmentSum' :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
- unstage :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => m' (TensorList Value dtypes)
- unstage' :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> m' (TensorList Value dtypes)
- varHandleOp :: forall m'. MonadBuild m' => DataType -> Shape -> m' (Tensor Value ResourceHandle)
- varHandleOp' :: forall m'. MonadBuild m' => OpParams -> DataType -> Shape -> m' (Tensor Value ResourceHandle)
- varIsInitializedOp :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
- varIsInitializedOp' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
- variable :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Ref dtype)
- variable' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Ref dtype)
- variableShape :: forall v'1 out_type m'. (MonadBuild m', OneOf '[Int32, Int64] out_type) => Tensor v'1 ResourceHandle -> m' (Tensor Value out_type)
- variableShape' :: forall v'1 out_type m'. (MonadBuild m', OneOf '[Int32, Int64] out_type) => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value out_type)
- variableV2 :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Ref dtype)
- variableV2' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Ref dtype)
- where' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build Int64
- where'' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build Int64
- wholeFileReader :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
- wholeFileReader' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
- wholeFileReaderV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
- wholeFileReaderV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
- workerHeartbeat :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value ByteString)
- workerHeartbeat' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value ByteString)
- writeAudioSummary :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 Float -> Tensor v'5 Float -> m' ControlNode
- writeAudioSummary' :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 Float -> Tensor v'5 Float -> m' ControlNode
- writeFile :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
- writeFile' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
- writeGraphSummary :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
- writeGraphSummary' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
- writeHistogramSummary :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 t -> m' ControlNode
- writeHistogramSummary' :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 t -> m' ControlNode
- writeImageSummary :: forall v'1 v'2 v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Word16, Word8, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 t -> Tensor v'5 Word8 -> m' ControlNode
- writeImageSummary' :: forall v'1 v'2 v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Word16, Word8, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 t -> Tensor v'5 Word8 -> m' ControlNode
- writeScalarSummary :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 t -> m' ControlNode
- writeScalarSummary' :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 t -> m' ControlNode
- writeSummary :: forall v'1 v'2 v'3 v'4 v'5 t m'. (MonadBuild m', TensorType t) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 t -> Tensor v'4 ByteString -> Tensor v'5 ByteString -> m' ControlNode
- writeSummary' :: forall v'1 v'2 v'3 v'4 v'5 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 t -> Tensor v'4 ByteString -> Tensor v'5 ByteString -> m' ControlNode
- zerosLike :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
- zerosLike' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
- zeta :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- zeta' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- zipDataset :: [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
- zipDataset' :: OpParams -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
- _Arg :: forall t m'. (MonadBuild m', TensorType t) => Int64 -> m' (Tensor Value t)
- _Arg' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> m' (Tensor Value t)
- _ArrayToList :: forall v'1 t out_types. (TensorType t, TensorTypes out_types) => [Tensor v'1 t] -> TensorList Build out_types
- _ArrayToList' :: forall v'1 t out_types. (TensorType t, TensorTypes out_types) => OpParams -> [Tensor v'1 t] -> TensorList Build out_types
- _ConfigureDistributedTPU :: forall v'1 m'. MonadBuild m' => [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
- _ConfigureDistributedTPU' :: forall v'1 m'. MonadBuild m' => OpParams -> [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
- _DisconnectHostFromDistributedTPUSystem :: forall m'. MonadBuild m' => m' (Tensor Value Int32)
- _DisconnectHostFromDistributedTPUSystem' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value Int32)
- _HostCast :: forall v'1 srcT dstT. (TensorType srcT, TensorType dstT) => Tensor v'1 srcT -> Tensor Build dstT
- _HostCast' :: forall v'1 srcT dstT. (TensorType srcT, TensorType dstT) => OpParams -> Tensor v'1 srcT -> Tensor Build dstT
- _HostRecv :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => Int64 -> m' (Tensor Value tensor_type)
- _HostRecv' :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => OpParams -> Int64 -> m' (Tensor Value tensor_type)
- _HostSend :: forall v'1 t m'. (MonadBuild m', TensorType t) => Int64 -> Tensor v'1 t -> m' ControlNode
- _HostSend' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
- _InitializeHostForDistributedTPU :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value Int32)
- _InitializeHostForDistributedTPU' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Int32)
- _ListToArray :: forall v'1 tin t. (TensorTypes tin, TensorType t) => Int64 -> TensorList v'1 tin -> [Tensor Build t]
- _ListToArray' :: forall v'1 tin t. (TensorTypes tin, TensorType t) => OpParams -> Int64 -> TensorList v'1 tin -> [Tensor Build t]
- _MklAdd :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
- _MklAdd' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
- _MklMaximum :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
- _MklMaximum' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
- _MklMul :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
- _MklMul' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
- _MklSquaredDifference :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
- _MklSquaredDifference' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
- _MklSub :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
- _MklSub' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
- _ParallelConcatStart :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Value dtype)
- _ParallelConcatStart' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Value dtype)
- _ParallelConcatUpdate :: forall v'1 v'2 t. TensorType t => Int64 -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- _ParallelConcatUpdate' :: forall v'1 v'2 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
- _Recv :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => Int64 -> m' (Tensor Value tensor_type)
- _Recv' :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => OpParams -> Int64 -> m' (Tensor Value tensor_type)
- _Retval :: forall v'1 t m'. (MonadBuild m', TensorType t) => Int64 -> Tensor v'1 t -> m' ControlNode
- _Retval' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
- _ScopedAllocator :: forall t m'. (MonadBuild m', TensorType t) => Int64 -> Int64 -> Shape -> m' (Tensor Value t)
- _ScopedAllocator' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Int64 -> Shape -> m' (Tensor Value t)
- _ScopedAllocatorConcat :: forall v'1 v'2 t m'. (MonadBuild m', TensorType t) => Int64 -> Shape -> Tensor v'1 t -> [Tensor v'2 t] -> m' (Tensor Value t)
- _ScopedAllocatorConcat' :: forall v'1 v'2 t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Shape -> Tensor v'1 t -> [Tensor v'2 t] -> m' (Tensor Value t)
- _ScopedAllocatorSplit :: forall v'1 v'2 t m'. (MonadBuild m', TensorType t) => Int64 -> Tensor v'1 t -> [Tensor v'2 t] -> m' [Tensor Value t]
- _ScopedAllocatorSplit' :: forall v'1 v'2 t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Tensor v'1 t -> [Tensor v'2 t] -> m' [Tensor Value t]
- _Send :: forall v'1 t m'. (MonadBuild m', TensorType t) => Int64 -> Tensor v'1 t -> m' ControlNode
- _Send' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
- _SetGlobalTPUArray :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' ControlNode
- _SetGlobalTPUArray' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' ControlNode
- _ShutdownDistributedTPU :: forall m'. MonadBuild m' => m' ControlNode
- _ShutdownDistributedTPU' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode
- _WaitForDistributedTPU :: forall v'1 m'. MonadBuild m' => [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
- _WaitForDistributedTPU' :: forall v'1 m'. MonadBuild m' => OpParams -> [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
- _XlaRecvAtHost :: forall v'1 toutputs m'. (MonadBuild m', TensorTypes toutputs) => Int64 -> Tensor v'1 ByteString -> m' (TensorList Value toutputs)
- _XlaRecvAtHost' :: forall v'1 toutputs m'. (MonadBuild m', TensorTypes toutputs) => OpParams -> Int64 -> Tensor v'1 ByteString -> m' (TensorList Value toutputs)
- _XlaSendFromHost :: forall v'1 v'2 tinputs m'. (MonadBuild m', TensorTypes tinputs) => Int64 -> TensorList v'1 tinputs -> Tensor v'2 ByteString -> m' ControlNode
- _XlaSendFromHost' :: forall v'1 v'2 tinputs m'. (MonadBuild m', TensorTypes tinputs) => OpParams -> Int64 -> TensorList v'1 tinputs -> Tensor v'2 ByteString -> m' ControlNode
Documentation
abort :: forall m'. MonadBuild m' => m' ControlNode Source #
abort' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode Source #
accumulatorNumAccumulated Source #
:: MonadBuild m' | |
=> Tensor Ref ByteString | handle |
-> m' (Tensor Value Int32) | num_accumulated |
accumulatorNumAccumulated' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> m' (Tensor Value Int32) | num_accumulated |
accumulatorSetGlobalStep Source #
:: MonadBuild m' | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Int64 | new_global_step |
-> m' ControlNode |
accumulatorSetGlobalStep' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Int64 | new_global_step |
-> m' ControlNode |
addManySparseToTensorsMap Source #
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 Int64 | sparse_indices |
-> Tensor v'2 t | sparse_values |
-> Tensor v'3 Int64 | sparse_shape |
-> m' (Tensor Value Int64) | sparse_handles |
addSparseToTensorsMap Source #
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 Int64 | sparse_indices |
-> Tensor v'2 t | sparse_values |
-> Tensor v'3 Int64 | sparse_shape |
-> m' (Tensor Value Int64) | sparse_handle |
:: MonadBuild m' | |
=> [DataType] | output_types |
-> m' (Tensor Value ResourceHandle) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | output_types |
-> m' (Tensor Value ResourceHandle) | handle |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor Ref t | var |
-> Tensor Ref t | m |
-> Tensor Ref t | v |
-> Tensor v'4 t | beta1_power |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | beta1 |
-> Tensor v'7 t | beta2 |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | m |
-> Tensor Ref t | v |
-> Tensor v'4 t | beta1_power |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | beta1 |
-> Tensor v'7 t | beta2 |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor Ref t | accum_update |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | epsilon |
-> Tensor v'7 t | grad |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor Ref t | var |
-> Tensor Ref t | gradient_accumulator |
-> Tensor Ref t | gradient_squared_accumulator |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 Int64 | global_step |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | gradient_accumulator |
-> Tensor Ref t | gradient_squared_accumulator |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 Int64 | global_step |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor Ref t | var |
-> Tensor Ref t | m |
-> Tensor Ref t | v |
-> Tensor v'4 t | beta1_power |
-> Tensor v'5 t | beta2_power |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | beta1 |
-> Tensor v'8 t | beta2 |
-> Tensor v'9 t | epsilon |
-> Tensor v'10 t | grad |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | m |
-> Tensor Ref t | v |
-> Tensor v'4 t | beta1_power |
-> Tensor v'5 t | beta2_power |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | beta1 |
-> Tensor v'8 t | beta2 |
-> Tensor v'9 t | epsilon |
-> Tensor v'10 t | grad |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | m |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | alpha |
-> Tensor v'5 t | sign_decay |
-> Tensor v'6 t | beta |
-> Tensor v'7 t | grad |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor Ref t | var |
-> Tensor Ref t | mg |
-> Tensor Ref t | ms |
-> Tensor Ref t | mom |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | rho |
-> Tensor v'7 t | momentum |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> m' (Tensor Ref t) | out |
applyCenteredRMSProp' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | mg |
-> Tensor Ref t | ms |
-> Tensor Ref t | mom |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | rho |
-> Tensor v'7 t | momentum |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor Ref t | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 t | lr_power |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor Ref t | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 t | lr_power |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor Ref t | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 t | l2_shrinkage |
-> Tensor v'9 t | lr_power |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor Ref t | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 t | l2_shrinkage |
-> Tensor v'9 t | lr_power |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | m |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | logbase |
-> Tensor v'5 t | sign_decay |
-> Tensor v'6 t | beta |
-> Tensor v'7 t | grad |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor Ref t | var |
-> Tensor Ref t | ms |
-> Tensor Ref t | mom |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | momentum |
-> Tensor v'7 t | epsilon |
-> Tensor v'8 t | grad |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | ms |
-> Tensor Ref t | mom |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | momentum |
-> Tensor v'7 t | epsilon |
-> Tensor v'8 t | grad |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', TensorTypes t) | |
=> Tensor v'1 Bool | condition |
-> TensorList v'2 t | data |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes t) | |
=> OpParams | |
-> Tensor v'1 Bool | condition |
-> TensorList v'2 t | data |
-> m' ControlNode |
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref t | ref |
-> Tensor v'2 t | value |
-> m' (Tensor Ref t) | output_ref |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref t | ref |
-> Tensor v'2 t | value |
-> m' (Tensor Ref t) | output_ref |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor v'1 ResourceHandle | resource |
-> Tensor v'2 dtype | value |
-> m' ControlNode |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource |
-> Tensor v'2 dtype | value |
-> m' ControlNode |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor v'1 ResourceHandle | resource |
-> Tensor v'2 dtype | value |
-> m' ControlNode |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource |
-> Tensor v'2 dtype | value |
-> m' ControlNode |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor v'1 ResourceHandle | resource |
-> Tensor v'2 dtype | value |
-> m' ControlNode |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource |
-> Tensor v'2 dtype | value |
-> m' ControlNode |
:: Float | sample_rate |
-> Tensor v'1 ByteString | tag |
-> Tensor v'2 Float | tensor |
-> Tensor Build ByteString | summary |
:: OpParams | |
-> Float | sample_rate |
-> Tensor v'1 ByteString | tag |
-> Tensor v'2 Float | tensor |
-> Tensor Build ByteString | summary |
:: Tensor v'1 ByteString | tag |
-> Tensor v'2 Float | tensor |
-> Tensor v'3 Float | sample_rate |
-> Tensor Build ByteString | summary |
:: OpParams | |
-> Tensor v'1 ByteString | tag |
-> Tensor v'2 Float | tensor |
-> Tensor v'3 Float | sample_rate |
-> Tensor Build ByteString | summary |
:: MonadBuild m' | |
=> [DataType] | component_types |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | component_types |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> Tensor Ref ByteString | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> m' ControlNode |
barrierIncompleteSize Source #
:: MonadBuild m' | |
=> Tensor Ref ByteString | handle |
-> m' (Tensor Value Int32) | size |
barrierIncompleteSize' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> m' (Tensor Value Int32) | size |
:: (MonadBuild m', TensorType t) | |
=> Int64 | component_index |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 ByteString | keys |
-> Tensor v'3 t | values |
-> m' ControlNode |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Int64 | component_index |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 ByteString | keys |
-> Tensor v'3 t | values |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor Ref ByteString | handle |
-> m' (Tensor Value Int32) | size |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> m' (Tensor Value Int32) | size |
:: (MonadBuild m', TensorTypes component_types) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | num_elements |
-> m' (Tensor Value Int64, Tensor Value ByteString, TensorList Value component_types) | (indices, keys, values)
|
:: (MonadBuild m', TensorTypes component_types) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | num_elements |
-> m' (Tensor Value Int64, Tensor Value ByteString, TensorList Value component_types) | (indices, keys, values)
|
:: TensorTypes t | |
=> Int64 | batch_timeout_micros |
-> Int64 | grad_timeout_micros |
-> Int64 | max_batch_size |
-> Int64 | num_batch_threads |
-> TensorList v'1 t | in_tensors |
-> (TensorList Build t, Tensor Build Int64, Tensor Build Int64) | (batched_tensors, batch_index, id)
|
:: TensorTypes t | |
=> OpParams | |
-> Int64 | batch_timeout_micros |
-> Int64 | grad_timeout_micros |
-> Int64 | max_batch_size |
-> Int64 | num_batch_threads |
-> TensorList v'1 t | in_tensors |
-> (TensorList Build t, Tensor Build Int64, Tensor Build Int64) | (batched_tensors, batch_index, id)
|
:: TensorType t | |
=> Tensor v'1 t | diagonal |
-> Tensor Build t | output |
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | diagonal |
-> Tensor Build t | output |
:: TensorType t | |
=> Tensor v'1 t | input |
-> Tensor Build t | diagonal |
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | input |
-> Tensor Build t | diagonal |
:: TensorType t | |
=> Tensor v'1 t | input |
-> Tensor v'2 t | diagonal |
-> Tensor Build t | output |
batchNormWithGlobalNormalizationGrad Source #
:: OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> Bool | scale_after_normalization |
-> Float | variance_epsilon |
-> Tensor v'1 t | t |
-> Tensor v'2 t | m |
-> Tensor v'3 t | v |
-> Tensor v'4 t | gamma |
-> Tensor v'5 t | backprop |
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t) | (dx, dm, dv, db, dg)
|
batchNormWithGlobalNormalizationGrad' Source #
:: OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> OpParams | |
-> Bool | scale_after_normalization |
-> Float | variance_epsilon |
-> Tensor v'1 t | t |
-> Tensor v'2 t | m |
-> Tensor v'3 t | v |
-> Tensor v'4 t | gamma |
-> Tensor v'5 t | backprop |
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t) | (dx, dm, dv, db, dg)
|
:: MonadBuild m' | |
=> Int64 | timestamp_millis: Table snapshot timestamp in millis since epoch. Relative (negative or zero) snapshot times are not allowed. For more details, see 'Table Decorators' in BigQuery docs. |
-> m' (Tensor Ref ByteString) | reader_handle: The handle to reference the Reader. |
A Reader that outputs rows from a BigQuery table as tensorflow Examples.
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | timestamp_millis: Table snapshot timestamp in millis since epoch. Relative (negative or zero) snapshot times are not allowed. For more details, see 'Table Decorators' in BigQuery docs. |
-> m' (Tensor Ref ByteString) | reader_handle: The handle to reference the Reader. |
boostedTreesCalculateBestGainsPerFeature Source #
:: Int64 | max_splits |
-> Tensor v'1 Int32 | node_id_range |
-> [Tensor v'2 Float] | stats_summary_list |
-> Tensor v'3 Float | l1 |
-> Tensor v'4 Float | l2 |
-> Tensor v'5 Float | tree_complexity |
-> Tensor v'6 Float | min_node_weight |
-> ([Tensor Build Int32], [Tensor Build Float], [Tensor Build Int32], [Tensor Build Float], [Tensor Build Float]) | (node_ids_list, gains_list, thresholds_list, left_node_contribs_list, right_node_contribs_list)
|
boostedTreesCalculateBestGainsPerFeature' Source #
:: OpParams | |
-> Int64 | max_splits |
-> Tensor v'1 Int32 | node_id_range |
-> [Tensor v'2 Float] | stats_summary_list |
-> Tensor v'3 Float | l1 |
-> Tensor v'4 Float | l2 |
-> Tensor v'5 Float | tree_complexity |
-> Tensor v'6 Float | min_node_weight |
-> ([Tensor Build Int32], [Tensor Build Float], [Tensor Build Int32], [Tensor Build Float], [Tensor Build Float]) | (node_ids_list, gains_list, thresholds_list, left_node_contribs_list, right_node_contribs_list)
|
boostedTreesCreateEnsemble Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> Tensor v'2 Int64 | stamp_token |
-> Tensor v'3 ByteString | tree_ensemble_serialized |
-> m' ControlNode |
boostedTreesCreateEnsemble' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> Tensor v'2 Int64 | stamp_token |
-> Tensor v'3 ByteString | tree_ensemble_serialized |
-> m' ControlNode |
boostedTreesDeserializeEnsemble Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> Tensor v'2 Int64 | stamp_token |
-> Tensor v'3 ByteString | tree_ensemble_serialized |
-> m' ControlNode |
boostedTreesDeserializeEnsemble' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> Tensor v'2 Int64 | stamp_token |
-> Tensor v'3 ByteString | tree_ensemble_serialized |
-> m' ControlNode |
boostedTreesEnsembleResourceHandleOp Source #
:: MonadBuild m' | |
=> m' (Tensor Value ResourceHandle) | resource |
boostedTreesEnsembleResourceHandleOp' Source #
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value ResourceHandle) | resource |
boostedTreesGetEnsembleStates Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> m' (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32, Tensor Value Int32, Tensor Value Int32) | (stamp_token, num_trees, num_finalized_trees, num_attempted_layers, last_layer_nodes_range)
|
boostedTreesGetEnsembleStates' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> m' (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32, Tensor Value Int32, Tensor Value Int32) | (stamp_token, num_trees, num_finalized_trees, num_attempted_layers, last_layer_nodes_range)
|
:: MonadBuild m' | |
=> Int64 | logits_dimension |
-> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> [Tensor v'2 Int32] | bucketized_features |
-> m' (Tensor Value Float) | logits |
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | logits_dimension |
-> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> [Tensor v'2 Int32] | bucketized_features |
-> m' (Tensor Value Float) | logits |
boostedTreesSerializeEnsemble Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> m' (Tensor Value Int64, Tensor Value ByteString) | (stamp_token, tree_ensemble_serialized)
|
boostedTreesSerializeEnsemble' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> m' (Tensor Value Int64, Tensor Value ByteString) | (stamp_token, tree_ensemble_serialized)
|
boostedTreesTrainingPredict Source #
:: MonadBuild m' | |
=> Int64 | logits_dimension |
-> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> Tensor v'2 Int32 | cached_tree_ids |
-> Tensor v'3 Int32 | cached_node_ids |
-> [Tensor v'4 Int32] | bucketized_features |
-> m' (Tensor Value Float, Tensor Value Int32, Tensor Value Int32) | (partial_logits, tree_ids, node_ids)
|
boostedTreesTrainingPredict' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | logits_dimension |
-> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> Tensor v'2 Int32 | cached_tree_ids |
-> Tensor v'3 Int32 | cached_node_ids |
-> [Tensor v'4 Int32] | bucketized_features |
-> m' (Tensor Value Float, Tensor Value Int32, Tensor Value Int32) | (partial_logits, tree_ids, node_ids)
|
boostedTreesUpdateEnsemble Source #
:: MonadBuild m' | |
=> Int64 | pruning_mode |
-> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> Tensor v'2 Int32 | feature_ids |
-> [Tensor v'3 Int32] | node_ids |
-> [Tensor v'4 Float] | gains |
-> [Tensor v'5 Int32] | thresholds |
-> [Tensor v'6 Float] | left_node_contribs |
-> [Tensor v'7 Float] | right_node_contribs |
-> Tensor v'8 Int32 | max_depth |
-> Tensor v'9 Float | learning_rate |
-> m' ControlNode |
boostedTreesUpdateEnsemble' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | pruning_mode |
-> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> Tensor v'2 Int32 | feature_ids |
-> [Tensor v'3 Int32] | node_ids |
-> [Tensor v'4 Float] | gains |
-> [Tensor v'5 Int32] | thresholds |
-> [Tensor v'6 Float] | left_node_contribs |
-> [Tensor v'7 Float] | right_node_contribs |
-> Tensor v'8 Int32 | max_depth |
-> Tensor v'9 Float | learning_rate |
-> m' ControlNode |
:: Int64 | beam_width |
-> Int64 | top_paths |
-> Tensor v'1 Float | inputs |
-> Tensor v'2 Int32 | sequence_length |
-> ([Tensor Build Int64], [Tensor Build Int64], [Tensor Build Int64], Tensor Build Float) | (decoded_indices, decoded_values, decoded_shape, log_probability)
|
cTCBeamSearchDecoder' Source #
:: OpParams | |
-> Int64 | beam_width |
-> Int64 | top_paths |
-> Tensor v'1 Float | inputs |
-> Tensor v'2 Int32 | sequence_length |
-> ([Tensor Build Int64], [Tensor Build Int64], [Tensor Build Int64], Tensor Build Float) | (decoded_indices, decoded_values, decoded_shape, log_probability)
|
:: (TensorType srcT, TensorType dstT) | |
=> Tensor v'1 srcT | x |
-> Tensor Build dstT | y |
:: (TensorType srcT, TensorType dstT) | |
=> OpParams | |
-> Tensor v'1 srcT | x |
-> Tensor Build dstT | y |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | writer |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | writer |
-> m' ControlNode |
conditionalAccumulator Source #
:: MonadBuild m' | |
=> DataType | dtype |
-> Shape | shape |
-> m' (Tensor Ref ByteString) | handle |
conditionalAccumulator' Source #
:: MonadBuild m' | |
=> OpParams | |
-> DataType | dtype |
-> Shape | shape |
-> m' (Tensor Ref ByteString) | handle |
configureDistributedTPU Source #
:: MonadBuild m' | |
=> m' (Tensor Value ByteString) | topology: A serialized tensorflow.tpu.TopologyProto that describes the TPU topology. |
An op that sets up the centralized structures for a distributed TPU
system.
configureDistributedTPU' Source #
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value ByteString) | topology: A serialized tensorflow.tpu.TopologyProto that describes the TPU topology. |
:: MonadBuild m' | |
=> Tensor v'1 Variant | mutex_lock |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 Variant | mutex_lock |
-> m' ControlNode |
controlTrigger :: forall m'. MonadBuild m' => m' ControlNode Source #
controlTrigger' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode Source #
:: TensorType t | |
=> Tensor v'1 t | input: Input tensor. |
-> Tensor Build t | output: Output tensor, deep-copied from input. |
Copy Op.
Performs CPU-to-CPU or GPU-to-GPU deep-copying of tensor, depending on the device on which the tensor is allocated. N.B.: If the all downstream attached debug ops are disabled given the current gRPC gating status, the output will simply forward the input tensor without deep-copying. See the documentation of Debug* ops for more details.
Unlike the CopyHost Op, this op does not have HostMemory constraint on its input or output.
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | input: Input tensor. |
-> Tensor Build t | output: Output tensor, deep-copied from input. |
:: TensorType t | |
=> Tensor v'1 t | input: Input tensor. |
-> Tensor Build t | output: Output tensor, deep-copied from input. |
Copy Host Op.
Performs CPU-to-CPU deep-copying of tensor. N.B.: If the all downstream attached debug ops are disabled given the current gRPC gating status, the output will simply forward the input tensor without deep-copying. See the documentation of Debug* ops for more details.
Unlike the Copy Op, this op has HostMemory constraint on its input or output.
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | input: Input tensor. |
-> Tensor Build t | output: Output tensor, deep-copied from input. |
createSummaryDbWriter Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 ByteString | db_uri |
-> Tensor v'3 ByteString | experiment_name |
-> Tensor v'4 ByteString | run_name |
-> Tensor v'5 ByteString | user_name |
-> m' ControlNode |
createSummaryDbWriter' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 ByteString | db_uri |
-> Tensor v'3 ByteString | experiment_name |
-> Tensor v'4 ByteString | run_name |
-> Tensor v'5 ByteString | user_name |
-> m' ControlNode |
createSummaryFileWriter Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 ByteString | logdir |
-> Tensor v'3 Int32 | max_queue |
-> Tensor v'4 Int32 | flush_millis |
-> Tensor v'5 ByteString | filename_suffix |
-> m' ControlNode |
createSummaryFileWriter' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 ByteString | logdir |
-> Tensor v'3 Int32 | max_queue |
-> Tensor v'4 Int32 | flush_millis |
-> Tensor v'5 ByteString | filename_suffix |
-> m' ControlNode |
:: OneOf '[Word16, Float] t | |
=> Tensor v'1 t | input: The local input to the sum. |
-> Tensor Build t | output: The sum of all the distributed inputs. |
An Op to sum inputs across replicated TPU instances. Each
instance supplies its own input, and the output of each is the sum of all the inputs.
:: (MonadBuild m', OneOf '[Word16, Double, Float] t) | |
=> Tensor v'1 t | input |
-> Tensor v'2 t | input_h |
-> Tensor v'3 t | input_c |
-> Tensor v'4 t | params |
-> Tensor v'5 t | output |
-> Tensor v'6 t | output_h |
-> Tensor v'7 t | output_c |
-> Tensor v'8 t | output_backprop |
-> Tensor v'9 t | output_h_backprop |
-> Tensor v'10 t | output_c_backprop |
-> Tensor v'11 t | reserve_space |
-> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t) | (input_backprop, input_h_backprop, input_c_backprop, params_backprop)
|
:: (MonadBuild m', OneOf '[Word16, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 t | input |
-> Tensor v'2 t | input_h |
-> Tensor v'3 t | input_c |
-> Tensor v'4 t | params |
-> Tensor v'5 t | output |
-> Tensor v'6 t | output_h |
-> Tensor v'7 t | output_c |
-> Tensor v'8 t | output_backprop |
-> Tensor v'9 t | output_h_backprop |
-> Tensor v'10 t | output_c_backprop |
-> Tensor v'11 t | reserve_space |
-> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t) | (input_backprop, input_h_backprop, input_c_backprop, params_backprop)
|
:: (MonadBuild m', OneOf '[Word16, Double, Float] t) | |
=> Tensor v'1 t | input |
-> Tensor v'2 t | input_h |
-> Tensor v'3 t | input_c |
-> Tensor v'4 t | params |
-> Tensor v'5 t | output |
-> Tensor v'6 t | output_h |
-> Tensor v'7 t | output_c |
-> Tensor v'8 t | output_backprop |
-> Tensor v'9 t | output_h_backprop |
-> Tensor v'10 t | output_c_backprop |
-> Tensor v'11 t | reserve_space |
-> Tensor v'12 Int8 | host_reserved |
-> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t) | (input_backprop, input_h_backprop, input_c_backprop, params_backprop)
|
:: (MonadBuild m', OneOf '[Word16, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 t | input |
-> Tensor v'2 t | input_h |
-> Tensor v'3 t | input_c |
-> Tensor v'4 t | params |
-> Tensor v'5 t | output |
-> Tensor v'6 t | output_h |
-> Tensor v'7 t | output_c |
-> Tensor v'8 t | output_backprop |
-> Tensor v'9 t | output_h_backprop |
-> Tensor v'10 t | output_c_backprop |
-> Tensor v'11 t | reserve_space |
-> Tensor v'12 Int8 | host_reserved |
-> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t) | (input_backprop, input_h_backprop, input_c_backprop, params_backprop)
|
:: (MonadBuild m', OneOf '[Word16, Double, Float] t) | |
=> Tensor v'1 t | input |
-> Tensor v'2 t | input_h |
-> Tensor v'3 t | input_c |
-> Tensor v'4 t | params |
-> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value Int8) | (output, output_h, output_c, reserve_space, host_reserved)
|
:: (MonadBuild m', OneOf '[Word16, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 t | input |
-> Tensor v'2 t | input_h |
-> Tensor v'3 t | input_c |
-> Tensor v'4 t | params |
-> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value Int8) | (output, output_h, output_c, reserve_space, host_reserved)
|
datasetToSingleElement Source #
:: TensorTypes output_types | |
=> Tensor v'1 Variant | dataset |
-> TensorList Build output_types | components |
datasetToSingleElement' Source #
:: TensorTypes output_types | |
=> OpParams | |
-> Tensor v'1 Variant | dataset |
-> TensorList Build output_types | components |
:: MonadBuild m' | |
=> Tensor v'1 Variant | input_dataset |
-> Tensor v'2 ByteString | filename |
-> Tensor v'3 ByteString | compression_type |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 Variant | input_dataset |
-> Tensor v'2 ByteString | filename |
-> Tensor v'3 ByteString | compression_type |
-> m' ControlNode |
debugGradientIdentity Source #
:: TensorType t | |
=> Tensor v'1 t | input |
-> Tensor Build t | output |
debugGradientIdentity' Source #
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | input |
-> Tensor Build t | output |
debugGradientRefIdentity Source #
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref t | input |
-> m' (Tensor Ref t) | output |
debugGradientRefIdentity' Source #
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref t | input |
-> m' (Tensor Ref t) | output |
:: TensorType t | |
=> Tensor v'1 t | input: Input tensor, non-Reference type. |
-> Tensor Build t | output: Output tensor that equals the input tensor. |
Debug Identity Op.
Provides an identity mapping of the non-Ref type input tensor for debugging.
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | input: Input tensor, non-Reference type. |
-> Tensor Build t | output: Output tensor that equals the input tensor. |
:: TensorType t | |
=> Tensor v'1 t | input: Input tensor, non-Reference type. |
-> Tensor Build Int64 | output: An integer output tensor that is the number of NaNs in the input. |
Debug NaN Value Counter Op
Counts number of NaNs in the input tensor, for debugging.
:: TensorType t | |
=> Tensor v'1 t | input: Input tensor, non-Reference type, float or double. |
-> Tensor Build Double | output: A double tensor of shape [14 + nDimensions], where nDimensions is the the number of dimensions of the tensor's shape. The elements of output are: [0]: is initialized (1.0) or not (0.0). [1]: total number of elements [2]: NaN element count [3]: generalized -inf count: elements <= lower_bound. lower_bound is -inf by default. [4]: negative element count (excluding -inf), if lower_bound is the default -inf. Otherwise, this is the count of elements > lower_bound and < 0. [5]: zero element count [6]: positive element count (excluding +inf), if upper_bound is the default -inf. Otherwise, this is the count of elements and 0. [7]: generalized +inf count, elements >= upper_bound. upper_bound is +inf by default. Output elements [1:8] are all zero, if the tensor is uninitialized. [8]: minimum of all non-inf and non-NaN elements. If uninitialized or no such element exists: +inf. [9]: maximum of all non-inf and non-NaN elements. If uninitialized or no such element exists: -inf. [10]: mean of all non-inf and non-NaN elements. If uninitialized or no such element exists: NaN. [11]: variance of all non-inf and non-NaN elements. If uninitialized or no such element exists: NaN. [12]: Data type of the tensor encoded as an enum integer. See the DataType proto for more details. [13]: Number of dimensions of the tensor (ndims). [14+]: Sizes of the dimensions. |
Debug Numeric Summary Op.
Provide a basic summary of numeric value types, range and distribution.
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | input: Input tensor, non-Reference type, float or double. |
-> Tensor Build Double | output: A double tensor of shape [14 + nDimensions], where nDimensions is the the number of dimensions of the tensor's shape. The elements of output are: [0]: is initialized (1.0) or not (0.0). [1]: total number of elements [2]: NaN element count [3]: generalized -inf count: elements <= lower_bound. lower_bound is -inf by default. [4]: negative element count (excluding -inf), if lower_bound is the default -inf. Otherwise, this is the count of elements > lower_bound and < 0. [5]: zero element count [6]: positive element count (excluding +inf), if upper_bound is the default -inf. Otherwise, this is the count of elements and 0. [7]: generalized +inf count, elements >= upper_bound. upper_bound is +inf by default. Output elements [1:8] are all zero, if the tensor is uninitialized. [8]: minimum of all non-inf and non-NaN elements. If uninitialized or no such element exists: +inf. [9]: maximum of all non-inf and non-NaN elements. If uninitialized or no such element exists: -inf. [10]: mean of all non-inf and non-NaN elements. If uninitialized or no such element exists: NaN. [11]: variance of all non-inf and non-NaN elements. If uninitialized or no such element exists: NaN. [12]: Data type of the tensor encoded as an enum integer. See the DataType proto for more details. [13]: Number of dimensions of the tensor (ndims). [14+]: Sizes of the dimensions. |
:: Tensor v'1 ByteString | input |
-> Tensor Build ByteString | output |
:: OpParams | |
-> Tensor v'1 ByteString | input |
-> Tensor Build ByteString | output |
:: OneOfs '[ByteString, Int32, Int64, Double, Float] oUT_TYPE | |
=> Tensor v'1 ByteString | records |
-> TensorList v'2 oUT_TYPE | record_defaults |
-> TensorList Build oUT_TYPE | output |
:: OneOfs '[ByteString, Int32, Int64, Double, Float] oUT_TYPE | |
=> OpParams | |
-> Tensor v'1 ByteString | records |
-> TensorList v'2 oUT_TYPE | record_defaults |
-> TensorList Build oUT_TYPE | output |
:: Tensor v'1 ByteString | bytes |
-> Tensor Build ByteString | output |
:: OpParams | |
-> Tensor v'1 ByteString | bytes |
-> Tensor Build ByteString | output |
:: Tensor v'1 ByteString | json_examples |
-> Tensor Build ByteString | binary_examples |
:: OpParams | |
-> Tensor v'1 ByteString | json_examples |
-> Tensor Build ByteString | binary_examples |
:: Tensor v'1 ByteString | contents |
-> Tensor Build Word8 | image |
:: TensorTypes output_types | |
=> Tensor v'1 ByteString | bytes |
-> (Tensor Build Int32, TensorList Build output_types) | (sizes, values)
|
:: TensorTypes output_types | |
=> OpParams | |
-> Tensor v'1 ByteString | bytes |
-> (Tensor Build Int32, TensorList Build output_types) | (sizes, values)
|
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 t | x |
-> m' (Tensor Value t) | y |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor v'1 t | x |
-> m' (Tensor Value t) | y |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | handle |
-> m' ControlNode |
denseToSparseSetOperation Source #
:: OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t | |
=> Tensor v'1 t | set1 |
-> Tensor v'2 Int64 | set2_indices |
-> Tensor v'3 t | set2_values |
-> Tensor v'4 Int64 | set2_shape |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64) | (result_indices, result_values, result_shape)
|
denseToSparseSetOperation' Source #
:: OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t | |
=> OpParams | |
-> Tensor v'1 t | set1 |
-> Tensor v'2 Int64 | set2_indices |
-> Tensor v'3 t | set2_values |
-> Tensor v'4 Int64 | set2_shape |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64) | (result_indices, result_values, result_shape)
|
:: TensorType t | |
=> Int64 | block_size |
-> Tensor v'1 t | input |
-> Tensor Build t | output |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | resource_handle |
-> Tensor v'2 Variant | serialized |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource_handle |
-> Tensor v'2 Variant | serialized |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | resource |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource |
-> m' ControlNode |
destroyTemporaryVariable Source #
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref t | ref |
-> m' (Tensor Value t) | value |
destroyTemporaryVariable' Source #
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref t | ref |
-> m' (Tensor Value t) | value |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor v'1 Int32 | shape |
-> m' (Tensor Value dtype) | output |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor v'1 Int32 | shape |
-> m' (Tensor Value dtype) | output |
:: Tensor v'1 ByteString | input |
-> Tensor Build ByteString | output |
:: OpParams | |
-> Tensor v'1 ByteString | input |
-> Tensor Build ByteString | output |
:: Tensor v'1 Word8 | image |
-> Tensor Build ByteString | contents |
:: TensorTypes tinput_types | |
=> Tensor v'1 Int32 | sizes |
-> TensorList v'2 tinput_types | values |
-> Tensor Build ByteString | bytes |
:: TensorTypes tinput_types | |
=> OpParams | |
-> Tensor v'1 Int32 | sizes |
-> TensorList v'2 tinput_types | values |
-> Tensor Build ByteString | bytes |
enqueueInQueueDataset Source #
:: (MonadBuild m', TensorTypes tcomponents) | |
=> Tensor v'1 Variant | queue |
-> TensorList v'2 tcomponents | components |
-> m' ControlNode |
enqueueInQueueDataset' Source #
:: (MonadBuild m', TensorTypes tcomponents) | |
=> OpParams | |
-> Tensor v'1 Variant | queue |
-> TensorList v'2 tcomponents | components |
-> m' ControlNode |
:: MonadBuild m' | |
=> [DataType] | component_types |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | component_types |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> [DataType] | component_types |
-> m' (Tensor Value ResourceHandle) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | component_types |
-> m' (Tensor Value ResourceHandle) | handle |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | resource |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource |
-> m' (Tensor Ref ByteString) | handle |
fixedLengthRecordReader Source #
:: MonadBuild m' | |
=> Int64 | record_bytes |
-> m' (Tensor Ref ByteString) | reader_handle |
fixedLengthRecordReader' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | record_bytes |
-> m' (Tensor Ref ByteString) | reader_handle |
fixedLengthRecordReaderV2 Source #
:: MonadBuild m' | |
=> Int64 | record_bytes |
-> m' (Tensor Value ResourceHandle) | reader_handle |
fixedLengthRecordReaderV2' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | record_bytes |
-> m' (Tensor Value ResourceHandle) | reader_handle |
fixedUnigramCandidateSampler Source #
:: MonadBuild m' | |
=> Int64 | num_sampled |
-> Int64 | num_true |
-> Int64 | range_max |
-> Bool | unique |
-> Tensor v'1 Int64 | true_classes |
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float) | (sampled_candidates, true_expected_count, sampled_expected_count)
|
fixedUnigramCandidateSampler' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | num_sampled |
-> Int64 | num_true |
-> Int64 | range_max |
-> Bool | unique |
-> Tensor v'1 Int64 | true_classes |
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float) | (sampled_candidates, true_expected_count, sampled_expected_count)
|
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | writer |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | writer |
-> m' ControlNode |
:: OneOf '[Float] t | |
=> Tensor v'1 t | x |
-> Tensor v'2 t | scale |
-> Tensor v'3 t | offset |
-> Tensor v'4 t | mean |
-> Tensor v'5 t | variance |
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t) | (y, batch_mean, batch_variance, reserve_space_1, reserve_space_2)
|
:: OneOf '[Float] t | |
=> OpParams | |
-> Tensor v'1 t | x |
-> Tensor v'2 t | scale |
-> Tensor v'3 t | offset |
-> Tensor v'4 t | mean |
-> Tensor v'5 t | variance |
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t) | (y, batch_mean, batch_variance, reserve_space_1, reserve_space_2)
|
:: OneOf '[Float] t | |
=> Tensor v'1 t | y_backprop |
-> Tensor v'2 t | x |
-> Tensor v'3 t | scale |
-> Tensor v'4 t | reserve_space_1 |
-> Tensor v'5 t | reserve_space_2 |
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t) | (x_backprop, scale_backprop, offset_backprop, reserve_space_3, reserve_space_4)
|
:: OneOf '[Float] t | |
=> OpParams | |
-> Tensor v'1 t | y_backprop |
-> Tensor v'2 t | x |
-> Tensor v'3 t | scale |
-> Tensor v'4 t | reserve_space_1 |
-> Tensor v'5 t | reserve_space_2 |
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t) | (x_backprop, scale_backprop, offset_backprop, reserve_space_3, reserve_space_4)
|
:: (OneOf '[Word16, Float] t, OneOf '[Float] u) | |
=> Tensor v'1 t | y_backprop |
-> Tensor v'2 t | x |
-> Tensor v'3 Float | scale |
-> Tensor v'4 u | reserve_space_1 |
-> Tensor v'5 u | reserve_space_2 |
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u) | (x_backprop, scale_backprop, offset_backprop, reserve_space_3, reserve_space_4)
|
fusedBatchNormGradV2' Source #
:: (OneOf '[Word16, Float] t, OneOf '[Float] u) | |
=> OpParams | |
-> Tensor v'1 t | y_backprop |
-> Tensor v'2 t | x |
-> Tensor v'3 Float | scale |
-> Tensor v'4 u | reserve_space_1 |
-> Tensor v'5 u | reserve_space_2 |
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u) | (x_backprop, scale_backprop, offset_backprop, reserve_space_3, reserve_space_4)
|
:: (OneOf '[Word16, Float] t, OneOf '[Float] u) | |
=> Tensor v'1 t | x |
-> Tensor v'2 u | scale |
-> Tensor v'3 u | offset |
-> Tensor v'4 u | mean |
-> Tensor v'5 u | variance |
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u) | (y, batch_mean, batch_variance, reserve_space_1, reserve_space_2)
|
:: (OneOf '[Word16, Float] t, OneOf '[Float] u) | |
=> OpParams | |
-> Tensor v'1 t | x |
-> Tensor v'2 u | scale |
-> Tensor v'3 u | offset |
-> Tensor v'4 u | mean |
-> Tensor v'5 u | variance |
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u) | (y, batch_mean, batch_variance, reserve_space_1, reserve_space_2)
|
gcsConfigureBlockCache Source #
:: MonadBuild m' | |
=> Tensor v'1 Word64 | max_cache_size |
-> Tensor v'2 Word64 | block_size |
-> Tensor v'3 Word64 | max_staleness |
-> m' ControlNode |
gcsConfigureBlockCache' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 Word64 | max_cache_size |
-> Tensor v'2 Word64 | block_size |
-> Tensor v'3 Word64 | max_staleness |
-> m' ControlNode |
gcsConfigureCredentials Source #
:: MonadBuild m' | |
=> Tensor v'1 ByteString | json |
-> m' ControlNode |
gcsConfigureCredentials' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | json |
-> m' ControlNode |
generateBigQueryReaderPartitions Source #
:: Int64 | num_partitions: Number of partitions to split the table into. |
-> Int64 | timestamp_millis: Table snapshot timestamp in millis since epoch. Relative (negative or zero) snapshot times are not allowed. For more details, see 'Table Decorators' in BigQuery docs. |
-> Tensor Build ByteString | partitions: Serialized table partitions. |
Generates serialized partition messages suitable for batch reads.
This op should not be used directly by clients. Instead, the bigquery_reader_ops.py file defines a clean interface to the reader.
generateBigQueryReaderPartitions' Source #
:: OpParams | |
-> Int64 | num_partitions: Number of partitions to split the table into. |
-> Int64 | timestamp_millis: Table snapshot timestamp in millis since epoch. Relative (negative or zero) snapshot times are not allowed. For more details, see 'Table Decorators' in BigQuery docs. |
-> Tensor Build ByteString | partitions: Serialized table partitions. |
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 t | value |
-> m' (Tensor Value ByteString) | handle |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor v'1 t | value |
-> m' (Tensor Value ByteString) | handle |
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 t | value |
-> m' (Tensor Value ResourceHandle) | handle |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor v'1 t | value |
-> m' (Tensor Value ResourceHandle) | handle |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor v'1 ByteString | handle |
-> m' (Tensor Value dtype) | value |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor v'1 ByteString | handle |
-> m' (Tensor Value dtype) | value |
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 t | input |
-> m' (Tensor Value t) | output |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor v'1 t | input |
-> m' (Tensor Value t) | output |
:: MonadBuild m' | |
=> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Ref ByteString) | table_handle |
:: MonadBuild m' | |
=> OpParams | |
-> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Ref ByteString) | table_handle |
:: MonadBuild m' | |
=> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Value ResourceHandle) | table_handle |
:: MonadBuild m' | |
=> OpParams | |
-> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Value ResourceHandle) | table_handle |
:: TensorTypes t | |
=> TensorList v'1 t | input |
-> TensorList Build t | output |
:: TensorTypes t | |
=> OpParams | |
-> TensorList v'1 t | input |
-> TensorList Build t | output |
:: MonadBuild m' | |
=> m' (Tensor Ref ByteString) | reader_handle |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Ref ByteString) | reader_handle |
:: MonadBuild m' | |
=> m' (Tensor Value ResourceHandle) | reader_handle |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value ResourceHandle) | reader_handle |
:: TensorType dtype | |
=> Shape | shape |
-> Tensor Build dtype | tensor |
:: TensorType dtype | |
=> OpParams | |
-> Shape | shape |
-> Tensor Build dtype | tensor |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 ByteString | event |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 ByteString | event |
-> m' ControlNode |
:: (MonadBuild m', TensorType dtype) | |
=> Shape | shape: The shape of the tensor. |
-> m' (Tensor Value dtype) | output: A tensor that will be provided using the infeed mechanism. |
A placeholder op for a value that will be fed into the computation.
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Shape | shape: The shape of the tensor. |
-> m' (Tensor Value dtype) | output: A tensor that will be provided using the infeed mechanism. |
:: (MonadBuild m', TensorTypes dtypes) | |
=> m' (TensorList Value dtypes) | outputs: A list of tensors that will be provided using the infeed mechanism. |
A placeholder op for multiple values that will be fed into the computation
simultaneously as an XLA tuple.
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> m' (TensorList Value dtypes) | outputs: A list of tensors that will be provided using the infeed mechanism. |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor v'1 dtype | input: A tensor that will be provided using the infeed mechanism. |
-> m' ControlNode |
An op which feeds a single Tensor value into the computation.
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor v'1 dtype | input: A tensor that will be provided using the infeed mechanism. |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes dtypes) | |
=> TensorList v'1 dtypes | inputs: A list of tensors that will be provided using the infeed mechanism. |
-> m' ControlNode |
An op which feeds multiple Tensor values into the computation as an XLA tuple.
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> TensorList v'1 dtypes | inputs: A list of tensors that will be provided using the infeed mechanism. |
-> m' ControlNode |
:: (MonadBuild m', TensorType tkey, TensorType tval) | |
=> Tensor Ref ByteString | table_handle |
-> Tensor v'2 tkey | keys |
-> Tensor v'3 tval | values |
-> m' ControlNode |
:: (MonadBuild m', TensorType tkey, TensorType tval) | |
=> OpParams | |
-> Tensor Ref ByteString | table_handle |
-> Tensor v'2 tkey | keys |
-> Tensor v'3 tval | values |
-> m' ControlNode |
initializeTableFromTextFile Source #
:: MonadBuild m' | |
=> Int64 | key_index |
-> Int64 | value_index |
-> Tensor Ref ByteString | table_handle |
-> Tensor v'2 ByteString | filename |
-> m' ControlNode |
initializeTableFromTextFile' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | key_index |
-> Int64 | value_index |
-> Tensor Ref ByteString | table_handle |
-> Tensor v'2 ByteString | filename |
-> m' ControlNode |
initializeTableFromTextFileV2 Source #
:: MonadBuild m' | |
=> Int64 | key_index |
-> Int64 | value_index |
-> Tensor v'1 ResourceHandle | table_handle |
-> Tensor v'2 ByteString | filename |
-> m' ControlNode |
initializeTableFromTextFileV2' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | key_index |
-> Int64 | value_index |
-> Tensor v'1 ResourceHandle | table_handle |
-> Tensor v'2 ByteString | filename |
-> m' ControlNode |
:: (MonadBuild m', TensorType tkey, TensorType tval) | |
=> Tensor v'1 ResourceHandle | table_handle |
-> Tensor v'2 tkey | keys |
-> Tensor v'3 tval | values |
-> m' ControlNode |
:: (MonadBuild m', TensorType tkey, TensorType tval) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | table_handle |
-> Tensor v'2 tkey | keys |
-> Tensor v'3 tval | values |
-> m' ControlNode |
isBoostedTreesEnsembleInitialized Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> m' (Tensor Value Bool) | is_initialized |
isBoostedTreesEnsembleInitialized' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | tree_ensemble_handle |
-> m' (Tensor Value Bool) | is_initialized |
isVariableInitialized Source #
:: (MonadBuild m', TensorType dtype) | |
=> Tensor Ref dtype | ref |
-> m' (Tensor Value Bool) | is_initialized |
isVariableInitialized' Source #
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor Ref dtype | ref |
-> m' (Tensor Value Bool) | is_initialized |
:: MonadBuild m' | |
=> [DataType] | output_types |
-> m' (Tensor Value ResourceHandle) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | output_types |
-> m' (Tensor Value ResourceHandle) | handle |
iteratorFromStringHandle Source #
:: MonadBuild m' | |
=> Tensor v'1 ByteString | string_handle |
-> m' (Tensor Value ResourceHandle) | resource_handle |
iteratorFromStringHandle' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | string_handle |
-> m' (Tensor Value ResourceHandle) | resource_handle |
:: (MonadBuild m', TensorTypes output_types) | |
=> Tensor v'1 ResourceHandle | iterator |
-> m' (TensorList Value output_types) | components |
:: (MonadBuild m', TensorTypes output_types) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | iterator |
-> m' (TensorList Value output_types) | components |
:: (MonadBuild m', TensorTypes output_types) | |
=> Tensor v'1 ResourceHandle | iterator |
-> m' (TensorList Value output_types) | components |
:: (MonadBuild m', TensorTypes output_types) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | iterator |
-> m' (TensorList Value output_types) | components |
iteratorToStringHandle Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | resource_handle |
-> m' (Tensor Value ByteString) | string_handle |
iteratorToStringHandle' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource_handle |
-> m' (Tensor Value ByteString) | string_handle |
:: MonadBuild m' | |
=> m' (Tensor Ref ByteString) | reader_handle |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Ref ByteString) | reader_handle |
learnedUnigramCandidateSampler Source #
:: MonadBuild m' | |
=> Int64 | num_sampled |
-> Int64 | num_true |
-> Int64 | range_max |
-> Bool | unique |
-> Tensor v'1 Int64 | true_classes |
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float) | (sampled_candidates, true_expected_count, sampled_expected_count)
|
learnedUnigramCandidateSampler' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | num_sampled |
-> Int64 | num_true |
-> Int64 | range_max |
-> Bool | unique |
-> Tensor v'1 Int64 | true_classes |
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float) | (sampled_candidates, true_expected_count, sampled_expected_count)
|
logUniformCandidateSampler Source #
:: MonadBuild m' | |
=> Int64 | num_sampled |
-> Int64 | num_true |
-> Int64 | range_max |
-> Bool | unique |
-> Tensor v'1 Int64 | true_classes |
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float) | (sampled_candidates, true_expected_count, sampled_expected_count)
|
logUniformCandidateSampler' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | num_sampled |
-> Int64 | num_true |
-> Int64 | range_max |
-> Bool | unique |
-> Tensor v'1 Int64 | true_classes |
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float) | (sampled_candidates, true_expected_count, sampled_expected_count)
|
:: (MonadBuild m', TensorType tkeys, TensorType tvalues) | |
=> Tensor Ref ByteString | table_handle |
-> m' (Tensor Value tkeys, Tensor Value tvalues) | (keys, values)
|
:: (MonadBuild m', TensorType tkeys, TensorType tvalues) | |
=> OpParams | |
-> Tensor Ref ByteString | table_handle |
-> m' (Tensor Value tkeys, Tensor Value tvalues) | (keys, values)
|
:: (MonadBuild m', TensorType tkeys, TensorType tvalues) | |
=> Tensor v'1 ResourceHandle | table_handle |
-> m' (Tensor Value tkeys, Tensor Value tvalues) | (keys, values)
|
:: (MonadBuild m', TensorType tkeys, TensorType tvalues) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | table_handle |
-> m' (Tensor Value tkeys, Tensor Value tvalues) | (keys, values)
|
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> Tensor Ref ByteString | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | default_value |
-> m' (Tensor Value tout) | values |
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> OpParams | |
-> Tensor Ref ByteString | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | default_value |
-> m' (Tensor Value tout) | values |
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> Tensor v'1 ResourceHandle | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | default_value |
-> m' (Tensor Value tout) | values |
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | default_value |
-> m' (Tensor Value tout) | values |
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> Tensor Ref ByteString | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | values |
-> m' ControlNode |
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> OpParams | |
-> Tensor Ref ByteString | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | values |
-> m' ControlNode |
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> Tensor v'1 ResourceHandle | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | values |
-> m' ControlNode |
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | values |
-> m' ControlNode |
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> Tensor Ref ByteString | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | values |
-> m' ControlNode |
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> OpParams | |
-> Tensor Ref ByteString | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | values |
-> m' ControlNode |
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> Tensor v'1 ResourceHandle | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | values |
-> m' ControlNode |
:: (MonadBuild m', TensorType tin, TensorType tout) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | table_handle |
-> Tensor v'2 tin | keys |
-> Tensor v'3 tout | values |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor Ref ByteString | table_handle |
-> m' (Tensor Value Int64) | size |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | table_handle |
-> m' (Tensor Value Int64) | size |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | table_handle |
-> m' (Tensor Value Int64) | size |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | table_handle |
-> m' (Tensor Value Int64) | size |
:: MonadBuild m' | |
=> Tensor v'1 Variant | dataset |
-> Tensor v'2 ResourceHandle | iterator |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 Variant | dataset |
-> Tensor v'2 ResourceHandle | iterator |
-> m' ControlNode |
:: MonadBuild m' | |
=> [DataType] | dtypes |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | dtypes |
-> m' ControlNode |
:: MonadBuild m' | |
=> [DataType] | dtypes |
-> m' (Tensor Value Int32) | size |
:: (MonadBuild m', TensorTypes dtypes) | |
=> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> m' (TensorList Value dtypes) | values |
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> m' (TensorList Value dtypes) | values |
:: (MonadBuild m', TensorTypes fake_dtypes) | |
=> [DataType] | dtypes |
-> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> TensorList v'3 fake_dtypes | values |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes fake_dtypes) | |
=> OpParams | |
-> [DataType] | dtypes |
-> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> TensorList v'3 fake_dtypes | values |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes dtypes) | |
=> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> m' (TensorList Value dtypes) | values |
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> m' (TensorList Value dtypes) | values |
:: (MonadBuild m', TensorTypes dtypes) | |
=> Tensor v'1 Int32 | indices |
-> m' (Tensor Value Int64, TensorList Value dtypes) | (key, values)
|
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> Tensor v'1 Int32 | indices |
-> m' (Tensor Value Int64, TensorList Value dtypes) | (key, values)
|
:: Tensor v'1 ByteString | pattern |
-> Tensor Build ByteString | filenames |
:: OpParams | |
-> Tensor v'1 ByteString | pattern |
-> Tensor Build ByteString | filenames |
:: TensorType t | |
=> Tensor v'1 t | diagonal |
-> Tensor Build t | output |
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | diagonal |
-> Tensor Build t | output |
:: TensorType t | |
=> Tensor v'1 t | input |
-> Tensor Build t | diagonal |
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | input |
-> Tensor Build t | diagonal |
:: TensorType t | |
=> Tensor v'1 t | input |
-> Tensor v'2 t | diagonal |
-> Tensor Build t | output |
:: [Tensor v'1 ByteString] | inputs |
-> Tensor Build ByteString | summary |
:: OpParams | |
-> [Tensor v'1 ByteString] | inputs |
-> Tensor Build ByteString | summary |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | checkpoint_prefixes |
-> Tensor v'2 ByteString | destination_prefix |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | checkpoint_prefixes |
-> Tensor v'2 ByteString | destination_prefix |
-> m' ControlNode |
mutableDenseHashTable Source #
:: (MonadBuild m', TensorType key_dtype) | |
=> DataType | value_dtype |
-> Tensor v'1 key_dtype | empty_key |
-> m' (Tensor Ref ByteString) | table_handle |
mutableDenseHashTable' Source #
:: (MonadBuild m', TensorType key_dtype) | |
=> OpParams | |
-> DataType | value_dtype |
-> Tensor v'1 key_dtype | empty_key |
-> m' (Tensor Ref ByteString) | table_handle |
mutableDenseHashTableV2 Source #
:: (MonadBuild m', TensorType key_dtype) | |
=> DataType | value_dtype |
-> Tensor v'1 key_dtype | empty_key |
-> m' (Tensor Value ResourceHandle) | table_handle |
mutableDenseHashTableV2' Source #
:: (MonadBuild m', TensorType key_dtype) | |
=> OpParams | |
-> DataType | value_dtype |
-> Tensor v'1 key_dtype | empty_key |
-> m' (Tensor Value ResourceHandle) | table_handle |
:: MonadBuild m' | |
=> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Ref ByteString) | table_handle |
:: MonadBuild m' | |
=> OpParams | |
-> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Ref ByteString) | table_handle |
mutableHashTableOfTensors Source #
:: MonadBuild m' | |
=> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Ref ByteString) | table_handle |
mutableHashTableOfTensors' Source #
:: MonadBuild m' | |
=> OpParams | |
-> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Ref ByteString) | table_handle |
mutableHashTableOfTensorsV2 Source #
:: MonadBuild m' | |
=> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Value ResourceHandle) | table_handle |
mutableHashTableOfTensorsV2' Source #
:: MonadBuild m' | |
=> OpParams | |
-> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Value ResourceHandle) | table_handle |
:: MonadBuild m' | |
=> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Value ResourceHandle) | table_handle |
:: MonadBuild m' | |
=> OpParams | |
-> DataType | key_dtype |
-> DataType | value_dtype |
-> m' (Tensor Value ResourceHandle) | table_handle |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | mutex |
-> m' (Tensor Value Variant) | mutex_lock |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | mutex |
-> m' (Tensor Value Variant) | mutex_lock |
:: MonadBuild m' | |
=> m' (Tensor Value ResourceHandle) | resource |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value ResourceHandle) | resource |
:: TensorType t | |
=> Tensor v'1 t | data |
-> Tensor Build t | output |
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | data |
-> Tensor Build t | output |
noOp :: forall m'. MonadBuild m' => m' ControlNode Source #
noOp' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode Source #
:: MonadBuild m' | |
=> [DataType] | dtypes |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | dtypes |
-> m' ControlNode |
orderedMapIncompleteSize Source #
:: MonadBuild m' | |
=> [DataType] | dtypes |
-> m' (Tensor Value Int32) | size |
:: (MonadBuild m', TensorTypes dtypes) | |
=> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> m' (TensorList Value dtypes) | values |
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> m' (TensorList Value dtypes) | values |
:: MonadBuild m' | |
=> [DataType] | dtypes |
-> m' (Tensor Value Int32) | size |
:: (MonadBuild m', TensorTypes fake_dtypes) | |
=> [DataType] | dtypes |
-> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> TensorList v'3 fake_dtypes | values |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes fake_dtypes) | |
=> OpParams | |
-> [DataType] | dtypes |
-> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> TensorList v'3 fake_dtypes | values |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes dtypes) | |
=> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> m' (TensorList Value dtypes) | values |
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> Tensor v'1 Int64 | key |
-> Tensor v'2 Int32 | indices |
-> m' (TensorList Value dtypes) | values |
orderedMapUnstageNoKey Source #
:: (MonadBuild m', TensorTypes dtypes) | |
=> Tensor v'1 Int32 | indices |
-> m' (Tensor Value Int64, TensorList Value dtypes) | (key, values)
|
orderedMapUnstageNoKey' Source #
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> Tensor v'1 Int32 | indices |
-> m' (Tensor Value Int64, TensorList Value dtypes) | (key, values)
|
:: (MonadBuild m', TensorType dtype) | |
=> Shape | shape: The shape of the tensor. |
-> m' (Tensor Value dtype) | output: A tensor that will be read from the device outfeed. |
Retrieves a single tensor from the computation outfeed. This operation will
block indefinitely until data is available.
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Shape | shape: The shape of the tensor. |
-> m' (Tensor Value dtype) | output: A tensor that will be read from the device outfeed. |
:: (MonadBuild m', TensorTypes dtypes) | |
=> m' (TensorList Value dtypes) | outputs: A list of tensors that will be read from the outfeed. |
Retrieve multiple values that will be emitted by the computation as an XLA
tuple. This operations will block indefinitely until data is available.
Output i
corresponds to XLA tuple element i
.
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> m' (TensorList Value dtypes) | outputs: A list of tensors that will be read from the outfeed. |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor v'1 dtype | input: A tensor that will be inserted into the outfeed queue. |
-> m' ControlNode |
An op which emits a single Tensor value from an XLA computation.
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor v'1 dtype | input: A tensor that will be inserted into the outfeed queue. |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes dtypes) | |
=> TensorList v'1 dtypes | inputs: A list of tensors that will be inserted into the outfeed queue as an XLA tuple. |
-> m' ControlNode |
An op which emits multiple Tensor values from an XLA computation.
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> TensorList v'1 dtypes | inputs: A list of tensors that will be inserted into the outfeed queue as an XLA tuple. |
-> m' ControlNode |
:: MonadBuild m' | |
=> [DataType] | component_types |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | component_types |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> [DataType] | component_types |
-> m' (Tensor Value ResourceHandle) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | component_types |
-> m' (Tensor Value ResourceHandle) | handle |
:: TensorType t | |
=> Shape | shape |
-> [Tensor v'1 t] | values |
-> Tensor Build t | output |
:: (OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] tdense) | |
=> Tensor v'1 ByteString | serialized |
-> Tensor v'2 ByteString | names |
-> [Tensor v'3 ByteString] | sparse_keys |
-> [Tensor v'4 ByteString] | dense_keys |
-> TensorList v'5 tdense | dense_defaults |
-> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense) | (sparse_indices, sparse_values, sparse_shapes, dense_values)
|
:: (OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] tdense) | |
=> OpParams | |
-> Tensor v'1 ByteString | serialized |
-> Tensor v'2 ByteString | names |
-> [Tensor v'3 ByteString] | sparse_keys |
-> [Tensor v'4 ByteString] | dense_keys |
-> TensorList v'5 tdense | dense_defaults |
-> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense) | (sparse_indices, sparse_values, sparse_shapes, dense_values)
|
:: (OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] tdense) | |
=> Int64 | num_sparse |
-> Tensor v'1 ByteString | serialized |
-> TensorList v'2 tdense | dense_defaults |
-> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense) | (sparse_indices, sparse_values, sparse_shapes, dense_values)
|
:: (OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] tdense) | |
=> OpParams | |
-> Int64 | num_sparse |
-> Tensor v'1 ByteString | serialized |
-> TensorList v'2 tdense | dense_defaults |
-> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense) | (sparse_indices, sparse_values, sparse_shapes, dense_values)
|
parseSingleSequenceExample Source #
:: (OneOfs '[ByteString, Int64, Float] context_sparse_types, OneOfs '[ByteString, Int64, Float] tcontext_dense, OneOfs '[ByteString, Int64, Float] feature_list_dense_types, OneOfs '[ByteString, Int64, Float] feature_list_sparse_types) | |
=> Tensor v'1 ByteString | serialized |
-> Tensor v'2 ByteString | feature_list_dense_missing_assumed_empty |
-> [Tensor v'3 ByteString] | context_sparse_keys |
-> [Tensor v'4 ByteString] | context_dense_keys |
-> [Tensor v'5 ByteString] | feature_list_sparse_keys |
-> [Tensor v'6 ByteString] | feature_list_dense_keys |
-> TensorList v'7 tcontext_dense | context_dense_defaults |
-> Tensor v'8 ByteString | debug_name |
-> ([Tensor Build Int64], TensorList Build context_sparse_types, [Tensor Build Int64], TensorList Build tcontext_dense, [Tensor Build Int64], TensorList Build feature_list_sparse_types, [Tensor Build Int64], TensorList Build feature_list_dense_types) | (context_sparse_indices, context_sparse_values, context_sparse_shapes, context_dense_values, feature_list_sparse_indices, feature_list_sparse_values, feature_list_sparse_shapes, feature_list_dense_values)
|
parseSingleSequenceExample' Source #
:: (OneOfs '[ByteString, Int64, Float] context_sparse_types, OneOfs '[ByteString, Int64, Float] tcontext_dense, OneOfs '[ByteString, Int64, Float] feature_list_dense_types, OneOfs '[ByteString, Int64, Float] feature_list_sparse_types) | |
=> OpParams | |
-> Tensor v'1 ByteString | serialized |
-> Tensor v'2 ByteString | feature_list_dense_missing_assumed_empty |
-> [Tensor v'3 ByteString] | context_sparse_keys |
-> [Tensor v'4 ByteString] | context_dense_keys |
-> [Tensor v'5 ByteString] | feature_list_sparse_keys |
-> [Tensor v'6 ByteString] | feature_list_dense_keys |
-> TensorList v'7 tcontext_dense | context_dense_defaults |
-> Tensor v'8 ByteString | debug_name |
-> ([Tensor Build Int64], TensorList Build context_sparse_types, [Tensor Build Int64], TensorList Build tcontext_dense, [Tensor Build Int64], TensorList Build feature_list_sparse_types, [Tensor Build Int64], TensorList Build feature_list_dense_types) | (context_sparse_indices, context_sparse_values, context_sparse_shapes, context_dense_values, feature_list_sparse_indices, feature_list_sparse_values, feature_list_sparse_shapes, feature_list_dense_values)
|
:: TensorType out_type | |
=> Tensor v'1 ByteString | serialized |
-> Tensor Build out_type | output |
:: TensorType out_type | |
=> OpParams | |
-> Tensor v'1 ByteString | serialized |
-> Tensor Build out_type | output |
:: TensorType dtype | |
=> Tensor Build dtype | output |
:: TensorType dtype | |
=> OpParams | |
-> Tensor Build dtype | output |
:: TensorType dtype | |
=> Shape | shape |
-> Tensor Build dtype | output |
:: TensorType dtype | |
=> OpParams | |
-> Shape | shape |
-> Tensor Build dtype | output |
placeholderWithDefault Source #
:: TensorType dtype | |
=> Shape | shape |
-> Tensor v'1 dtype | input |
-> Tensor Build dtype | output |
:: TensorType t | |
=> Tensor v'1 t | input |
-> Tensor Build t | output |
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | input |
-> Tensor Build t | output |
:: (MonadBuild m', TensorType t, TensorTypes u) | |
=> Tensor v'1 t | input |
-> TensorList v'2 u | data |
-> m' (Tensor Value t) | output |
:: (MonadBuild m', TensorType t, TensorTypes u) | |
=> OpParams | |
-> Tensor v'1 t | input |
-> TensorList v'2 u | data |
-> m' (Tensor Value t) | output |
:: MonadBuild m' | |
=> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> m' (Tensor Value ResourceHandle) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value ResourceHandle) | handle |
quantizeDownAndShrinkRange Source #
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> Tensor v'1 tinput | input |
-> Tensor v'2 Float | input_min |
-> Tensor v'3 Float | input_max |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (output, output_min, output_max)
|
quantizeDownAndShrinkRange' Source #
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> OpParams | |
-> Tensor v'1 tinput | input |
-> Tensor v'2 Float | input_min |
-> Tensor v'3 Float | input_max |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (output, output_min, output_max)
|
:: (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) | |
=> Tensor v'1 t1 | x |
-> Tensor v'2 t2 | y |
-> Tensor v'3 Float | min_x |
-> Tensor v'4 Float | max_x |
-> Tensor v'5 Float | min_y |
-> Tensor v'6 Float | max_y |
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float) | (z, min_z, max_z)
|
:: (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) | |
=> OpParams | |
-> Tensor v'1 t1 | x |
-> Tensor v'2 t2 | y |
-> Tensor v'3 Float | min_x |
-> Tensor v'4 Float | max_x |
-> Tensor v'5 Float | min_y |
-> Tensor v'6 Float | max_y |
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float) | (z, min_z, max_z)
|
quantizedBatchNormWithGlobalNormalization Source #
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> Bool | scale_after_normalization |
-> Float | variance_epsilon |
-> Tensor v'1 tinput | t |
-> Tensor v'2 Float | t_min |
-> Tensor v'3 Float | t_max |
-> Tensor v'4 tinput | m |
-> Tensor v'5 Float | m_min |
-> Tensor v'6 Float | m_max |
-> Tensor v'7 tinput | v |
-> Tensor v'8 Float | v_min |
-> Tensor v'9 Float | v_max |
-> Tensor v'10 tinput | beta |
-> Tensor v'11 Float | beta_min |
-> Tensor v'12 Float | beta_max |
-> Tensor v'13 tinput | gamma |
-> Tensor v'14 Float | gamma_min |
-> Tensor v'15 Float | gamma_max |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (result, result_min, result_max)
|
quantizedBatchNormWithGlobalNormalization' Source #
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> OpParams | |
-> Bool | scale_after_normalization |
-> Float | variance_epsilon |
-> Tensor v'1 tinput | t |
-> Tensor v'2 Float | t_min |
-> Tensor v'3 Float | t_max |
-> Tensor v'4 tinput | m |
-> Tensor v'5 Float | m_min |
-> Tensor v'6 Float | m_max |
-> Tensor v'7 tinput | v |
-> Tensor v'8 Float | v_min |
-> Tensor v'9 Float | v_max |
-> Tensor v'10 tinput | beta |
-> Tensor v'11 Float | beta_min |
-> Tensor v'12 Float | beta_max |
-> Tensor v'13 tinput | gamma |
-> Tensor v'14 Float | gamma_min |
-> Tensor v'15 Float | gamma_max |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (result, result_min, result_max)
|
:: (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> Tensor v'1 t1 | input |
-> Tensor v'2 t2 | bias |
-> Tensor v'3 Float | min_input |
-> Tensor v'4 Float | max_input |
-> Tensor v'5 Float | min_bias |
-> Tensor v'6 Float | max_bias |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (output, min_out, max_out)
|
:: (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> OpParams | |
-> Tensor v'1 t1 | input |
-> Tensor v'2 t2 | bias |
-> Tensor v'3 Float | min_input |
-> Tensor v'4 Float | max_input |
-> Tensor v'5 Float | min_bias |
-> Tensor v'6 Float | max_bias |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (output, min_out, max_out)
|
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> Tensor v'1 tinput | input |
-> Tensor v'2 tfilter | filter |
-> Tensor v'3 Float | min_input |
-> Tensor v'4 Float | max_input |
-> Tensor v'5 Float | min_filter |
-> Tensor v'6 Float | max_filter |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (output, min_output, max_output)
|
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> OpParams | |
-> Tensor v'1 tinput | input |
-> Tensor v'2 tfilter | filter |
-> Tensor v'3 Float | min_input |
-> Tensor v'4 Float | max_input |
-> Tensor v'5 Float | min_filter |
-> Tensor v'6 Float | max_filter |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (output, min_output, max_output)
|
:: (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) | |
=> Tensor v'1 t1 | a |
-> Tensor v'2 t2 | b |
-> Tensor v'3 Float | min_a |
-> Tensor v'4 Float | max_a |
-> Tensor v'5 Float | min_b |
-> Tensor v'6 Float | max_b |
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float) | (out, min_out, max_out)
|
:: (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) | |
=> OpParams | |
-> Tensor v'1 t1 | a |
-> Tensor v'2 t2 | b |
-> Tensor v'3 Float | min_a |
-> Tensor v'4 Float | max_a |
-> Tensor v'5 Float | min_b |
-> Tensor v'6 Float | max_b |
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float) | (out, min_out, max_out)
|
:: (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) | |
=> Tensor v'1 t1 | x |
-> Tensor v'2 t2 | y |
-> Tensor v'3 Float | min_x |
-> Tensor v'4 Float | max_x |
-> Tensor v'5 Float | min_y |
-> Tensor v'6 Float | max_y |
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float) | (z, min_z, max_z)
|
:: (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] toutput) | |
=> OpParams | |
-> Tensor v'1 t1 | x |
-> Tensor v'2 t2 | y |
-> Tensor v'3 Float | min_x |
-> Tensor v'4 Float | max_x |
-> Tensor v'5 Float | min_y |
-> Tensor v'6 Float | max_y |
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float) | (z, min_z, max_z)
|
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> Tensor v'1 tinput | features |
-> Tensor v'2 Float | min_features |
-> Tensor v'3 Float | max_features |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (activations, min_activations, max_activations)
|
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> OpParams | |
-> Tensor v'1 tinput | features |
-> Tensor v'2 Float | min_features |
-> Tensor v'3 Float | max_features |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (activations, min_activations, max_activations)
|
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> Tensor v'1 tinput | features |
-> Tensor v'2 Float | min_features |
-> Tensor v'3 Float | max_features |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (activations, min_activations, max_activations)
|
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> OpParams | |
-> Tensor v'1 tinput | features |
-> Tensor v'2 Float | min_features |
-> Tensor v'3 Float | max_features |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (activations, min_activations, max_activations)
|
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> Tensor v'1 tinput | features |
-> Tensor v'2 Float | max_value |
-> Tensor v'3 Float | min_features |
-> Tensor v'4 Float | max_features |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (activations, min_activations, max_activations)
|
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> OpParams | |
-> Tensor v'1 tinput | features |
-> Tensor v'2 Float | max_value |
-> Tensor v'3 Float | min_features |
-> Tensor v'4 Float | max_features |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (activations, min_activations, max_activations)
|
:: MonadBuild m' | |
=> Tensor Ref ByteString | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes component_types) | |
=> Tensor Ref ByteString | handle |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes component_types) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes component_types) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | n |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes component_types) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | n |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes component_types) | |
=> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | n |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes component_types) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | n |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes component_types) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | n |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes component_types) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | n |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes component_types) | |
=> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | n |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes component_types) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | n |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes component_types) | |
=> Tensor v'1 ResourceHandle | handle |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes component_types) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> m' (TensorList Value component_types) | components |
:: (MonadBuild m', TensorTypes tcomponents) | |
=> Tensor Ref ByteString | handle |
-> TensorList v'2 tcomponents | components |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes tcomponents) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> TensorList v'2 tcomponents | components |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes tcomponents) | |
=> Tensor Ref ByteString | handle |
-> TensorList v'2 tcomponents | components |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes tcomponents) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> TensorList v'2 tcomponents | components |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes tcomponents) | |
=> Tensor v'1 ResourceHandle | handle |
-> TensorList v'2 tcomponents | components |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes tcomponents) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> TensorList v'2 tcomponents | components |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes tcomponents) | |
=> Tensor v'1 ResourceHandle | handle |
-> TensorList v'2 tcomponents | components |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes tcomponents) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> TensorList v'2 tcomponents | components |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor Ref ByteString | handle |
-> m' (Tensor Value Bool) | is_closed |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> m' (Tensor Value Bool) | is_closed |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | handle |
-> m' (Tensor Value Bool) | is_closed |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> m' (Tensor Value Bool) | is_closed |
:: MonadBuild m' | |
=> Tensor Ref ByteString | handle |
-> m' (Tensor Value Int32) | size |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> m' (Tensor Value Int32) | size |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | handle |
-> m' (Tensor Value Int32) | size |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> m' (Tensor Value Int32) | size |
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 t | value |
-> m' (Tensor Value t) | output |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor v'1 t | value |
-> m' (Tensor Value t) | output |
:: MonadBuild m' | |
=> [DataType] | component_types |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | component_types |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> [DataType] | component_types |
-> m' (Tensor Value ResourceHandle) | handle |
randomShuffleQueueV2' Source #
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | component_types |
-> m' (Tensor Value ResourceHandle) | handle |
:: Tensor v'1 ByteString | filename |
-> Tensor Build ByteString | contents |
:: OpParams | |
-> Tensor v'1 ByteString | filename |
-> Tensor Build ByteString | contents |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor v'1 ResourceHandle | resource |
-> m' (Tensor Value dtype) | value |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource |
-> m' (Tensor Value dtype) | value |
readerNumRecordsProduced Source #
:: MonadBuild m' | |
=> Tensor Ref ByteString | reader_handle |
-> m' (Tensor Value Int64) | records_produced |
readerNumRecordsProduced' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | reader_handle |
-> m' (Tensor Value Int64) | records_produced |
readerNumRecordsProducedV2 Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | reader_handle |
-> m' (Tensor Value Int64) | records_produced |
readerNumRecordsProducedV2' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | reader_handle |
-> m' (Tensor Value Int64) | records_produced |
readerNumWorkUnitsCompleted Source #
:: MonadBuild m' | |
=> Tensor Ref ByteString | reader_handle |
-> m' (Tensor Value Int64) | units_completed |
readerNumWorkUnitsCompleted' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | reader_handle |
-> m' (Tensor Value Int64) | units_completed |
readerNumWorkUnitsCompletedV2 Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | reader_handle |
-> m' (Tensor Value Int64) | units_completed |
readerNumWorkUnitsCompletedV2' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | reader_handle |
-> m' (Tensor Value Int64) | units_completed |
:: MonadBuild m' | |
=> Tensor Ref ByteString | reader_handle |
-> Tensor Ref ByteString | queue_handle |
-> m' (Tensor Value ByteString, Tensor Value ByteString) | (key, value)
|
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | reader_handle |
-> Tensor Ref ByteString | queue_handle |
-> m' (Tensor Value ByteString, Tensor Value ByteString) | (key, value)
|
:: MonadBuild m' | |
=> Tensor Ref ByteString | reader_handle |
-> Tensor Ref ByteString | queue_handle |
-> Tensor v'3 Int64 | num_records |
-> m' (Tensor Value ByteString, Tensor Value ByteString) | (keys, values)
|
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | reader_handle |
-> Tensor Ref ByteString | queue_handle |
-> Tensor v'3 Int64 | num_records |
-> m' (Tensor Value ByteString, Tensor Value ByteString) | (keys, values)
|
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | reader_handle |
-> Tensor v'2 ResourceHandle | queue_handle |
-> Tensor v'3 Int64 | num_records |
-> m' (Tensor Value ByteString, Tensor Value ByteString) | (keys, values)
|
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | reader_handle |
-> Tensor v'2 ResourceHandle | queue_handle |
-> Tensor v'3 Int64 | num_records |
-> m' (Tensor Value ByteString, Tensor Value ByteString) | (keys, values)
|
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | reader_handle |
-> Tensor v'2 ResourceHandle | queue_handle |
-> m' (Tensor Value ByteString, Tensor Value ByteString) | (key, value)
|
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | reader_handle |
-> Tensor v'2 ResourceHandle | queue_handle |
-> m' (Tensor Value ByteString, Tensor Value ByteString) | (key, value)
|
:: MonadBuild m' | |
=> Tensor Ref ByteString | reader_handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | reader_handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | reader_handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | reader_handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor Ref ByteString | reader_handle |
-> Tensor v'2 ByteString | state |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | reader_handle |
-> Tensor v'2 ByteString | state |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | reader_handle |
-> Tensor v'2 ByteString | state |
-> m' ControlNode |
readerRestoreStateV2' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | reader_handle |
-> Tensor v'2 ByteString | state |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor Ref ByteString | reader_handle |
-> m' (Tensor Value ByteString) | state |
readerSerializeState' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | reader_handle |
-> m' (Tensor Value ByteString) | state |
readerSerializeStateV2 Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | reader_handle |
-> m' (Tensor Value ByteString) | state |
readerSerializeStateV2' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | reader_handle |
-> m' (Tensor Value ByteString) | state |
:: MonadBuild m' | |
=> m' (Tensor Value ByteString) | records |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value ByteString) | records |
:: Tensor v'1 ByteString | inputs |
-> Tensor v'2 Int32 | reduction_indices |
-> Tensor Build ByteString | output |
:: OpParams | |
-> Tensor v'1 ByteString | inputs |
-> Tensor v'2 Int32 | reduction_indices |
-> Tensor Build ByteString | output |
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref t | data |
-> m' (Tensor Ref t) | output |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref t | data |
-> m' (Tensor Ref t) | output |
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref t | data |
-> m' (Tensor Ref t) | output |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref t | data |
-> m' (Tensor Ref t) | output |
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref t | input |
-> m' (Tensor Ref t) | output |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref t | input |
-> m' (Tensor Ref t) | output |
:: (MonadBuild m', TensorType t) | |
=> [Tensor Ref t] | inputs |
-> m' (Tensor Ref t, Tensor Value Int32) | (output, value_index)
|
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> [Tensor Ref t] | inputs |
-> m' (Tensor Ref t, Tensor Value Int32) | (output, value_index)
|
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref t | data |
-> m' (Tensor Ref t) | output |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref t | data |
-> m' (Tensor Ref t) | output |
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 Int32 | index |
-> [Tensor Ref t] | inputs |
-> m' (Tensor Ref t) | output |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor v'1 Int32 | index |
-> [Tensor Ref t] | inputs |
-> m' (Tensor Ref t) | output |
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref t | data |
-> Tensor v'2 Bool | pred |
-> m' (Tensor Ref t, Tensor Ref t) | (output_false, output_true)
|
:: Tensor v'1 ByteString | input |
-> Tensor v'2 ByteString | pattern |
-> Tensor Build Bool | output |
:: OpParams | |
-> Tensor v'1 ByteString | input |
-> Tensor v'2 ByteString | pattern |
-> Tensor Build Bool | output |
:: Tensor v'1 ByteString | input |
-> Tensor v'2 ByteString | pattern |
-> Tensor v'3 ByteString | rewrite |
-> Tensor Build ByteString | output |
:: OpParams | |
-> Tensor v'1 ByteString | input |
-> Tensor v'2 ByteString | pattern |
-> Tensor v'3 ByteString | rewrite |
-> Tensor Build ByteString | output |
remoteFusedGraphExecute Source #
:: (TensorTypes tinputs, TensorTypes toutputs) | |
=> TensorList v'1 tinputs | inputs |
-> TensorList Build toutputs | outputs |
remoteFusedGraphExecute' Source #
:: (TensorTypes tinputs, TensorTypes toutputs) | |
=> OpParams | |
-> TensorList v'1 tinputs | inputs |
-> TensorList Build toutputs | outputs |
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> Tensor v'1 tinput | input |
-> Tensor v'2 Float | input_min |
-> Tensor v'3 Float | input_max |
-> Tensor v'4 Float | requested_output_min |
-> Tensor v'5 Float | requested_output_max |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (output, output_min, output_max)
|
:: (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) | |
=> OpParams | |
-> Tensor v'1 tinput | input |
-> Tensor v'2 Float | input_min |
-> Tensor v'3 Float | input_max |
-> Tensor v'4 Float | requested_output_min |
-> Tensor v'5 Float | requested_output_max |
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float) | (output, output_min, output_max)
|
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | m |
-> Tensor v'3 ResourceHandle | v |
-> Tensor v'4 t | beta1_power |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | beta1 |
-> Tensor v'7 t | beta2 |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | m |
-> Tensor v'3 ResourceHandle | v |
-> Tensor v'4 t | beta1_power |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | beta1 |
-> Tensor v'7 t | beta2 |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> m' ControlNode |
resourceApplyAdadelta Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | accum_update |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | epsilon |
-> Tensor v'7 t | grad |
-> m' ControlNode |
resourceApplyAdadelta' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | accum_update |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | epsilon |
-> Tensor v'7 t | grad |
-> m' ControlNode |
resourceApplyAdagradDA Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | gradient_accumulator |
-> Tensor v'3 ResourceHandle | gradient_squared_accumulator |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 Int64 | global_step |
-> m' ControlNode |
resourceApplyAdagradDA' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | gradient_accumulator |
-> Tensor v'3 ResourceHandle | gradient_squared_accumulator |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 Int64 | global_step |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | m |
-> Tensor v'3 ResourceHandle | v |
-> Tensor v'4 t | beta1_power |
-> Tensor v'5 t | beta2_power |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | beta1 |
-> Tensor v'8 t | beta2 |
-> Tensor v'9 t | epsilon |
-> Tensor v'10 t | grad |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | m |
-> Tensor v'3 ResourceHandle | v |
-> Tensor v'4 t | beta1_power |
-> Tensor v'5 t | beta2_power |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | beta1 |
-> Tensor v'8 t | beta2 |
-> Tensor v'9 t | epsilon |
-> Tensor v'10 t | grad |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | m |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | alpha |
-> Tensor v'5 t | sign_decay |
-> Tensor v'6 t | beta |
-> Tensor v'7 t | grad |
-> m' ControlNode |
resourceApplyAddSign' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | m |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | alpha |
-> Tensor v'5 t | sign_decay |
-> Tensor v'6 t | beta |
-> Tensor v'7 t | grad |
-> m' ControlNode |
resourceApplyCenteredRMSProp Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | mg |
-> Tensor v'3 ResourceHandle | ms |
-> Tensor v'4 ResourceHandle | mom |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | rho |
-> Tensor v'7 t | momentum |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> m' ControlNode |
resourceApplyCenteredRMSProp' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | mg |
-> Tensor v'3 ResourceHandle | ms |
-> Tensor v'4 ResourceHandle | mom |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | rho |
-> Tensor v'7 t | momentum |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 t | lr_power |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 t | lr_power |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 t | l2_shrinkage |
-> Tensor v'9 t | lr_power |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | l1 |
-> Tensor v'7 t | l2 |
-> Tensor v'8 t | l2_shrinkage |
-> Tensor v'9 t | lr_power |
-> m' ControlNode |
resourceApplyPowerSign Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | m |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | logbase |
-> Tensor v'5 t | sign_decay |
-> Tensor v'6 t | beta |
-> Tensor v'7 t | grad |
-> m' ControlNode |
resourceApplyPowerSign' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | m |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | logbase |
-> Tensor v'5 t | sign_decay |
-> Tensor v'6 t | beta |
-> Tensor v'7 t | grad |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | ms |
-> Tensor v'3 ResourceHandle | mom |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | momentum |
-> Tensor v'7 t | epsilon |
-> Tensor v'8 t | grad |
-> m' ControlNode |
resourceApplyRMSProp' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | ms |
-> Tensor v'3 ResourceHandle | mom |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | momentum |
-> Tensor v'7 t | epsilon |
-> Tensor v'8 t | grad |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Int32, Int64] t) | |
=> Int64 | limit |
-> Tensor v'1 ResourceHandle | resource |
-> m' (Tensor Value t) | output |
:: (MonadBuild m', OneOf '[Int32, Int64] t) | |
=> OpParams | |
-> Int64 | limit |
-> Tensor v'1 ResourceHandle | resource |
-> m' (Tensor Value t) | output |
:: (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | resource |
-> Tensor v'2 tindices | indices |
-> m' (Tensor Value dtype) | output |
:: (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource |
-> Tensor v'2 tindices | indices |
-> m' (Tensor Value dtype) | output |
resourceScatterNdUpdate Source #
:: (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | ref |
-> Tensor v'2 tindices | indices |
-> Tensor v'3 t | updates |
-> m' ControlNode |
resourceScatterNdUpdate' Source #
:: (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | ref |
-> Tensor v'2 tindices | indices |
-> Tensor v'3 t | updates |
-> m' ControlNode |
resourceScatterUpdate Source #
:: (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | resource |
-> Tensor v'2 tindices | indices |
-> Tensor v'3 dtype | updates |
-> m' ControlNode |
resourceScatterUpdate' Source #
:: (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource |
-> Tensor v'2 tindices | indices |
-> Tensor v'3 dtype | updates |
-> m' ControlNode |
resourceSparseApplyAdadelta Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | accum_update |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | epsilon |
-> Tensor v'7 t | grad |
-> Tensor v'8 tindices | indices |
-> m' ControlNode |
resourceSparseApplyAdadelta' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | accum_update |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | epsilon |
-> Tensor v'7 t | grad |
-> Tensor v'8 tindices | indices |
-> m' ControlNode |
resourceSparseApplyAdagrad Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> m' ControlNode |
resourceSparseApplyAdagrad' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> m' ControlNode |
resourceSparseApplyAdagradDA Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | gradient_accumulator |
-> Tensor v'3 ResourceHandle | gradient_squared_accumulator |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 Int64 | global_step |
-> m' ControlNode |
resourceSparseApplyAdagradDA' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | gradient_accumulator |
-> Tensor v'3 ResourceHandle | gradient_squared_accumulator |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 Int64 | global_step |
-> m' ControlNode |
resourceSparseApplyCenteredRMSProp Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | mg |
-> Tensor v'3 ResourceHandle | ms |
-> Tensor v'4 ResourceHandle | mom |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | rho |
-> Tensor v'7 t | momentum |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> Tensor v'10 tindices | indices |
-> m' ControlNode |
resourceSparseApplyCenteredRMSProp' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | mg |
-> Tensor v'3 ResourceHandle | ms |
-> Tensor v'4 ResourceHandle | mom |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | rho |
-> Tensor v'7 t | momentum |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> Tensor v'10 tindices | indices |
-> m' ControlNode |
resourceSparseApplyFtrl Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 t | lr_power |
-> m' ControlNode |
resourceSparseApplyFtrl' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 t | lr_power |
-> m' ControlNode |
resourceSparseApplyFtrlV2 Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 t | l2_shrinkage |
-> Tensor v'10 t | lr_power |
-> m' ControlNode |
resourceSparseApplyFtrlV2' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 ResourceHandle | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 t | l2_shrinkage |
-> Tensor v'10 t | lr_power |
-> m' ControlNode |
resourceSparseApplyMomentum Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | momentum |
-> m' ControlNode |
resourceSparseApplyMomentum' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | momentum |
-> m' ControlNode |
resourceSparseApplyProximalAdagrad Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | l1 |
-> Tensor v'5 t | l2 |
-> Tensor v'6 t | grad |
-> Tensor v'7 tindices | indices |
-> m' ControlNode |
resourceSparseApplyProximalAdagrad' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | accum |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | l1 |
-> Tensor v'5 t | l2 |
-> Tensor v'6 t | grad |
-> Tensor v'7 tindices | indices |
-> m' ControlNode |
resourceSparseApplyProximalGradientDescent Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 t | alpha |
-> Tensor v'3 t | l1 |
-> Tensor v'4 t | l2 |
-> Tensor v'5 t | grad |
-> Tensor v'6 tindices | indices |
-> m' ControlNode |
resourceSparseApplyProximalGradientDescent' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 t | alpha |
-> Tensor v'3 t | l1 |
-> Tensor v'4 t | l2 |
-> Tensor v'5 t | grad |
-> Tensor v'6 tindices | indices |
-> m' ControlNode |
resourceSparseApplyRMSProp Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | ms |
-> Tensor v'3 ResourceHandle | mom |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | momentum |
-> Tensor v'7 t | epsilon |
-> Tensor v'8 t | grad |
-> Tensor v'9 tindices | indices |
-> m' ControlNode |
resourceSparseApplyRMSProp' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | var |
-> Tensor v'2 ResourceHandle | ms |
-> Tensor v'3 ResourceHandle | mom |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | momentum |
-> Tensor v'7 t | epsilon |
-> Tensor v'8 t | grad |
-> Tensor v'9 tindices | indices |
-> m' ControlNode |
resourceStridedSliceAssign Source #
:: (MonadBuild m', TensorType t, OneOf '[Int32, Int64] index) | |
=> Tensor v'1 ResourceHandle | ref |
-> Tensor v'2 index | begin |
-> Tensor v'3 index | end |
-> Tensor v'4 index | strides |
-> Tensor v'5 t | value |
-> m' ControlNode |
resourceStridedSliceAssign' Source #
:: (MonadBuild m', TensorType t, OneOf '[Int32, Int64] index) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | ref |
-> Tensor v'2 index | begin |
-> Tensor v'3 index | end |
-> Tensor v'4 index | strides |
-> Tensor v'5 t | value |
-> m' ControlNode |
:: (MonadBuild m', TensorType dt) | |
=> Tensor v'1 ByteString | file_pattern |
-> Tensor v'2 ByteString | tensor_name |
-> m' (Tensor Value dt) | tensor |
:: (MonadBuild m', TensorType dt) | |
=> OpParams | |
-> Tensor v'1 ByteString | file_pattern |
-> Tensor v'2 ByteString | tensor_name |
-> m' (Tensor Value dt) | tensor |
:: (MonadBuild m', TensorType dt) | |
=> Tensor v'1 ByteString | file_pattern |
-> Tensor v'2 ByteString | tensor_name |
-> Tensor v'3 ByteString | shape_and_slice |
-> m' (Tensor Value dt) | tensor |
:: (MonadBuild m', TensorType dt) | |
=> OpParams | |
-> Tensor v'1 ByteString | file_pattern |
-> Tensor v'2 ByteString | tensor_name |
-> Tensor v'3 ByteString | shape_and_slice |
-> m' (Tensor Value dt) | tensor |
:: (MonadBuild m', TensorTypes dtypes) | |
=> Tensor v'1 ByteString | prefix |
-> Tensor v'2 ByteString | tensor_names |
-> Tensor v'3 ByteString | shape_and_slices |
-> m' (TensorList Value dtypes) | tensors |
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> Tensor v'1 ByteString | prefix |
-> Tensor v'2 ByteString | tensor_names |
-> Tensor v'3 ByteString | shape_and_slices |
-> m' (TensorList Value dtypes) | tensors |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | address |
-> Tensor v'2 ByteString | method |
-> Tensor v'3 ByteString | request |
-> m' (Tensor Value ByteString) | response |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | address |
-> Tensor v'2 ByteString | method |
-> Tensor v'3 ByteString | request |
-> m' (Tensor Value ByteString) | response |
:: (MonadBuild m', TensorTypes t) | |
=> Tensor v'1 ByteString | filename |
-> Tensor v'2 ByteString | tensor_names |
-> TensorList v'3 t | data |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes t) | |
=> OpParams | |
-> Tensor v'1 ByteString | filename |
-> Tensor v'2 ByteString | tensor_names |
-> TensorList v'3 t | data |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes t) | |
=> Tensor v'1 ByteString | filename |
-> Tensor v'2 ByteString | tensor_names |
-> Tensor v'3 ByteString | shapes_and_slices |
-> TensorList v'4 t | data |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes t) | |
=> OpParams | |
-> Tensor v'1 ByteString | filename |
-> Tensor v'2 ByteString | tensor_names |
-> Tensor v'3 ByteString | shapes_and_slices |
-> TensorList v'4 t | data |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes dtypes) | |
=> Tensor v'1 ByteString | prefix |
-> Tensor v'2 ByteString | tensor_names |
-> Tensor v'3 ByteString | shape_and_slices |
-> TensorList v'4 dtypes | tensors |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> Tensor v'1 ByteString | prefix |
-> Tensor v'2 ByteString | tensor_names |
-> Tensor v'3 ByteString | shape_and_slices |
-> TensorList v'4 dtypes | tensors |
-> m' ControlNode |
:: Tensor v'1 ByteString | input |
-> Tensor Build Int64 | output |
:: Float | l1 |
-> Float | l2 |
-> Int64 | num_inner_iterations |
-> Int64 | num_loss_partitions |
-> [Tensor v'1 Int64] | sparse_example_indices |
-> [Tensor v'2 Int64] | sparse_feature_indices |
-> [Tensor v'3 Float] | sparse_feature_values |
-> [Tensor v'4 Float] | dense_features |
-> Tensor v'5 Float | example_weights |
-> Tensor v'6 Float | example_labels |
-> [Tensor v'7 Int64] | sparse_indices |
-> [Tensor v'8 Float] | sparse_weights |
-> [Tensor v'9 Float] | dense_weights |
-> Tensor v'10 Float | example_state_data |
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float]) | (out_example_state_data, out_delta_sparse_weights, out_delta_dense_weights)
|
:: OpParams | |
-> Float | l1 |
-> Float | l2 |
-> Int64 | num_inner_iterations |
-> Int64 | num_loss_partitions |
-> [Tensor v'1 Int64] | sparse_example_indices |
-> [Tensor v'2 Int64] | sparse_feature_indices |
-> [Tensor v'3 Float] | sparse_feature_values |
-> [Tensor v'4 Float] | dense_features |
-> Tensor v'5 Float | example_weights |
-> Tensor v'6 Float | example_labels |
-> [Tensor v'7 Int64] | sparse_indices |
-> [Tensor v'8 Float] | sparse_weights |
-> [Tensor v'9 Float] | dense_weights |
-> Tensor v'10 Float | example_state_data |
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float]) | (out_example_state_data, out_delta_sparse_weights, out_delta_dense_weights)
|
:: MonadBuild m' | |
=> Float | l1 |
-> Float | l2 |
-> [Tensor Ref Float] | weights |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Float | l1 |
-> Float | l2 |
-> [Tensor Ref Float] | weights |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | resource_handle |
-> m' (Tensor Value Variant) | serialized |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource_handle |
-> m' (Tensor Value Variant) | serialized |
:: TensorType t | |
=> Tensor v'1 t | tensor |
-> Tensor Build ByteString | serialized |
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | tensor |
-> Tensor Build ByteString | serialized |
setStatsAggregatorDataset Source #
:: MonadBuild m' | |
=> [DataType] | output_types |
-> Tensor v'1 Variant | input_dataset |
-> Tensor v'2 ResourceHandle | stats_aggregator |
-> m' (Tensor Value Variant) | handle |
setStatsAggregatorDataset' Source #
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | output_types |
-> Tensor v'1 Variant | input_dataset |
-> Tensor v'2 ResourceHandle | stats_aggregator |
-> m' (Tensor Value Variant) | handle |
:: Tensor v'1 ByteString | basename |
-> Tensor v'2 Int32 | shard |
-> Tensor v'3 Int32 | num_shards |
-> Tensor Build ByteString | filename |
:: OpParams | |
-> Tensor v'1 ByteString | basename |
-> Tensor v'2 Int32 | shard |
-> Tensor v'3 Int32 | num_shards |
-> Tensor Build ByteString | filename |
:: Tensor v'1 ByteString | basename |
-> Tensor v'2 Int32 | num_shards |
-> Tensor Build ByteString | filename |
:: OpParams | |
-> Tensor v'1 ByteString | basename |
-> Tensor v'2 Int32 | num_shards |
-> Tensor Build ByteString | filename |
shutdownDistributedTPU :: forall m'. MonadBuild m' => m' ControlNode Source #
An op that shuts down a running distributed TPU system. The Op returns
an error if no system is running.
shutdownDistributedTPU' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode Source #
:: MonadBuild m' | |
=> Int64 | batch_size |
-> m' (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64, Tensor Value Int32, Tensor Value Int64, Tensor Value Int32, Tensor Value Int32) | (vocab_word, vocab_freq, words_per_epoch, current_epoch, total_words_processed, examples, labels)
|
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | batch_size |
-> m' (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64, Tensor Value Int32, Tensor Value Int64, Tensor Value Int32, Tensor Value Int32) | (vocab_word, vocab_freq, words_per_epoch, current_epoch, total_words_processed, examples, labels)
|
:: TensorType t | |
=> Int64 | block_size |
-> Tensor v'1 t | input |
-> Tensor Build t | output |
sparseAccumulatorApplyGradient Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) | |
=> Bool | has_known_shape |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Int64 | local_step |
-> Tensor v'3 Int64 | gradient_indices |
-> Tensor v'4 dtype | gradient_values |
-> Tensor v'5 Int64 | gradient_shape |
-> m' ControlNode |
sparseAccumulatorApplyGradient' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) | |
=> OpParams | |
-> Bool | has_known_shape |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Int64 | local_step |
-> Tensor v'3 Int64 | gradient_indices |
-> Tensor v'4 dtype | gradient_values |
-> Tensor v'5 Int64 | gradient_shape |
-> m' ControlNode |
sparseAccumulatorTakeGradient Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | num_required |
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64) | (indices, values, shape)
|
sparseAccumulatorTakeGradient' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | num_required |
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64) | (indices, values, shape)
|
:: (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] treal) | |
=> Tensor v'1 Int64 | a_indices |
-> Tensor v'2 t | a_values |
-> Tensor v'3 Int64 | a_shape |
-> Tensor v'4 Int64 | b_indices |
-> Tensor v'5 t | b_values |
-> Tensor v'6 Int64 | b_shape |
-> Tensor v'7 treal | thresh |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64) | (sum_indices, sum_values, sum_shape)
|
:: (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] treal) | |
=> OpParams | |
-> Tensor v'1 Int64 | a_indices |
-> Tensor v'2 t | a_values |
-> Tensor v'3 Int64 | a_shape |
-> Tensor v'4 Int64 | b_indices |
-> Tensor v'5 t | b_values |
-> Tensor v'6 Int64 | b_shape |
-> Tensor v'7 treal | thresh |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64) | (sum_indices, sum_values, sum_shape)
|
:: OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> Tensor v'1 t | backprop_val_grad |
-> Tensor v'2 Int64 | a_indices |
-> Tensor v'3 Int64 | b_indices |
-> Tensor v'4 Int64 | sum_indices |
-> (Tensor Build t, Tensor Build t) | (a_val_grad, b_val_grad)
|
:: OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> OpParams | |
-> Tensor v'1 t | backprop_val_grad |
-> Tensor v'2 Int64 | a_indices |
-> Tensor v'3 Int64 | b_indices |
-> Tensor v'4 Int64 | sum_indices |
-> (Tensor Build t, Tensor Build t) | (a_val_grad, b_val_grad)
|
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor Ref t | accum_update |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | epsilon |
-> Tensor v'7 t | grad |
-> Tensor v'8 tindices | indices |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor Ref t | accum_update |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | epsilon |
-> Tensor v'7 t | grad |
-> Tensor v'8 tindices | indices |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor Ref t | var |
-> Tensor Ref t | gradient_accumulator |
-> Tensor Ref t | gradient_squared_accumulator |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 Int64 | global_step |
-> m' (Tensor Ref t) | out |
sparseApplyAdagradDA' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | gradient_accumulator |
-> Tensor Ref t | gradient_squared_accumulator |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 Int64 | global_step |
-> m' (Tensor Ref t) | out |
sparseApplyCenteredRMSProp Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor Ref t | var |
-> Tensor Ref t | mg |
-> Tensor Ref t | ms |
-> Tensor Ref t | mom |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | rho |
-> Tensor v'7 t | momentum |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> Tensor v'10 tindices | indices |
-> m' (Tensor Ref t) | out |
sparseApplyCenteredRMSProp' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | mg |
-> Tensor Ref t | ms |
-> Tensor Ref t | mom |
-> Tensor v'5 t | lr |
-> Tensor v'6 t | rho |
-> Tensor v'7 t | momentum |
-> Tensor v'8 t | epsilon |
-> Tensor v'9 t | grad |
-> Tensor v'10 tindices | indices |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor Ref t | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 t | lr_power |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor Ref t | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 t | lr_power |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor Ref t | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 t | l2_shrinkage |
-> Tensor v'10 t | lr_power |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor Ref t | linear |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | lr |
-> Tensor v'7 t | l1 |
-> Tensor v'8 t | l2 |
-> Tensor v'9 t | l2_shrinkage |
-> Tensor v'10 t | lr_power |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | momentum |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | grad |
-> Tensor v'5 tindices | indices |
-> Tensor v'6 t | momentum |
-> m' (Tensor Ref t) | out |
sparseApplyProximalAdagrad Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | l1 |
-> Tensor v'5 t | l2 |
-> Tensor v'6 t | grad |
-> Tensor v'7 tindices | indices |
-> m' (Tensor Ref t) | out |
sparseApplyProximalAdagrad' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | accum |
-> Tensor v'3 t | lr |
-> Tensor v'4 t | l1 |
-> Tensor v'5 t | l2 |
-> Tensor v'6 t | grad |
-> Tensor v'7 tindices | indices |
-> m' (Tensor Ref t) | out |
sparseApplyProximalGradientDescent Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor Ref t | var |
-> Tensor v'2 t | alpha |
-> Tensor v'3 t | l1 |
-> Tensor v'4 t | l2 |
-> Tensor v'5 t | grad |
-> Tensor v'6 tindices | indices |
-> m' (Tensor Ref t) | out |
sparseApplyProximalGradientDescent' Source #
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor v'2 t | alpha |
-> Tensor v'3 t | l1 |
-> Tensor v'4 t | l2 |
-> Tensor v'5 t | grad |
-> Tensor v'6 tindices | indices |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> Tensor Ref t | var |
-> Tensor Ref t | ms |
-> Tensor Ref t | mom |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | momentum |
-> Tensor v'7 t | epsilon |
-> Tensor v'8 t | grad |
-> Tensor v'9 tindices | indices |
-> m' (Tensor Ref t) | out |
:: (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) | |
=> OpParams | |
-> Tensor Ref t | var |
-> Tensor Ref t | ms |
-> Tensor Ref t | mom |
-> Tensor v'4 t | lr |
-> Tensor v'5 t | rho |
-> Tensor v'6 t | momentum |
-> Tensor v'7 t | epsilon |
-> Tensor v'8 t | grad |
-> Tensor v'9 tindices | indices |
-> m' (Tensor Ref t) | out |
sparseConditionalAccumulator Source #
:: MonadBuild m' | |
=> DataType | dtype |
-> Shape | shape |
-> m' (Tensor Ref ByteString) | handle |
sparseConditionalAccumulator' Source #
:: MonadBuild m' | |
=> OpParams | |
-> DataType | dtype |
-> Shape | shape |
-> m' (Tensor Ref ByteString) | handle |
:: (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types, OneOf '[ByteString, Int64] out_type) | |
=> Int64 | hash_key |
-> Bool | hashed_output |
-> DataType | internal_type |
-> Int64 | num_buckets |
-> [Tensor v'1 Int64] | indices |
-> TensorList v'2 sparse_types | values |
-> [Tensor v'3 Int64] | shapes |
-> TensorList v'4 dense_types | dense_inputs |
-> (Tensor Build Int64, Tensor Build out_type, Tensor Build Int64) | (output_indices, output_values, output_shape)
|
:: (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types, OneOf '[ByteString, Int64] out_type) | |
=> OpParams | |
-> Int64 | hash_key |
-> Bool | hashed_output |
-> DataType | internal_type |
-> Int64 | num_buckets |
-> [Tensor v'1 Int64] | indices |
-> TensorList v'2 sparse_types | values |
-> [Tensor v'3 Int64] | shapes |
-> TensorList v'4 dense_types | dense_inputs |
-> (Tensor Build Int64, Tensor Build out_type, Tensor Build Int64) | (output_indices, output_values, output_shape)
|
:: TensorType t | |
=> Tensor v'1 Int64 | indices |
-> Tensor v'2 t | values |
-> Tensor v'3 Int64 | dense_shape |
-> Tensor v'4 t | default_value |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Bool, Tensor Build Int64) | (output_indices, output_values, empty_row_indicator, reverse_index_map)
|
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 Int64 | indices |
-> Tensor v'2 t | values |
-> Tensor v'3 Int64 | dense_shape |
-> Tensor v'4 t | default_value |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Bool, Tensor Build Int64) | (output_indices, output_values, empty_row_indicator, reverse_index_map)
|
sparseReduceMaxSparse Source #
:: OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> Tensor v'1 Int64 | input_indices |
-> Tensor v'2 t | input_values |
-> Tensor v'3 Int64 | input_shape |
-> Tensor v'4 Int32 | reduction_axes |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64) | (output_indices, output_values, output_shape)
|
sparseReduceMaxSparse' Source #
:: OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> OpParams | |
-> Tensor v'1 Int64 | input_indices |
-> Tensor v'2 t | input_values |
-> Tensor v'3 Int64 | input_shape |
-> Tensor v'4 Int32 | reduction_axes |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64) | (output_indices, output_values, output_shape)
|
sparseReduceSumSparse Source #
:: OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> Tensor v'1 Int64 | input_indices |
-> Tensor v'2 t | input_values |
-> Tensor v'3 Int64 | input_shape |
-> Tensor v'4 Int32 | reduction_axes |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64) | (output_indices, output_values, output_shape)
|
sparseReduceSumSparse' Source #
:: OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> OpParams | |
-> Tensor v'1 Int64 | input_indices |
-> Tensor v'2 t | input_values |
-> Tensor v'3 Int64 | input_shape |
-> Tensor v'4 Int32 | reduction_axes |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64) | (output_indices, output_values, output_shape)
|
:: OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> Tensor v'1 Int64 | a_indices |
-> Tensor v'2 t | a_values |
-> Tensor v'3 Int64 | a_shape |
-> Tensor v'4 Int64 | b_indices |
-> Tensor v'5 t | b_values |
-> Tensor v'6 Int64 | b_shape |
-> (Tensor Build Int64, Tensor Build t) | (output_indices, output_values)
|
:: OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> OpParams | |
-> Tensor v'1 Int64 | a_indices |
-> Tensor v'2 t | a_values |
-> Tensor v'3 Int64 | a_shape |
-> Tensor v'4 Int64 | b_indices |
-> Tensor v'5 t | b_values |
-> Tensor v'6 Int64 | b_shape |
-> (Tensor Build Int64, Tensor Build t) | (output_indices, output_values)
|
:: OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> Tensor v'1 Int64 | a_indices |
-> Tensor v'2 t | a_values |
-> Tensor v'3 Int64 | a_shape |
-> Tensor v'4 Int64 | b_indices |
-> Tensor v'5 t | b_values |
-> Tensor v'6 Int64 | b_shape |
-> (Tensor Build Int64, Tensor Build t) | (output_indices, output_values)
|
:: OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t | |
=> OpParams | |
-> Tensor v'1 Int64 | a_indices |
-> Tensor v'2 t | a_values |
-> Tensor v'3 Int64 | a_shape |
-> Tensor v'4 Int64 | b_indices |
-> Tensor v'5 t | b_values |
-> Tensor v'6 Int64 | b_shape |
-> (Tensor Build Int64, Tensor Build t) | (output_indices, output_values)
|
sparseTensorSliceDataset Source #
:: (MonadBuild m', TensorType tvalues) | |
=> Tensor v'1 Int64 | indices |
-> Tensor v'2 tvalues | values |
-> Tensor v'3 Int64 | dense_shape |
-> m' (Tensor Value Variant) | handle |
sparseToSparseSetOperation Source #
:: OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t | |
=> Tensor v'1 Int64 | set1_indices |
-> Tensor v'2 t | set1_values |
-> Tensor v'3 Int64 | set1_shape |
-> Tensor v'4 Int64 | set2_indices |
-> Tensor v'5 t | set2_values |
-> Tensor v'6 Int64 | set2_shape |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64) | (result_indices, result_values, result_shape)
|
sparseToSparseSetOperation' Source #
:: OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t | |
=> OpParams | |
-> Tensor v'1 Int64 | set1_indices |
-> Tensor v'2 t | set1_values |
-> Tensor v'3 Int64 | set1_shape |
-> Tensor v'4 Int64 | set2_indices |
-> Tensor v'5 t | set2_values |
-> Tensor v'6 Int64 | set2_shape |
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64) | (result_indices, result_values, result_shape)
|
:: MonadBuild m' | |
=> [DataType] | output_types |
-> Tensor v'1 ByteString | driver_name |
-> Tensor v'2 ByteString | data_source_name |
-> Tensor v'3 ByteString | query |
-> m' (Tensor Value Variant) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | output_types |
-> Tensor v'1 ByteString | driver_name |
-> Tensor v'2 ByteString | data_source_name |
-> Tensor v'3 ByteString | query |
-> m' (Tensor Value Variant) | handle |
:: MonadBuild m' | |
=> DataType | elem_type |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> DataType | elem_type |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> Tensor Ref ByteString | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> m' ControlNode |
:: (MonadBuild m', TensorType elem_type) | |
=> Tensor Ref ByteString | handle |
-> m' (Tensor Value elem_type) | elem |
:: (MonadBuild m', TensorType elem_type) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> m' (Tensor Value elem_type) | elem |
:: (MonadBuild m', TensorType elem_type) | |
=> Tensor v'1 ResourceHandle | handle |
-> m' (Tensor Value elem_type) | elem |
:: (MonadBuild m', TensorType elem_type) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> m' (Tensor Value elem_type) | elem |
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 t | elem |
-> m' (Tensor Value t) | output |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 t | elem |
-> m' (Tensor Value t) | output |
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 t | elem |
-> m' (Tensor Value t) | output |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 t | elem |
-> m' (Tensor Value t) | output |
:: MonadBuild m' | |
=> DataType | elem_type |
-> Tensor v'1 Int32 | max_size |
-> m' (Tensor Value ResourceHandle) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> DataType | elem_type |
-> Tensor v'1 Int32 | max_size |
-> m' (Tensor Value ResourceHandle) | handle |
:: (MonadBuild m', TensorTypes dtypes) | |
=> TensorList v'1 dtypes | values |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> TensorList v'1 dtypes | values |
-> m' ControlNode |
:: MonadBuild m' | |
=> [DataType] | dtypes |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> [DataType] | dtypes |
-> m' ControlNode |
:: (MonadBuild m', TensorTypes dtypes) | |
=> Tensor v'1 Int32 | index |
-> m' (TensorList Value dtypes) | values |
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> Tensor v'1 Int32 | index |
-> m' (TensorList Value dtypes) | values |
statsAggregatorHandle Source #
:: MonadBuild m' | |
=> m' (Tensor Value ResourceHandle) | handle |
statsAggregatorHandle' Source #
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value ResourceHandle) | handle |
statsAggregatorSummary Source #
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | iterator |
-> m' (Tensor Value ByteString) | summary |
statsAggregatorSummary' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | iterator |
-> m' (Tensor Value ByteString) | summary |
:: TensorType t | |
=> Tensor v'1 t | input |
-> Tensor Build t | output |
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | input |
-> Tensor Build t | output |
:: [Tensor v'1 ByteString] | inputs |
-> Tensor Build ByteString | output |
:: OpParams | |
-> [Tensor v'1 ByteString] | inputs |
-> Tensor Build ByteString | output |
:: Tensor v'1 ByteString | input |
-> Tensor v'2 ByteString | delimiter |
-> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64) | (indices, values, shape)
|
:: OpParams | |
-> Tensor v'1 ByteString | input |
-> Tensor v'2 ByteString | delimiter |
-> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64) | (indices, values, shape)
|
:: Tensor v'1 ByteString | input |
-> Tensor Build ByteString | output |
:: OpParams | |
-> Tensor v'1 ByteString | input |
-> Tensor Build ByteString | output |
:: OneOf '[Int32, Int64] t | |
=> Tensor v'1 ByteString | input |
-> Tensor v'2 t | pos |
-> Tensor v'3 t | len |
-> Tensor Build ByteString | output |
:: MonadBuild m' | |
=> m' (Tensor Value ResourceHandle) | writer |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value ResourceHandle) | writer |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | filenames |
-> Tensor v'2 ByteString | compression_type |
-> Tensor v'3 Int64 | buffer_size |
-> m' (Tensor Value Variant) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | filenames |
-> Tensor v'2 ByteString | compression_type |
-> Tensor v'3 Int64 | buffer_size |
-> m' (Tensor Value Variant) | handle |
:: MonadBuild m' | |
=> m' (Tensor Ref ByteString) | reader_handle |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Ref ByteString) | reader_handle |
:: MonadBuild m' | |
=> m' (Tensor Value ResourceHandle) | reader_handle |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value ResourceHandle) | reader_handle |
:: Tensor Build ByteString | output |
tPUCompilationResult' Source #
:: OpParams | |
-> Tensor Build ByteString | output |
tPUEmbeddingActivations Source #
:: Int64 | lookup_id: Identifier of the set of embedding indices which produced these activations. |
-> Int64 | table_id: The id of the table in the embedding layer configuration from which these activations were computed. |
-> Tensor v'1 Float | embedding_variable: A trainable variable, enabling optimizers to find this op. |
-> Tensor v'2 Float | sliced_activations: The embedding activations Tensor to return. |
-> Tensor Build Float | output |
An op enabling differentiation of TPU Embeddings.
This op simply returns its first input, which is assumed to have been sliced from the Tensors returned by TPUEmbeddingDequeueActivations. The presence of this op, and its first argument being a trainable Variable, enables automatic differentiation of graphs containing embeddings via the TPU Embedding Python libraries.
tPUEmbeddingActivations' Source #
:: OpParams | |
-> Int64 | lookup_id: Identifier of the set of embedding indices which produced these activations. |
-> Int64 | table_id: The id of the table in the embedding layer configuration from which these activations were computed. |
-> Tensor v'1 Float | embedding_variable: A trainable variable, enabling optimizers to find this op. |
-> Tensor v'2 Float | sliced_activations: The embedding activations Tensor to return. |
-> Tensor Build Float | output |
tPUEmbeddingEnqueueSparseBatch Source #
:: MonadBuild m' | |
=> [Tensor v'1 Int32] | sample_indices: A list of rank 1 Tensors specifying row indices of the COO sparse matrix representing the embedding lookups for each table. |
-> [Tensor v'2 Int32] | embedding_indices: A list of rank 1 Tensors specifying column indices of the COO sparse matrix representing the embedding lookups for each table. |
-> [Tensor v'3 Float] | aggregation_weights: A list of rank 1 Tensors specifying the nonzero values of the COO sparse matrix representing the embedding lookups for each table. |
-> m' ControlNode |
An op that feeds a batch of embedding indices and weights to the TPU.
Embedding lookups are equivalent to sparse-dense matrix multiplications: the sparse matrix contains nonzeros in column j in order to retrieve row j from the embedding table.
The three Tensor list arguments (sample_indices, embedding_indices, and aggregation_weights) represent these sparse matrices in COO format. The Tensor lists each have one entry for each embedding table specified in the model. For the kth embedding table, the three Tensors at position k in the list specify a COO-format sparse matrix. For the kth table, the row indices, column indices, and nonzero values of the COO sparse matrix are specified by sample_indices[k], embedding_indices[k], and aggregation_weights[k], respectively. Entries must be sorted by row index, then by column index.
There should be at most one TPUEmbeddingEnqueueSparseBatch op in a signle training step per TPU shard.
tPUEmbeddingEnqueueSparseBatch' Source #
:: MonadBuild m' | |
=> OpParams | |
-> [Tensor v'1 Int32] | sample_indices: A list of rank 1 Tensors specifying row indices of the COO sparse matrix representing the embedding lookups for each table. |
-> [Tensor v'2 Int32] | embedding_indices: A list of rank 1 Tensors specifying column indices of the COO sparse matrix representing the embedding lookups for each table. |
-> [Tensor v'3 Float] | aggregation_weights: A list of rank 1 Tensors specifying the nonzero values of the COO sparse matrix representing the embedding lookups for each table. |
-> m' ControlNode |
tPUEmbeddingLoadAdagradParameters Source #
:: MonadBuild m' | |
=> Int64 | host_id: Which CPU host in the distributed training job will execute this op. |
-> Int64 | num_hosts: The number of CPU hosts in the distributed training job. |
-> Int64 | table_id: The id of the table specified in the embedding_config. |
-> Tensor v'1 Float | parameters: The shard of the embedding table resident on the host executing this op. For single-TPU models, this is the entire embedding table. |
-> Tensor v'2 Float | accumulators: Shard of the Adagrad accumulators resident on the host executing this op. |
-> m' ControlNode |
Load an embedding table shard into TensorNode memories for use with Adagrad.
TPU embeddings use dedicated per-optimizer Ops for loading and retrieving trainable variables and optimizer state from TPU memory. This op enables functionality equivalent to AdagradOptimizer.
tPUEmbeddingLoadAdagradParameters' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | host_id: Which CPU host in the distributed training job will execute this op. |
-> Int64 | num_hosts: The number of CPU hosts in the distributed training job. |
-> Int64 | table_id: The id of the table specified in the embedding_config. |
-> Tensor v'1 Float | parameters: The shard of the embedding table resident on the host executing this op. For single-TPU models, this is the entire embedding table. |
-> Tensor v'2 Float | accumulators: Shard of the Adagrad accumulators resident on the host executing this op. |
-> m' ControlNode |
tPUEmbeddingLoadGradientDescentParameters Source #
:: MonadBuild m' | |
=> Int64 | host_id: Which CPU host in the distributed training job will execute this op. |
-> Int64 | num_hosts: The number of CPU hosts in the distributed training job. |
-> Int64 | table_id: The id of the table specified in the tpu_embedding_config. |
-> Tensor v'1 Float | parameters: The shard of the embedding table resident on the host executing this op. For single-TPU models, this is the entire embedding table. |
-> m' ControlNode |
Load an embedding table shard into TPU memory for use with GradientDescent.
TPU embeddings use dedicated per-optimizer Ops for loading and retrieving trainable variables and optimizer state from TPU memory. This op enables functionality equivalent to GradientDescentOptimizer.
tPUEmbeddingLoadGradientDescentParameters' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | host_id: Which CPU host in the distributed training job will execute this op. |
-> Int64 | num_hosts: The number of CPU hosts in the distributed training job. |
-> Int64 | table_id: The id of the table specified in the tpu_embedding_config. |
-> Tensor v'1 Float | parameters: The shard of the embedding table resident on the host executing this op. For single-TPU models, this is the entire embedding table. |
-> m' ControlNode |
tPUEmbeddingReceiveActivations Source #
:: MonadBuild m' | |
=> Int64 | num_tables: The number of output activation tensors, equal to the number of embedding tables in the model. |
-> m' [Tensor Value Float] | outputs: A TensorList of embedding activations containing one Tensor per embedding table in the model. |
An op that receives embedding activations on the TPU.
The TPU system performs the embedding lookups and aggregations specified by the arguments to TPUEmbeddingEnqueueSparseBatch. The results of these aggregations are visible to the Tensorflow Graph as the outputs of a TPUEmbeddingDequeueActivations Op. This op returns a list containing one Tensor of activations per table specified in the model. There can be at most one ReceieveActivations op in the TPU graph.
tPUEmbeddingRetrieveAdagradParameters Source #
:: MonadBuild m' | |
=> Int64 | host_id: Which CPU host in the distributed training job will execute this op. |
-> Int64 | num_hosts: The number of CPU hosts in the distributed training job. |
-> Int64 | table_id: The id of the table specified in the embedding_config_json. |
-> m' (Tensor Value Float, Tensor Value Float) | (parameters, accumulators)
|
Retrieve an embedding table shard from TPU memory.
TPU embeddings use dedicated per-optimizer Ops for loading and retrieving trainable variables and optimizer state from TPU memory. This op enables functionality equivalent to AdagradOptimizer.
tPUEmbeddingRetrieveAdagradParameters' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | host_id: Which CPU host in the distributed training job will execute this op. |
-> Int64 | num_hosts: The number of CPU hosts in the distributed training job. |
-> Int64 | table_id: The id of the table specified in the embedding_config_json. |
-> m' (Tensor Value Float, Tensor Value Float) | (parameters, accumulators)
|
tPUEmbeddingRetrieveGradientDescentParameters Source #
:: MonadBuild m' | |
=> Int64 | host_id: Which CPU host in the distributed training job will execute this op. |
-> Int64 | num_hosts: The number of CPU hosts in the distributed training job. |
-> Int64 | table_id: The id of the table specified in tpu_embedding_config. |
-> m' (Tensor Value Float) | parameters |
Retrieve an embedding table shard from TPU memory.
TPU embeddings use dedicated per-optimizer Ops for loading and retrieving trainable variables and optimizer state from TPU memory. This op enables functionality equivalent to GradientDescentOptimizer.
tPUEmbeddingSendGradients Source #
:: MonadBuild m' | |
=> [Tensor v'1 Float] | gradients: A TensorList of gradients with which to update embedding tables. |
-> m' ControlNode |
An op that performs gradient updates of embedding tables.
The TensorList argument has the same length and shapes as the return value of TPUEmbeddingReceiveActivations, but contains gradients of the model's loss with respect to the embedding activations. The embedding tables are updated from these gradients via the optimizer specified in the configuration given to tpu.initialize_system.
tPUEmbeddingSendGradients' Source #
:: MonadBuild m' | |
=> OpParams | |
-> [Tensor v'1 Float] | gradients: A TensorList of gradients with which to update embedding tables. |
-> m' ControlNode |
:: MonadBuild m' | |
=> Int64 | num_replicas |
-> m' ControlNode |
tPUReplicateMetadata' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | num_replicas |
-> m' ControlNode |
:: TensorType t | |
=> [Tensor v'1 t] | inputs |
-> Tensor Build t | output |
Operator that connects N unreplicated inputs to an N-way replicated TPU computation.
:: TensorType t | |
=> OpParams | |
-> [Tensor v'1 t] | inputs |
-> Tensor Build t | output |
:: TensorType t | |
=> Int64 | num_replicas |
-> Tensor v'1 t | input |
-> [Tensor Build t] | outputs |
Operator that connects the output of an N-way replicated TPU computation to N separate outputs.
:: (MonadBuild m', TensorType dtype) | |
=> Shape | shape |
-> m' (Tensor Ref dtype) | ref |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Shape | shape |
-> m' (Tensor Ref dtype) | ref |
:: MonadBuild m' | |
=> DataType | dtype |
-> Tensor v'1 Int32 | size |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> DataType | dtype |
-> Tensor v'1 Int32 | size |
-> m' (Tensor Ref ByteString) | handle |
:: MonadBuild m' | |
=> Tensor Ref ByteString | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | handle |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> m' ControlNode |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value dtype, Tensor Value Int64) | (value, lengths)
|
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value dtype, Tensor Value Int64) | (value, lengths)
|
:: TensorType dtype | |
=> Tensor v'1 ByteString | handle |
-> Tensor v'2 Float | flow_in |
-> (Tensor Build dtype, Tensor Build Int64) | (value, lengths)
|
:: (MonadBuild m', TensorType dtype) | |
=> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value dtype, Tensor Value Int64) | (value, lengths)
|
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value dtype, Tensor Value Int64) | (value, lengths)
|
:: (MonadBuild m', TensorType dtype) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | indices |
-> Tensor v'3 Float | flow_in |
-> m' (Tensor Value dtype) | value |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | indices |
-> Tensor v'3 Float | flow_in |
-> m' (Tensor Value dtype) | value |
:: TensorType dtype | |
=> Tensor v'1 ByteString | handle |
-> Tensor v'2 Int32 | indices |
-> Tensor v'3 Float | flow_in |
-> Tensor Build dtype | value |
:: TensorType dtype | |
=> OpParams | |
-> Tensor v'1 ByteString | handle |
-> Tensor v'2 Int32 | indices |
-> Tensor v'3 Float | flow_in |
-> Tensor Build dtype | value |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | indices |
-> Tensor v'3 Float | flow_in |
-> m' (Tensor Value dtype) | value |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | indices |
-> Tensor v'3 Float | flow_in |
-> m' (Tensor Value dtype) | value |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Ref ByteString) | grad_handle |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Ref ByteString) | grad_handle |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value ByteString) | grad_handle |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value ByteString) | grad_handle |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value ResourceHandle, Tensor Value Float) | (grad_handle, flow_out)
|
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value ResourceHandle, Tensor Value Float) | (grad_handle, flow_out)
|
:: (MonadBuild m', TensorType dtype) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value dtype) | value |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value dtype) | value |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | index |
-> Tensor v'3 Float | flow_in |
-> m' (Tensor Value dtype) | value |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | index |
-> Tensor v'3 Float | flow_in |
-> m' (Tensor Value dtype) | value |
:: TensorType dtype | |
=> Tensor v'1 ByteString | handle |
-> Tensor v'2 Int32 | index |
-> Tensor v'3 Float | flow_in |
-> Tensor Build dtype | value |
:: TensorType dtype | |
=> OpParams | |
-> Tensor v'1 ByteString | handle |
-> Tensor v'2 Int32 | index |
-> Tensor v'3 Float | flow_in |
-> Tensor Build dtype | value |
:: (MonadBuild m', TensorType dtype) | |
=> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | index |
-> Tensor v'3 Float | flow_in |
-> m' (Tensor Value dtype) | value |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | index |
-> Tensor v'3 Float | flow_in |
-> m' (Tensor Value dtype) | value |
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | indices |
-> Tensor v'3 t | value |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | indices |
-> Tensor v'3 t | value |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | indices |
-> Tensor v'3 t | value |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
tensorArrayScatterV3' Source #
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | indices |
-> Tensor v'3 t | value |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: MonadBuild m' | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value Int32) | size |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value Int32) | size |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value Int32) | size |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Float | flow_in |
-> m' (Tensor Value Int32) | size |
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 t | value |
-> Tensor v'3 Int64 | lengths |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 t | value |
-> Tensor v'3 Int64 | lengths |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 t | value |
-> Tensor v'3 Int64 | lengths |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 t | value |
-> Tensor v'3 Int64 | lengths |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 t | value |
-> Tensor v'3 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 t | value |
-> Tensor v'3 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: MonadBuild m' | |
=> DataType | dtype |
-> Tensor v'1 Int32 | size |
-> m' (Tensor Value ByteString) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> DataType | dtype |
-> Tensor v'1 Int32 | size |
-> m' (Tensor Value ByteString) | handle |
:: MonadBuild m' | |
=> DataType | dtype |
-> Tensor v'1 Int32 | size |
-> m' (Tensor Value ResourceHandle, Tensor Value Float) | (handle, flow)
|
:: (MonadBuild m', TensorType t) | |
=> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | index |
-> Tensor v'3 t | value |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor Ref ByteString | handle |
-> Tensor v'2 Int32 | index |
-> Tensor v'3 t | value |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | index |
-> Tensor v'3 t | value |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | handle |
-> Tensor v'2 Int32 | index |
-> Tensor v'3 t | value |
-> Tensor v'4 Float | flow_in |
-> m' (Tensor Value Float) | flow_out |
:: (MonadBuild m', TensorTypes toutput_types) | |
=> TensorList v'1 toutput_types | components |
-> m' (Tensor Value Variant) | handle |
:: (MonadBuild m', TensorTypes toutput_types) | |
=> OpParams | |
-> TensorList v'1 toutput_types | components |
-> m' (Tensor Value Variant) | handle |
:: TensorType element_dtype | |
=> Tensor v'1 Variant | input_handle |
-> Tensor Build element_dtype | tensor |
:: (MonadBuild m', TensorTypes toutput_types) | |
=> TensorList v'1 toutput_types | components |
-> m' (Tensor Value Variant) | handle |
:: (MonadBuild m', TensorTypes toutput_types) | |
=> OpParams | |
-> TensorList v'1 toutput_types | components |
-> m' (Tensor Value Variant) | handle |
:: TensorType t | |
=> Tensor v'1 t | tensor |
-> Tensor Build ByteString | summary |
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | tensor |
-> Tensor Build ByteString | summary |
:: TensorType t | |
=> Tensor v'1 ByteString | tag |
-> Tensor v'2 t | tensor |
-> Tensor v'3 ByteString | serialized_summary_metadata |
-> Tensor Build ByteString | summary |
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 ByteString | tag |
-> Tensor v'2 t | tensor |
-> Tensor v'3 ByteString | serialized_summary_metadata |
-> Tensor Build ByteString | summary |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | filenames |
-> Tensor v'2 ByteString | compression_type |
-> Tensor v'3 Int64 | buffer_size |
-> m' (Tensor Value Variant) | handle |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | filenames |
-> Tensor v'2 ByteString | compression_type |
-> Tensor v'3 Int64 | buffer_size |
-> m' (Tensor Value Variant) | handle |
:: MonadBuild m' | |
=> m' (Tensor Ref ByteString) | reader_handle |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Ref ByteString) | reader_handle |
:: MonadBuild m' | |
=> m' (Tensor Value ResourceHandle) | reader_handle |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value ResourceHandle) | reader_handle |
threadUnsafeUnigramCandidateSampler Source #
:: MonadBuild m' | |
=> Int64 | num_sampled |
-> Int64 | num_true |
-> Int64 | range_max |
-> Bool | unique |
-> Tensor v'1 Int64 | true_classes |
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float) | (sampled_candidates, true_expected_count, sampled_expected_count)
|
threadUnsafeUnigramCandidateSampler' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | num_sampled |
-> Int64 | num_true |
-> Int64 | range_max |
-> Bool | unique |
-> Tensor v'1 Int64 | true_classes |
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float) | (sampled_candidates, true_expected_count, sampled_expected_count)
|
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value Double) | ts |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | address |
-> Tensor v'2 ByteString | method |
-> Tensor v'3 ByteString | request |
-> m' (Tensor Value ByteString, Tensor Value Int32, Tensor Value ByteString) | (response, status_code, status_message)
|
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | address |
-> Tensor v'2 ByteString | method |
-> Tensor v'3 ByteString | request |
-> m' (Tensor Value ByteString, Tensor Value Int32, Tensor Value ByteString) | (response, status_code, status_message)
|
uniformCandidateSampler Source #
:: MonadBuild m' | |
=> Int64 | num_sampled |
-> Int64 | num_true |
-> Int64 | range_max |
-> Bool | unique |
-> Tensor v'1 Int64 | true_classes |
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float) | (sampled_candidates, true_expected_count, sampled_expected_count)
|
uniformCandidateSampler' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Int64 | num_sampled |
-> Int64 | num_true |
-> Int64 | range_max |
-> Bool | unique |
-> Tensor v'1 Int64 | true_classes |
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float) | (sampled_candidates, true_expected_count, sampled_expected_count)
|
:: (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) | |
=> Tensor v'1 t | data |
-> Tensor v'2 tindices | segment_ids |
-> Tensor v'3 tnumsegments | num_segments |
-> Tensor Build t | output |
:: (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) | |
=> OpParams | |
-> Tensor v'1 t | data |
-> Tensor v'2 tindices | segment_ids |
-> Tensor v'3 tnumsegments | num_segments |
-> Tensor Build t | output |
:: (MonadBuild m', TensorTypes dtypes) | |
=> m' (TensorList Value dtypes) | values |
:: (MonadBuild m', TensorTypes dtypes) | |
=> OpParams | |
-> m' (TensorList Value dtypes) | values |
:: MonadBuild m' | |
=> DataType | dtype |
-> Shape | shape |
-> m' (Tensor Value ResourceHandle) | resource |
:: MonadBuild m' | |
=> OpParams | |
-> DataType | dtype |
-> Shape | shape |
-> m' (Tensor Value ResourceHandle) | resource |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | resource |
-> m' (Tensor Value Bool) | is_initialized |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | resource |
-> m' (Tensor Value Bool) | is_initialized |
:: (MonadBuild m', TensorType dtype) | |
=> Shape | shape |
-> m' (Tensor Ref dtype) | ref |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Shape | shape |
-> m' (Tensor Ref dtype) | ref |
:: (MonadBuild m', OneOf '[Int32, Int64] out_type) | |
=> Tensor v'1 ResourceHandle | input |
-> m' (Tensor Value out_type) | output |
:: (MonadBuild m', OneOf '[Int32, Int64] out_type) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | input |
-> m' (Tensor Value out_type) | output |
:: (MonadBuild m', TensorType dtype) | |
=> Shape | shape |
-> m' (Tensor Ref dtype) | ref |
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Shape | shape |
-> m' (Tensor Ref dtype) | ref |
:: MonadBuild m' | |
=> m' (Tensor Ref ByteString) | reader_handle |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Ref ByteString) | reader_handle |
:: MonadBuild m' | |
=> m' (Tensor Value ResourceHandle) | reader_handle |
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value ResourceHandle) | reader_handle |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | request: A string tensor containing a serialized WorkerHeartbeatRequest |
-> m' (Tensor Value ByteString) | response: A string tensor containing a serialized WorkerHeartbeatResponse |
Worker heartbeat op.
Heartbeats may be sent periodically to indicate the coordinator is still active, to retrieve the current worker status and to expedite shutdown when necessary.
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | request: A string tensor containing a serialized WorkerHeartbeatRequest |
-> m' (Tensor Value ByteString) | response: A string tensor containing a serialized WorkerHeartbeatResponse |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 Int64 | step |
-> Tensor v'3 ByteString | tag |
-> Tensor v'4 Float | tensor |
-> Tensor v'5 Float | sample_rate |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 Int64 | step |
-> Tensor v'3 ByteString | tag |
-> Tensor v'4 Float | tensor |
-> Tensor v'5 Float | sample_rate |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | filename |
-> Tensor v'2 ByteString | contents |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | filename |
-> Tensor v'2 ByteString | contents |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 Int64 | step |
-> Tensor v'3 ByteString | tensor |
-> m' ControlNode |
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 Int64 | step |
-> Tensor v'3 ByteString | tensor |
-> m' ControlNode |
writeHistogramSummary Source #
:: (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 Int64 | step |
-> Tensor v'3 ByteString | tag |
-> Tensor v'4 t | values |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Word16, Word8, Float] t) | |
=> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 Int64 | step |
-> Tensor v'3 ByteString | tag |
-> Tensor v'4 t | tensor |
-> Tensor v'5 Word8 | bad_color |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Word16, Word8, Float] t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 Int64 | step |
-> Tensor v'3 ByteString | tag |
-> Tensor v'4 t | tensor |
-> Tensor v'5 Word8 | bad_color |
-> m' ControlNode |
:: (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) | |
=> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 Int64 | step |
-> Tensor v'3 ByteString | tag |
-> Tensor v'4 t | value |
-> m' ControlNode |
:: (MonadBuild m', TensorType t) | |
=> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 Int64 | step |
-> Tensor v'3 t | tensor |
-> Tensor v'4 ByteString | tag |
-> Tensor v'5 ByteString | summary_metadata |
-> m' ControlNode |
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Tensor v'1 ResourceHandle | writer |
-> Tensor v'2 Int64 | step |
-> Tensor v'3 t | tensor |
-> Tensor v'4 ByteString | tag |
-> Tensor v'5 ByteString | summary_metadata |
-> m' ControlNode |
:: TensorType t | |
=> OpParams | |
-> Tensor v'1 t | x |
-> Tensor Build t | y |
:: (MonadBuild m', TensorType t) | |
=> Int64 | index: This argument is the index-th argument of the function. |
-> m' (Tensor Value t) | output: The argument. |
A graph node which represents an argument to a function.
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Int64 | index: This argument is the index-th argument of the function. |
-> m' (Tensor Value t) | output: The argument. |
:: (TensorType t, TensorTypes out_types) | |
=> [Tensor v'1 t] | input |
-> TensorList Build out_types | output |
Converts an array of tensors to a list of tensors.
:: (TensorType t, TensorTypes out_types) | |
=> OpParams | |
-> [Tensor v'1 t] | input |
-> TensorList Build out_types | output |
_ConfigureDistributedTPU Source #
:: MonadBuild m' | |
=> [Tensor v'1 Int32] | inputs: A scalar tensor for each host indicating how many TPU chips there are on the host. |
-> m' (Tensor Value ByteString) | output: A tensor containing a TPUHostConfiguration proto serialized to a string, containing the information necessary to initialize the chips in a host. |
An op that sets up the centralized structures for a distributed TPU
system.
_ConfigureDistributedTPU' Source #
:: MonadBuild m' | |
=> OpParams | |
-> [Tensor v'1 Int32] | inputs: A scalar tensor for each host indicating how many TPU chips there are on the host. |
-> m' (Tensor Value ByteString) | output: A tensor containing a TPUHostConfiguration proto serialized to a string, containing the information necessary to initialize the chips in a host. |
_DisconnectHostFromDistributedTPUSystem Source #
:: MonadBuild m' | |
=> m' (Tensor Value Int32) | number_of_tpu_chips: A scalar tensor containing the number of TPU chips on the host. |
An op that disconnects the TPUs on a host from a running distributed
TPU system.
_DisconnectHostFromDistributedTPUSystem' Source #
:: MonadBuild m' | |
=> OpParams | |
-> m' (Tensor Value Int32) | number_of_tpu_chips: A scalar tensor containing the number of TPU chips on the host. |
:: (TensorType srcT, TensorType dstT) | |
=> Tensor v'1 srcT | x |
-> Tensor Build dstT | y |
Cast x of type SrcT to y of DstT.
_HostCast requires its input and produces its output in host memory.
:: (TensorType srcT, TensorType dstT) | |
=> OpParams | |
-> Tensor v'1 srcT | x |
-> Tensor Build dstT | y |
:: (MonadBuild m', TensorType tensor_type) | |
=> Int64 | send_device_incarnation: The current incarnation of send_device. |
-> m' (Tensor Value tensor_type) | tensor: The tensor to receive. |
Receives the named tensor from send_device on recv_device.
_HostRecv requires its input on host memory whereas _Recv requires its input on device memory.
:: (MonadBuild m', TensorType tensor_type) | |
=> OpParams | |
-> Int64 | send_device_incarnation: The current incarnation of send_device. |
-> m' (Tensor Value tensor_type) | tensor: The tensor to receive. |
:: (MonadBuild m', TensorType t) | |
=> Int64 | send_device_incarnation: The current incarnation of send_device. |
-> Tensor v'1 t | tensor: The tensor to send. |
-> m' ControlNode |
Sends the named tensor from send_device to recv_device.
_HostSend requires its input on host memory whereas _Send requires its input on device memory.
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Int64 | send_device_incarnation: The current incarnation of send_device. |
-> Tensor v'1 t | tensor: The tensor to send. |
-> m' ControlNode |
_InitializeHostForDistributedTPU Source #
:: MonadBuild m' | |
=> Tensor v'1 ByteString | input: A string containing the address of the UberDriver to connect to. |
-> m' (Tensor Value Int32) | tpu_ids: A vector containing the global TPU id of each TPU on the host. |
An op that connects each chip on the host to a centralized UberDriver to allow
them to operate as a distributed system with chips in other hosts.
_InitializeHostForDistributedTPU' Source #
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | input: A string containing the address of the UberDriver to connect to. |
-> m' (Tensor Value Int32) | tpu_ids: A vector containing the global TPU id of each TPU on the host. |
:: (TensorTypes tin, TensorType t) | |
=> Int64 | N |
-> TensorList v'1 tin | input |
-> [Tensor Build t] | output |
Converts a list of tensors to an array of tensors.
:: (TensorTypes tin, TensorType t) | |
=> OpParams | |
-> Int64 | N |
-> TensorList v'1 tin | input |
-> [Tensor Build t] | output |
:: OneOf '[Complex Double, Complex Float, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t | |
=> Tensor v'1 t | x |
-> Tensor v'2 t | y |
-> Tensor v'3 Word8 | mkl_x |
-> Tensor v'4 Word8 | mkl_y |
-> (Tensor Build t, Tensor Build Word8) | (z, mkl_z)
|
Returns x + y element-wise.
- NOTE*:
Add
supports broadcasting.AddN
does not. More about broadcasting here
:: OneOf '[Int32, Int64, Word16, Double, Float] t | |
=> Tensor v'1 t | x |
-> Tensor v'2 t | y |
-> Tensor v'3 Word8 | mkl_x |
-> Tensor v'4 Word8 | mkl_y |
-> (Tensor Build t, Tensor Build Word8) | (z, mkl_z)
|
Returns the max of x and y (i.e. x > y ? x : y) element-wise.
- NOTE*:
Maximum
supports broadcasting. More about broadcasting here
:: OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t | |
=> Tensor v'1 t | x |
-> Tensor v'2 t | y |
-> Tensor v'3 Word8 | mkl_x |
-> Tensor v'4 Word8 | mkl_y |
-> (Tensor Build t, Tensor Build Word8) | (z, mkl_z)
|
Returns x * y element-wise.
- NOTE*:
Mul
supports broadcasting. More about broadcasting here
_MklSquaredDifference Source #
:: OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t | |
=> Tensor v'1 t | x |
-> Tensor v'2 t | y |
-> Tensor v'3 Word8 | mkl_x |
-> Tensor v'4 Word8 | mkl_y |
-> (Tensor Build t, Tensor Build Word8) | (z, mkl_z)
|
Returns (x - y)(x - y) element-wise.
- NOTE*:
SquaredDifference
supports broadcasting. More about broadcasting here
:: OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t | |
=> Tensor v'1 t | x |
-> Tensor v'2 t | y |
-> Tensor v'3 Word8 | mkl_x |
-> Tensor v'4 Word8 | mkl_y |
-> (Tensor Build t, Tensor Build Word8) | (z, mkl_z)
|
Returns x - y element-wise.
- NOTE*:
Sub
supports broadcasting. More about broadcasting here
:: (MonadBuild m', TensorType dtype) | |
=> Shape | shape: 1-D |
-> m' (Tensor Value dtype) | output: An empty Tensor of the specified type. |
Creates an empty Tensor with shape shape
and type dtype
.
The memory can optionally be initialized. This is usually useful in conjunction with inplace operations.
_ParallelConcatStart' Source #
:: (MonadBuild m', TensorType dtype) | |
=> OpParams | |
-> Shape | shape: 1-D |
-> m' (Tensor Value dtype) | output: An empty Tensor of the specified type. |
_ParallelConcatUpdate Source #
:: TensorType t | |
=> Int64 | loc: A scalar indicating the index of the first dimension such that value[loc, :] is updated. |
-> Tensor v'1 t | value: A |
-> Tensor v'2 t | update: A |
-> Tensor Build t | output: |
Updates input value
at loc
with update
.
If you use this function you will almost certainly want to add a control dependency as done in the implementation of parallel_stack to avoid race conditions.
_ParallelConcatUpdate' Source #
:: TensorType t | |
=> OpParams | |
-> Int64 | loc: A scalar indicating the index of the first dimension such that value[loc, :] is updated. |
-> Tensor v'1 t | value: A |
-> Tensor v'2 t | update: A |
-> Tensor Build t | output: |
:: (MonadBuild m', TensorType tensor_type) | |
=> Int64 | send_device_incarnation: The current incarnation of send_device. |
-> m' (Tensor Value tensor_type) | tensor: The tensor to receive. |
Receives the named tensor from send_device on recv_device.
:: (MonadBuild m', TensorType tensor_type) | |
=> OpParams | |
-> Int64 | send_device_incarnation: The current incarnation of send_device. |
-> m' (Tensor Value tensor_type) | tensor: The tensor to receive. |
:: (MonadBuild m', TensorType t) | |
=> Int64 | index: This return value is the index-th return value of the function. |
-> Tensor v'1 t | input: The return value. |
-> m' ControlNode |
A graph node which represents a return value of a function.
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Int64 | index: This return value is the index-th return value of the function. |
-> Tensor v'1 t | input: The return value. |
-> m' ControlNode |
:: (MonadBuild m', TensorType t) | |
=> Int64 | expected_call_count |
-> Int64 | id |
-> Shape | shape |
-> m' (Tensor Value t) | output |
Allocates a mutable tensor that becomes available to appropriately annotated
downstream Ops as backing store for their output tensor allocations via the ScopedAllocatorMgr. Returns a reference to this value.
This is an experimental op for internal use only. It is possible to use this op in unsafe ways.
shapes
is a list of the shapes of the tensors that are to be allocated
by this ScopedAllocator.
shape
is the shape of the output of this Op, i.e. the 1D backing tensor
from which the individual allocated tensors are aliased.
sa_name
is the name assigned to the Node, for connectivity specification
and debugging.
id
is a non-negative integer scope_id
handled by the ScopedAllocatorMgr.
expected_call_count
is the number of individual tensors expected to
be allocated from the backing tensor.
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Int64 | expected_call_count |
-> Int64 | id |
-> Shape | shape |
-> m' (Tensor Value t) | output |
_ScopedAllocatorConcat Source #
:: (MonadBuild m', TensorType t) | |
=> Int64 | id |
-> Shape | shape |
-> Tensor v'1 t | backing |
-> [Tensor v'2 t] | inputs |
-> m' (Tensor Value t) | output |
Acts like a Concat Op that merges multple tensors into one, however it must
only be used in conjunction with a ScopedAllocator which is backing the memory of all of its input tensors so that actually it just outputs a read-only reference to that ScopedAllocator's backing tensor.
This is an experimental op for internal use only. It is possible to use this op in unsafe ways.
backing
is the backing tensor, i.e. the output of an upstream ScopedAllocator.
inputs
is a list of nominal input tensors, all of which must be aliases
to regions of the backing tensor. These will be outputs of upstream nodes
that allocate their outputs from the same ScopedAllocator.
shape
is the shape of the output, which will usually be the same shape as
the input backing tensor.
reshape
is true iff the output shape is to be different from that of
the input backing tensor.
sa_name
is the Node name of the upstream ScopedAllocator.
id
is the scope_id identifying the upstream ScopedAllocator.
N
is the number of nominal inputs to be concatenated.
_ScopedAllocatorConcat' Source #
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Int64 | id |
-> Shape | shape |
-> Tensor v'1 t | backing |
-> [Tensor v'2 t] | inputs |
-> m' (Tensor Value t) | output |
_ScopedAllocatorSplit Source #
:: (MonadBuild m', TensorType t) | |
=> Int64 | id |
-> Tensor v'1 t | concat |
-> [Tensor v'2 t] | split |
-> m' [Tensor Value t] | output |
Acts roughly like a SplitV Op that splits one tensor into multiple tensors
but must only be used in conjunction with corresponding ScopedAllocator and ScopedAllocatorConcat instances. In practice it is provided as inputs the backing tensor as first input, which contains the concatenated values, and a list of alias tensors as its other input and it simply outputs that second list.
This is an experimental op for internal use only. It is possible to use this op in unsafe ways.
concat
is the single output produced by an upstream ScopedAllocatorConcat
node. This is actually the backing tensor from a ScopedAllocator node
upstream of the ScopedAllocatorConcat.
split
is a list of tensors aliased from the backing tensor. It will
become the output of this ScopedAllocatorSplit node.
'type' is the common DataType of all of the input and output tensors.
sa_name
is the Node name of the upstream ScopedAllocator.
id
is the scope_id identifying the upstream ScopedAllocator.
N
is the number of split tensors.
shapes
is a list of the split tensor shapes.
_ScopedAllocatorSplit' Source #
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Int64 | id |
-> Tensor v'1 t | concat |
-> [Tensor v'2 t] | split |
-> m' [Tensor Value t] | output |
:: (MonadBuild m', TensorType t) | |
=> Int64 | send_device_incarnation: The current incarnation of send_device. |
-> Tensor v'1 t | tensor: The tensor to send. |
-> m' ControlNode |
Sends the named tensor from send_device to recv_device.
:: (MonadBuild m', TensorType t) | |
=> OpParams | |
-> Int64 | send_device_incarnation: The current incarnation of send_device. |
-> Tensor v'1 t | tensor: The tensor to send. |
-> m' ControlNode |
:: MonadBuild m' | |
=> Tensor v'1 ByteString | topology: A serialized tensorflow.tpu.TopologyProto that describes the TPU topology. |
-> m' ControlNode |
An op that informs a host of the global ids of all the of TPUs in the
system.
:: MonadBuild m' | |
=> OpParams | |
-> Tensor v'1 ByteString | topology: A serialized tensorflow.tpu.TopologyProto that describes the TPU topology. |
-> m' ControlNode |
_ShutdownDistributedTPU :: forall m'. MonadBuild m' => m' ControlNode Source #
An op that shuts down a running distributed TPU system. The Op returns
an error if no system is running. This Op must be run on the same TPU_SYSTEM device as the corresponding _ConfigureDistributedTPU was run to start the system, and must be run only after _DisconnectHostFromDistributedTPUSystem has completed on every host in the system.
_ShutdownDistributedTPU' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode Source #
_WaitForDistributedTPU Source #
:: MonadBuild m' | |
=> [Tensor v'1 Int32] | inputs: For each initialized host, a vector giving the global TPU id of each TPU on the host. |
-> m' (Tensor Value ByteString) | topology: A serialized tensorflow.tpu.TopologyProto that describes the TPU topology. |
An op that blocks execution until a distributed TPU system has
started up. This Op must be run on the same TPU_SYSTEM device as _ConfigureDistributedTPU, and takes an inputs the outputs from the _InitializeHostForDistributedTPU Ops.
_WaitForDistributedTPU' Source #
:: MonadBuild m' | |
=> OpParams | |
-> [Tensor v'1 Int32] | inputs: For each initialized host, a vector giving the global TPU id of each TPU on the host. |
-> m' (Tensor Value ByteString) | topology: A serialized tensorflow.tpu.TopologyProto that describes the TPU topology. |
:: (MonadBuild m', TensorTypes toutputs) | |
=> Int64 | device_ordinal: The device to use. |
-> Tensor v'1 ByteString | dynamic_key: The key sent at runtime by the compile node to identify which execution the transfer corresponds to. |
-> m' (TensorList Value toutputs) | outputs: A list of tensors that will be received from the XLA computation. |
A placeholder op for multiple values that will be sent to TensorFlow from a
running XLA computation.
:: (MonadBuild m', TensorTypes toutputs) | |
=> OpParams | |
-> Int64 | device_ordinal: The device to use. |
-> Tensor v'1 ByteString | dynamic_key: The key sent at runtime by the compile node to identify which execution the transfer corresponds to. |
-> m' (TensorList Value toutputs) | outputs: A list of tensors that will be received from the XLA computation. |
:: (MonadBuild m', TensorTypes tinputs) | |
=> Int64 | device_ordinal: The device to use. |
-> TensorList v'1 tinputs | inputs: A list of tensors that will be sent to the XLA computation. |
-> Tensor v'2 ByteString | dynamic_key: The key sent at runtime by the compile node to identify which execution the transfer corresponds to. |
-> m' ControlNode |
A placeholder op for multiple values that will be sent from TensorFlow to a
running XLA computation.
:: (MonadBuild m', TensorTypes tinputs) | |
=> OpParams | |
-> Int64 | device_ordinal: The device to use. |
-> TensorList v'1 tinputs | inputs: A list of tensors that will be sent to the XLA computation. |
-> Tensor v'2 ByteString | dynamic_key: The key sent at runtime by the compile node to identify which execution the transfer corresponds to. |
-> m' ControlNode |