Safe Haskell	None
Language	Haskell2010

TensorFlow.GenOps.Core

Synopsis

abort :: forall m'. MonadBuild m' => m' ControlNode
abort' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode
abs :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
abs' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, 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, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
acos' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, 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 :: forall v'1 v'2 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustContrastv2' :: forall v'1 v'2 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustHue :: forall v'1 v'2 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustHue' :: forall v'1 v'2 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustSaturation :: forall v'1 v'2 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustSaturation' :: forall v'1 v'2 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
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)
allToAll :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Int64 -> Int64 -> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
allToAll' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Int64 -> Int64 -> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
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)
anonymousIteratorV2 :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousIteratorV2' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousMemoryCache :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousMemoryCache' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousMultiDeviceIterator :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousMultiDeviceIterator' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousRandomSeedGenerator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousRandomSeedGenerator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousSeedGenerator :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 Int64 -> Tensor v'2 Int64 -> Tensor v'3 Bool -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousSeedGenerator' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> Tensor v'3 Bool -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
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)
applyAdagradV2 :: 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)
applyAdagradV2' :: 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)
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, Bool, 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, Bool, 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, Bool, 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, Bool, 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 Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Double, Float] t => Tensor v'1 t -> Tensor Build ByteString
asString' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build ByteString
asin :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
asin' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, 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
assertCardinalityDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
assertCardinalityDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
assertNextDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
assertNextDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
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, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
atan' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, 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
autoShardDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
autoShardDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
avgPool :: forall v'1 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
avgPool' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
avgPool3D :: forall v'1 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
avgPool3D' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
avgPool3DGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
avgPool3DGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
avgPoolGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
avgPoolGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
bandedTriangularSolve :: 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
bandedTriangularSolve' :: 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
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
batchDatasetV2 :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Bool -> Tensor Build Variant
batchDatasetV2' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Bool -> 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, Int64, 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, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatMulV2 :: 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
batchMatMulV2' :: 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
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
besselI0 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselI0' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
besselI0e :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselI0e' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
besselI1 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselI1' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
besselI1e :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselI1e' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
besselJ0 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselJ0' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
besselJ1 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselJ1' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
besselK0 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselK0' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
besselK0e :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselK0e' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
besselK1 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselK1' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
besselK1e :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselK1e' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
besselY0 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselY0' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
besselY1 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
besselY1' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> 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
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, Word32, Word64, Word8, Double, Float] t, OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, 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, Word32, Word64, Word8, Double, Float] t, OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, 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
blockLSTM :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t. OneOf '[Word16, Float] t => Tensor v'1 Int64 -> 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 Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTM' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 Int64 -> 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 Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMGrad :: 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 v'16 v'17 v'18 t. OneOf '[Word16, Float] t => Bool -> Tensor v'1 Int64 -> 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 t -> Tensor v'13 t -> Tensor v'14 t -> Tensor v'15 t -> Tensor v'16 t -> Tensor v'17 t -> Tensor v'18 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMGrad' :: 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 v'16 v'17 v'18 t. OneOf '[Word16, Float] t => OpParams -> Bool -> Tensor v'1 Int64 -> 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 t -> Tensor v'13 t -> Tensor v'14 t -> Tensor v'15 t -> Tensor v'16 t -> Tensor v'17 t -> Tensor v'18 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMGradV2 :: 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 v'16 v'17 v'18 t. OneOf '[Word16, Float] t => Bool -> Tensor v'1 Int64 -> 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 t -> Tensor v'13 t -> Tensor v'14 t -> Tensor v'15 t -> Tensor v'16 t -> Tensor v'17 t -> Tensor v'18 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMGradV2' :: 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 v'16 v'17 v'18 t. OneOf '[Word16, Float] t => OpParams -> Bool -> Tensor v'1 Int64 -> 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 t -> Tensor v'13 t -> Tensor v'14 t -> Tensor v'15 t -> Tensor v'16 t -> Tensor v'17 t -> Tensor v'18 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t. OneOf '[Word16, Float] t => Tensor v'1 Int64 -> 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 Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 Int64 -> 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 Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
boostedTreesAggregateStats :: Int64 -> Int64 -> Tensor v'1 Int32 -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Int32 -> Tensor Build Float
boostedTreesAggregateStats' :: OpParams -> Int64 -> Int64 -> Tensor v'1 Int32 -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Int32 -> Tensor Build Float
boostedTreesBucketize :: [Tensor v'1 Float] -> [Tensor v'2 Float] -> [Tensor Build Int32]
boostedTreesBucketize' :: OpParams -> [Tensor v'1 Float] -> [Tensor v'2 Float] -> [Tensor Build Int32]
boostedTreesCalculateBestFeatureSplit :: 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 Int32, Tensor Build Float, Tensor Build Float, Tensor Build ByteString)
boostedTreesCalculateBestFeatureSplit' :: 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 Int32, Tensor Build Float, Tensor Build Float, Tensor Build ByteString)
boostedTreesCalculateBestFeatureSplitV2 :: Int64 -> Tensor v'1 Int32 -> [Tensor v'2 Float] -> Tensor v'3 ByteString -> Tensor v'4 Int32 -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32, Tensor Build Int32, Tensor Build Int32, Tensor Build Float, Tensor Build Float, Tensor Build ByteString)
boostedTreesCalculateBestFeatureSplitV2' :: OpParams -> Int64 -> Tensor v'1 Int32 -> [Tensor v'2 Float] -> Tensor v'3 ByteString -> Tensor v'4 Int32 -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32, Tensor Build Int32, Tensor Build Int32, Tensor Build Float, Tensor Build Float, Tensor Build ByteString)
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])
boostedTreesCenterBias :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> m' (Tensor Value Bool)
boostedTreesCenterBias' :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> m' (Tensor Value Bool)
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
boostedTreesCreateQuantileStreamResource :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Float -> Tensor v'3 Int64 -> m' ControlNode
boostedTreesCreateQuantileStreamResource' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> Tensor v'3 Int64 -> 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)
boostedTreesExampleDebugOutputs :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> [Tensor v'2 Int32] -> m' (Tensor Value ByteString)
boostedTreesExampleDebugOutputs' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> [Tensor v'2 Int32] -> m' (Tensor Value ByteString)
boostedTreesFlushQuantileSummaries :: forall v'1 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
boostedTreesFlushQuantileSummaries' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
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)
boostedTreesMakeQuantileSummaries :: [Tensor v'1 Float] -> Tensor v'2 Float -> Tensor v'3 Float -> [Tensor Build Float]
boostedTreesMakeQuantileSummaries' :: OpParams -> [Tensor v'1 Float] -> Tensor v'2 Float -> Tensor v'3 Float -> [Tensor Build Float]
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)
boostedTreesQuantileStreamResourceAddSummaries :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> [Tensor v'2 Float] -> m' ControlNode
boostedTreesQuantileStreamResourceAddSummaries' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> [Tensor v'2 Float] -> m' ControlNode
boostedTreesQuantileStreamResourceDeserialize :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> [Tensor v'2 Float] -> m' ControlNode
boostedTreesQuantileStreamResourceDeserialize' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> [Tensor v'2 Float] -> m' ControlNode
boostedTreesQuantileStreamResourceFlush :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
boostedTreesQuantileStreamResourceFlush' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
boostedTreesQuantileStreamResourceGetBucketBoundaries :: forall v'1 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
boostedTreesQuantileStreamResourceGetBucketBoundaries' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
boostedTreesQuantileStreamResourceHandleOp :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
boostedTreesQuantileStreamResourceHandleOp' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
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)
boostedTreesSparseAggregateStats :: Int64 -> Int64 -> Tensor v'1 Int32 -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Int32 -> Tensor v'5 Int32 -> Tensor v'6 Int32 -> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32)
boostedTreesSparseAggregateStats' :: OpParams -> Int64 -> Int64 -> Tensor v'1 Int32 -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Int32 -> Tensor v'5 Int32 -> Tensor v'6 Int32 -> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32)
boostedTreesSparseCalculateBestFeatureSplit :: Int64 -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor v'4 Int32 -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32, Tensor Build Int32, Tensor Build Float, Tensor Build Float, Tensor Build ByteString)
boostedTreesSparseCalculateBestFeatureSplit' :: OpParams -> Int64 -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor v'4 Int32 -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32, Tensor Build Int32, Tensor Build Float, Tensor Build Float, Tensor Build 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
boostedTreesUpdateEnsembleV2 :: 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 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> [Tensor v'2 Int32] -> [Tensor v'3 Int32] -> [Tensor v'4 Int32] -> [Tensor v'5 Float] -> [Tensor v'6 Int32] -> [Tensor v'7 Float] -> [Tensor v'8 Float] -> [Tensor v'9 ByteString] -> Tensor v'10 Int32 -> Tensor v'11 Float -> Tensor v'12 Int32 -> m' ControlNode
boostedTreesUpdateEnsembleV2' :: 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 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> [Tensor v'2 Int32] -> [Tensor v'3 Int32] -> [Tensor v'4 Int32] -> [Tensor v'5 Float] -> [Tensor v'6 Int32] -> [Tensor v'7 Float] -> [Tensor v'8 Float] -> [Tensor v'9 ByteString] -> Tensor v'10 Int32 -> Tensor v'11 Float -> Tensor v'12 Int32 -> 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
cSRSparseMatrixComponents :: forall v'1 v'2 type'. OneOf '[Complex Double, Complex Float, Double, Float] type' => Tensor v'1 Variant -> Tensor v'2 Int32 -> (Tensor Build Int32, Tensor Build Int32, Tensor Build type')
cSRSparseMatrixComponents' :: forall v'1 v'2 type'. OneOf '[Complex Double, Complex Float, Double, Float] type' => OpParams -> Tensor v'1 Variant -> Tensor v'2 Int32 -> (Tensor Build Int32, Tensor Build Int32, Tensor Build type')
cSRSparseMatrixToDense :: forall v'1 type'. OneOf '[Complex Double, Complex Float, Double, Float] type' => Tensor v'1 Variant -> Tensor Build type'
cSRSparseMatrixToDense' :: forall v'1 type'. OneOf '[Complex Double, Complex Float, Double, Float] type' => OpParams -> Tensor v'1 Variant -> Tensor Build type'
cSRSparseMatrixToSparseTensor :: forall v'1 type'. OneOf '[Complex Double, Complex Float, Double, Float] type' => Tensor v'1 Variant -> (Tensor Build Int64, Tensor Build type', Tensor Build Int64)
cSRSparseMatrixToSparseTensor' :: forall v'1 type'. OneOf '[Complex Double, Complex Float, Double, Float] type' => OpParams -> Tensor v'1 Variant -> (Tensor Build Int64, Tensor Build type', Tensor Build Int64)
cSVDataset :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 output_types m'. (MonadBuild m', OneOfs '[ByteString, Int32, Int64, Double, Float] output_types) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> Tensor v'4 Bool -> Tensor v'5 ByteString -> Tensor v'6 Bool -> Tensor v'7 ByteString -> Tensor v'8 Int64 -> TensorList v'9 output_types -> m' (Tensor Value Variant)
cSVDataset' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 output_types m'. (MonadBuild m', OneOfs '[ByteString, Int32, Int64, Double, Float] output_types) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> Tensor v'4 Bool -> Tensor v'5 ByteString -> Tensor v'6 Bool -> Tensor v'7 ByteString -> Tensor v'8 Int64 -> TensorList v'9 output_types -> m' (Tensor Value Variant)
cTCBeamSearchDecoder :: forall v'1 v'2 t. OneOf '[Double, Float] t => Int64 -> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> ([Tensor Build Int64], [Tensor Build Int64], [Tensor Build Int64], Tensor Build t)
cTCBeamSearchDecoder' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Int64 -> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> ([Tensor Build Int64], [Tensor Build Int64], [Tensor Build Int64], Tensor Build t)
cTCGreedyDecoder :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64, Tensor Build t)
cTCGreedyDecoder' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64, Tensor Build t)
cTCLoss :: forall v'1 v'2 v'3 v'4 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> (Tensor Build t, Tensor Build t)
cTCLoss' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> (Tensor Build t, Tensor Build t)
cTCLossV2 :: Tensor v'1 Float -> Tensor v'2 Int64 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> (Tensor Build Float, Tensor Build Float)
cTCLossV2' :: 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
cacheDatasetV2 :: forall v'1 v'2 v'3 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor v'3 ResourceHandle -> m' (Tensor Value Variant)
cacheDatasetV2' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor v'3 ResourceHandle -> m' (Tensor Value 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 m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => ByteString -> Tensor v'1 t -> m' (Tensor Value t)
checkNumerics' :: forall v'1 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => OpParams -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
checkNumericsV2 :: forall v'1 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => ByteString -> Tensor v'1 t -> m' (Tensor Value t)
checkNumericsV2' :: forall v'1 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => OpParams -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
cholesky :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
cholesky' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
choleskyGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
choleskyGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
chooseFastestDataset :: Int64 -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
chooseFastestDataset' :: OpParams -> Int64 -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
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 '[Bool, Int32, Int64, Word16, Double, Float] t) => Int64 -> Int64 -> Int64 -> Shape -> m' (Tensor Value t)
collectiveBcastRecv' :: forall t m'. (MonadBuild m', OneOf '[Bool, Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> Int64 -> Int64 -> Shape -> m' (Tensor Value t)
collectiveBcastSend :: forall v'1 t m'. (MonadBuild m', OneOf '[Bool, 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 '[Bool, Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> Int64 -> Int64 -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
collectiveGather :: 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)
collectiveGather' :: 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)
collectivePermute :: 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 Int32 -> Tensor Build t
collectivePermute' :: 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 Int32 -> Tensor Build t
collectiveReduce :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => ByteString -> Int64 -> Int64 -> Int64 -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
collectiveReduce' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> ByteString -> Int64 -> Int64 -> Int64 -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
combinedNonMaxSuppression :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build Float, Tensor Build Float, Tensor Build Float, Tensor Build Int32)
combinedNonMaxSuppression' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 Float -> Tensor v'6 Float -> (Tensor Build Float, Tensor Build Float, Tensor Build Float, Tensor Build Int32)
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
compressElement :: forall v'1 input_types. TensorTypes input_types => TensorList v'1 input_types -> Tensor Build Variant
compressElement' :: forall v'1 input_types. TensorTypes input_types => OpParams -> TensorList v'1 input_types -> Tensor Build Variant
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)
configureTPUEmbedding :: forall m'. MonadBuild m' => ByteString -> m' ControlNode
configureTPUEmbedding' :: forall m'. MonadBuild m' => OpParams -> ByteString -> m' ControlNode
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 '[Int32, Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
conv2D' :: forall v'1 v'2 t. OneOf '[Int32, Word16, Double, Float] t => OpParams -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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 '[Int32, Word16, Double, Float] t => ByteString -> 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 '[Int32, Word16, Double, Float] t => OpParams -> ByteString -> 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 => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
conv3D' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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) => ByteString -> 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 -> ByteString -> 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 v'2 t. OneOf '[Int32, Word16, Word32, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
crossReplicaSum' :: forall v'1 v'2 t. OneOf '[Int32, Word16, Word32, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> 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)
cudnnRNNBackpropV3 :: 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 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 Int32 -> 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 t -> Tensor v'13 Int8 -> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNNBackpropV3' :: 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 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 Int32 -> 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 t -> Tensor v'13 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
cudnnRNNCanonicalToParamsV2 :: 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
cudnnRNNCanonicalToParamsV2' :: 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])
cudnnRNNParamsToCanonicalV2 :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => Int64 -> Int64 -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 t -> ([Tensor Build t], [Tensor Build t])
cudnnRNNParamsToCanonicalV2' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => OpParams -> Int64 -> 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)
cudnnRNNV3 :: forall v'1 v'2 v'3 v'4 v'5 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 Int32 -> m' (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value Int8)
cudnnRNNV3' :: forall v'1 v'2 v'3 v'4 v'5 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 Int32 -> 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
cumulativeLogsumexp :: forall v'1 v'2 t tidx. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
cumulativeLogsumexp' :: forall v'1 v'2 t tidx. (OneOf '[Word16, 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
dataServiceDataset :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 m'. MonadBuild m' => [DataType] -> Tensor v'1 Int64 -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> Tensor v'5 ByteString -> Tensor v'6 Int64 -> Tensor v'7 ResourceHandle -> m' (Tensor Value Variant)
dataServiceDataset' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Int64 -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> Tensor v'5 ByteString -> Tensor v'6 Int64 -> Tensor v'7 ResourceHandle -> m' (Tensor Value Variant)
datasetCardinality :: Tensor v'1 Variant -> Tensor Build Int64
datasetCardinality' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int64
datasetFromGraph :: Tensor v'1 ByteString -> Tensor Build Variant
datasetFromGraph' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Variant
datasetToGraph :: Tensor v'1 Variant -> Tensor Build ByteString
datasetToGraph' :: OpParams -> Tensor v'1 Variant -> Tensor Build ByteString
datasetToGraphV2 :: Tensor v'1 Variant -> Tensor Build ByteString
datasetToGraphV2' :: OpParams -> Tensor v'1 Variant -> Tensor Build ByteString
datasetToSingleElement :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => Tensor v'1 Variant -> m' (TensorList Value output_types)
datasetToSingleElement' :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => OpParams -> Tensor v'1 Variant -> m' (TensorList Value 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
dawsn :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
dawsn' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
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
debugIdentityV2 :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value t)
debugIdentityV2' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value 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
debugNumericSummaryV2 :: forall v'1 output_dtype t. (OneOf '[Double, Float] output_dtype, TensorType t) => Tensor v'1 t -> Tensor Build output_dtype
debugNumericSummaryV2' :: forall v'1 output_dtype t. (OneOf '[Double, Float] output_dtype, TensorType t) => OpParams -> Tensor v'1 t -> Tensor Build output_dtype
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
decodePaddedRaw :: forall v'1 v'2 out_type. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] out_type => Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build out_type
decodePaddedRaw' :: forall v'1 v'2 out_type. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] out_type => OpParams -> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build out_type
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 => ByteString -> Tensor v'1 ByteString -> (Tensor Build Int32, TensorList Build output_types)
decodeProtoV2' :: forall v'1 output_types. TensorTypes output_types => OpParams -> ByteString -> Tensor v'1 ByteString -> (Tensor Build Int32, TensorList Build output_types)
decodeRaw :: forall v'1 out_type. OneOf '[Complex Double, Complex Float, Bool, 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 '[Complex Double, Complex Float, Bool, 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)
deleteIterator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deleteIterator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deleteMemoryCache :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deleteMemoryCache' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deleteMultiDeviceIterator :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> [Tensor v'2 ResourceHandle] -> Tensor v'3 Variant -> m' ControlNode
deleteMultiDeviceIterator' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> [Tensor v'2 ResourceHandle] -> Tensor v'3 Variant -> m' ControlNode
deleteRandomSeedGenerator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deleteRandomSeedGenerator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deleteSeedGenerator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deleteSeedGenerator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
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
denseBincount :: forall v'1 v'2 v'3 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t) => Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor v'3 t -> Tensor Build t
denseBincount' :: forall v'1 v'2 v'3 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t) => OpParams -> Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor v'3 t -> Tensor Build t
denseCountSparseOutput :: forall v'1 v'2 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type) => Bool -> Tensor v'1 t -> Tensor v'2 output_type -> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)
denseCountSparseOutput' :: forall v'1 v'2 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type) => OpParams -> Bool -> Tensor v'1 t -> Tensor v'2 output_type -> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)
denseToCSRSparseMatrix :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor Build Variant
denseToCSRSparseMatrix' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor Build Variant
denseToDenseSetOperation :: forall v'1 v'2 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => ByteString -> 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 -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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 => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
depthwiseConv2dNative' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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 => ByteString -> 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 -> ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
dequantize :: forall v'1 v'2 v'3 t dtype. (OneOf '[Int16, Int32, Word16, Word8] t, OneOf '[Word16, Float] dtype) => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build dtype
dequantize' :: forall v'1 v'2 v'3 t dtype. (OneOf '[Int16, Int32, Word16, Word8] t, OneOf '[Word16, Float] dtype) => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build dtype
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) => ByteString -> Tensor Ref t -> m' (Tensor Value t)
destroyTemporaryVariable' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> Tensor Ref t -> m' (Tensor Value t)
deviceIndex :: Tensor Build Int32
deviceIndex' :: OpParams -> Tensor Build Int32
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 => ByteString -> 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 -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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 => ByteString -> 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 -> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
directedInterleaveDataset :: [DataType] -> Tensor v'1 Variant -> [Tensor v'2 Variant] -> Tensor Build Variant
directedInterleaveDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> [Tensor v'2 Variant] -> Tensor Build Variant
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
divNoNan :: 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
divNoNan' :: 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
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
drawBoundingBoxesV2 :: forall v'1 v'2 v'3 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build t
drawBoundingBoxesV2' :: forall v'1 v'2 v'3 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build t
dummyIterationCounter :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
dummyIterationCounter' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
dummyMemoryCache :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
dummyMemoryCache' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
dummySeedGenerator :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
dummySeedGenerator' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
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
eagerPyFunc :: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout) => ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
eagerPyFunc' :: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout) => OpParams -> ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
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
eig :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float, Double, Float] t, OneOf '[Complex Double, Complex Float] tout) => Tensor v'1 t -> (Tensor Build tout, Tensor Build tout)
eig' :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float, Double, Float] t, OneOf '[Complex Double, Complex Float] tout) => OpParams -> Tensor v'1 t -> (Tensor Build tout, Tensor Build tout)
einsum :: forall v'1 t. TensorType t => ByteString -> [Tensor v'1 t] -> Tensor Build t
einsum' :: forall v'1 t. TensorType t => OpParams -> ByteString -> [Tensor v'1 t] -> Tensor Build t
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 v'2 shape_type. OneOf '[Int32, Int64] shape_type => DataType -> Tensor v'1 shape_type -> Tensor v'2 Int32 -> Tensor Build Variant
emptyTensorList' :: 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
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
encodeJpegVariableQuality :: Tensor v'1 Word8 -> Tensor v'2 Int32 -> Tensor Build ByteString
encodeJpegVariableQuality' :: OpParams -> Tensor v'1 Word8 -> Tensor v'2 Int32 -> 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 => ByteString -> Tensor v'1 Int32 -> TensorList v'2 tinput_types -> Tensor Build ByteString
encodeProto' :: forall v'1 v'2 tinput_types. TensorTypes tinput_types => OpParams -> ByteString -> 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
enqueueTPUEmbeddingIntegerBatch :: forall v'1 v'2 m'. MonadBuild m' => [Tensor v'1 Int32] -> Tensor v'2 ByteString -> m' ControlNode
enqueueTPUEmbeddingIntegerBatch' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> [Tensor v'1 Int32] -> Tensor v'2 ByteString -> m' ControlNode
enqueueTPUEmbeddingRaggedTensorBatch :: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3) => [Tensor v'1 t1] -> [Tensor v'2 t2] -> [Tensor v'3 t3] -> Tensor v'4 ByteString -> m' ControlNode
enqueueTPUEmbeddingRaggedTensorBatch' :: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3) => OpParams -> [Tensor v'1 t1] -> [Tensor v'2 t2] -> [Tensor v'3 t3] -> Tensor v'4 ByteString -> m' ControlNode
enqueueTPUEmbeddingSparseBatch :: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3) => [Tensor v'1 t1] -> [Tensor v'2 t2] -> [Tensor v'3 t3] -> Tensor v'4 ByteString -> m' ControlNode
enqueueTPUEmbeddingSparseBatch' :: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3) => OpParams -> [Tensor v'1 t1] -> [Tensor v'2 t2] -> [Tensor v'3 t3] -> Tensor v'4 ByteString -> m' ControlNode
enqueueTPUEmbeddingSparseTensorBatch :: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3) => [Tensor v'1 t1] -> [Tensor v'2 t2] -> [Tensor v'3 t3] -> Tensor v'4 ByteString -> m' ControlNode
enqueueTPUEmbeddingSparseTensorBatch' :: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3) => OpParams -> [Tensor v'1 t1] -> [Tensor v'2 t2] -> [Tensor v'3 t3] -> Tensor v'4 ByteString -> m' ControlNode
ensureShape :: forall v'1 t. TensorType t => Shape -> Tensor v'1 t -> Tensor Build t
ensureShape' :: forall v'1 t. TensorType t => OpParams -> Shape -> Tensor v'1 t -> Tensor Build t
enter :: forall v'1 t. TensorType t => ByteString -> Tensor v'1 t -> Tensor Build t
enter' :: forall v'1 t. TensorType t => OpParams -> ByteString -> 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, Word32, Word64, 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, Word32, Word64, 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
erfinv :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
erfinv' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
euclideanNorm :: 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
euclideanNorm' :: 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
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
experimentalAssertNextDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
experimentalAssertNextDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
experimentalAutoShardDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
experimentalAutoShardDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
experimentalBytesProducedStatsDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
experimentalBytesProducedStatsDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
experimentalCSVDataset :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 output_types m'. (MonadBuild m', OneOfs '[ByteString, Int32, Int64, Double, Float] output_types) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> Tensor v'4 Bool -> Tensor v'5 ByteString -> Tensor v'6 Bool -> Tensor v'7 ByteString -> Tensor v'8 Int64 -> TensorList v'9 output_types -> m' (Tensor Value Variant)
experimentalCSVDataset' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 output_types m'. (MonadBuild m', OneOfs '[ByteString, Int32, Int64, Double, Float] output_types) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> Tensor v'4 Bool -> Tensor v'5 ByteString -> Tensor v'6 Bool -> Tensor v'7 ByteString -> Tensor v'8 Int64 -> TensorList v'9 output_types -> m' (Tensor Value Variant)
experimentalChooseFastestDataset :: Int64 -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
experimentalChooseFastestDataset' :: OpParams -> Int64 -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
experimentalDatasetCardinality :: Tensor v'1 Variant -> Tensor Build Int64
experimentalDatasetCardinality' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int64
experimentalDatasetToTFRecord :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' ControlNode
experimentalDatasetToTFRecord' :: 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
experimentalDenseToSparseBatchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
experimentalDenseToSparseBatchDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
experimentalDirectedInterleaveDataset :: [DataType] -> Tensor v'1 Variant -> [Tensor v'2 Variant] -> Tensor Build Variant
experimentalDirectedInterleaveDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> [Tensor v'2 Variant] -> Tensor Build Variant
experimentalIgnoreErrorsDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalIgnoreErrorsDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalIteratorGetDevice :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
experimentalIteratorGetDevice' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
experimentalLMDBDataset :: forall v'1 m'. MonadBuild m' => [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
experimentalLMDBDataset' :: forall v'1 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
experimentalLatencyStatsDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
experimentalLatencyStatsDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
experimentalMatchingFilesDataset :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value Variant)
experimentalMatchingFilesDataset' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
experimentalMaxIntraOpParallelismDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalMaxIntraOpParallelismDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalNonSerializableDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalNonSerializableDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalParseExampleDataset :: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense => [DataType] -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> TensorList v'3 tdense -> Tensor Build Variant
experimentalParseExampleDataset' :: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense => OpParams -> [DataType] -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> TensorList v'3 tdense -> Tensor Build Variant
experimentalPrivateThreadPoolDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalPrivateThreadPoolDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalRandomDataset :: forall v'1 v'2 m'. MonadBuild m' => [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value Variant)
experimentalRandomDataset' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value Variant)
experimentalRebatchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalRebatchDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalSetStatsAggregatorDataset :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> m' (Tensor Value Variant)
experimentalSetStatsAggregatorDataset' :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> m' (Tensor Value Variant)
experimentalSleepDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalSleepDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalSlidingWindowDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build Variant
experimentalSlidingWindowDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build Variant
experimentalSqlDataset :: 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)
experimentalSqlDataset' :: 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)
experimentalStatsAggregatorHandle :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
experimentalStatsAggregatorHandle' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
experimentalStatsAggregatorSummary :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
experimentalStatsAggregatorSummary' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
experimentalThreadPoolDataset :: forall v'1 v'2 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' (Tensor Value Variant)
experimentalThreadPoolDataset' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' (Tensor Value Variant)
experimentalThreadPoolHandle :: forall m'. MonadBuild m' => ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
experimentalThreadPoolHandle' :: forall m'. MonadBuild m' => OpParams -> ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
experimentalUnbatchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalUnbatchDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalUniqueDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalUniqueDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
expint :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
expint' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> 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
extractGlimpseV2 :: Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build Float
extractGlimpseV2' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build Float
extractImagePatches :: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
extractImagePatches' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> ByteString -> 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
extractVolumePatches :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
extractVolumePatches' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
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
fakeParam :: forall dtype. TensorType dtype => Shape -> Tensor Build dtype
fakeParam' :: forall dtype. TensorType dtype => OpParams -> Shape -> Tensor Build dtype
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
filterByLastComponentDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
filterByLastComponentDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
fingerprint :: forall v'1 v'2 t. TensorType t => Tensor v'1 t -> Tensor v'2 ByteString -> Tensor Build Word8
fingerprint' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 ByteString -> Tensor Build Word8
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)
fixedLengthRecordDatasetV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> Tensor v'6 ByteString -> m' (Tensor Value Variant)
fixedLengthRecordDatasetV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> Tensor v'6 ByteString -> 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, Word64, 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, Word64, 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
fresnelCos :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
fresnelCos' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
fresnelSin :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
fresnelSin' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> 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)
fusedBatchNormGradV3 :: forall v'1 v'2 v'3 v'4 v'5 v'6 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 v'6 u -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u)
fusedBatchNormGradV3' :: forall v'1 v'2 v'3 v'4 v'5 v'6 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 v'6 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)
fusedBatchNormV3 :: 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, Tensor Build u)
fusedBatchNormV3' :: 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, Tensor Build u)
fusedPadConv2D :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => ByteString -> ByteString -> 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 '[Word16, Double, Float] t => OpParams -> ByteString -> ByteString -> 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 '[Word16, Double, Float] t => ByteString -> ByteString -> 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 '[Word16, Double, Float] t => OpParams -> ByteString -> ByteString -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 t -> Tensor Build t
gRUBlockCell :: forall v'1 v'2 v'3 v'4 v'5 v'6 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 v'6 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
gRUBlockCell' :: forall v'1 v'2 v'3 v'4 v'5 v'6 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 v'6 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
gRUBlockCellGrad :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 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 v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
gRUBlockCellGrad' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 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 v'6 t -> Tensor v'7 t -> Tensor v'8 t -> Tensor v'9 t -> Tensor v'10 t -> (Tensor Build t, Tensor Build t, Tensor Build 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
generateBoundingBoxProposals :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Int32 -> Tensor v'7 Float -> (Tensor Build Float, Tensor Build Float)
generateBoundingBoxProposals' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Int32 -> Tensor v'7 Float -> (Tensor Build Float, Tensor Build Float)
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
hostConst :: forall dtype. TensorType dtype => Tensor Build dtype
hostConst' :: forall dtype. TensorType dtype => OpParams -> Tensor Build dtype
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 :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT' :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => OpParams -> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT2D :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT2D' :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => OpParams -> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT3D :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT3D' :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => OpParams -> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
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
igammaGradA :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
igammaGradA' :: 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
ignoreErrorsDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
ignoreErrorsDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
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
imageProjectiveTransformV2 :: forall v'1 v'2 v'3 dtype. OneOf '[Int32, Int64, Word16, Word8, Double, Float] dtype => ByteString -> Tensor v'1 dtype -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor Build dtype
imageProjectiveTransformV2' :: forall v'1 v'2 v'3 dtype. OneOf '[Int32, Int64, Word16, Word8, Double, Float] dtype => OpParams -> ByteString -> Tensor v'1 dtype -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor Build dtype
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 => ByteString -> Shape -> Tensor Build dtype
immutableConst' :: forall dtype. TensorType dtype => OpParams -> ByteString -> 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
infeedEnqueuePrelinearizedBuffer :: forall v'1 m'. MonadBuild m' => Tensor v'1 Variant -> m' ControlNode
infeedEnqueuePrelinearizedBuffer' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 Variant -> 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
initializeTableFromDataset :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
initializeTableFromDataset' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> 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, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
inv' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, 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)
isBoostedTreesQuantileStreamResourceInitialized :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
isBoostedTreesQuantileStreamResourceInitialized' :: 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' => ByteString -> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
iterator' :: forall m'. MonadBuild m' => OpParams -> ByteString -> [DataType] -> ByteString -> 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)
iteratorFromStringHandleV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
iteratorFromStringHandleV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
iteratorGetDevice :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
iteratorGetDevice' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
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)
iteratorGetNextAsOptional :: forall v'1 m'. MonadBuild m' => [DataType] -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
iteratorGetNextAsOptional' :: forall v'1 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
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)
iteratorV2 :: forall m'. MonadBuild m' => ByteString -> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
iteratorV2' :: forall m'. MonadBuild m' => OpParams -> ByteString -> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
kMC2ChainInitialization :: Tensor v'1 Float -> Tensor v'2 Int64 -> Tensor Build Int64
kMC2ChainInitialization' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Int64 -> Tensor Build Int64
kmeansPlusPlusInitialization :: Tensor v'1 Float -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build Float
kmeansPlusPlusInitialization' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build Float
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
lMDBDataset :: forall v'1 m'. MonadBuild m' => [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
lMDBDataset' :: forall v'1 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
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
lSTMBlockCell :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t. OneOf '[Word16, 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 Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
lSTMBlockCell' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t. OneOf '[Word16, 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 Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
lSTMBlockCellGrad :: 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 v'16 t. OneOf '[Word16, Float] t => Bool -> 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 t -> Tensor v'13 t -> Tensor v'14 t -> Tensor v'15 t -> Tensor v'16 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
lSTMBlockCellGrad' :: 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 v'16 t. OneOf '[Word16, Float] t => OpParams -> Bool -> 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 t -> Tensor v'13 t -> Tensor v'14 t -> Tensor v'15 t -> Tensor v'16 t -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build 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
leakyRelu :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
leakyRelu' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
leakyReluGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
leakyReluGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
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)
loadTPUEmbeddingADAMParameters :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingADAMParameters' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingADAMParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingADAMParametersGradAccumDebug' :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingAdadeltaParameters :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingAdadeltaParameters' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingAdadeltaParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingAdadeltaParametersGradAccumDebug' :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingAdagradParameters :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
loadTPUEmbeddingAdagradParameters' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
loadTPUEmbeddingAdagradParametersGradAccumDebug :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingAdagradParametersGradAccumDebug' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingCenteredRMSPropParameters :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingCenteredRMSPropParameters' :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingFTRLParameters :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingFTRLParameters' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingFTRLParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingFTRLParametersGradAccumDebug' :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingMDLAdagradLightParameters :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingMDLAdagradLightParameters' :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingMomentumParameters :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
loadTPUEmbeddingMomentumParameters' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
loadTPUEmbeddingMomentumParametersGradAccumDebug :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingMomentumParametersGradAccumDebug' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingProximalAdagradParameters :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
loadTPUEmbeddingProximalAdagradParameters' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
loadTPUEmbeddingProximalAdagradParametersGradAccumDebug :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingProximalAdagradParametersGradAccumDebug' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingProximalYogiParameters :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingProximalYogiParameters' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingProximalYogiParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingProximalYogiParametersGradAccumDebug' :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingRMSPropParameters :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingRMSPropParameters' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
loadTPUEmbeddingRMSPropParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingRMSPropParametersGradAccumDebug' :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
loadTPUEmbeddingStochasticGradientDescentParameters :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> m' ControlNode
loadTPUEmbeddingStochasticGradientDescentParameters' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> m' ControlNode
loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
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, Word16, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t)
logMatrixDeterminant' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, 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
lookupTableRemoveV2 :: forall v'1 v'2 tin m'. (MonadBuild m', TensorType tin) => Tensor v'1 ResourceHandle -> Tensor v'2 tin -> m' ControlNode
lookupTableRemoveV2' :: forall v'1 v'2 tin m'. (MonadBuild m', TensorType tin) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tin -> 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
lowerBound :: forall v'1 v'2 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
lowerBound' :: forall v'1 v'2 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
lu :: forall v'1 t output_idx_type. (OneOf '[Complex Double, Complex Float, Word16, Double, Float] t, OneOf '[Int32, Int64] output_idx_type) => Tensor v'1 t -> (Tensor Build t, Tensor Build output_idx_type)
lu' :: forall v'1 t output_idx_type. (OneOf '[Complex Double, Complex Float, Word16, Double, Float] t, OneOf '[Int32, Int64] output_idx_type) => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build output_idx_type)
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, Int64, 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, Int64, 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
matchingFilesDataset :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value Variant)
matchingFilesDataset' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
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, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
matrixDeterminant' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, 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
matrixDiagPartV2 :: 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
matrixDiagPartV2' :: 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
matrixDiagPartV3 :: 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
matrixDiagPartV3' :: 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
matrixDiagV2 :: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 t -> Tensor Build t
matrixDiagV2' :: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 t -> Tensor Build t
matrixDiagV3 :: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 t -> Tensor Build t
matrixDiagV3' :: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 t -> Tensor Build t
matrixExponential :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
matrixExponential' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
matrixInverse :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
matrixInverse' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, 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
matrixSetDiagV2 :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build t
matrixSetDiagV2' :: forall v'1 v'2 v'3 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build t
matrixSetDiagV3 :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build t
matrixSetDiagV3' :: forall v'1 v'2 v'3 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build t
matrixSolve :: 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
matrixSolve' :: 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
matrixSolveLs :: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Word16, 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, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Double -> Tensor Build t
matrixSquareRoot :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
matrixSquareRoot' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
matrixTriangularSolve :: 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
matrixTriangularSolve' :: 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
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
maxIntraOpParallelismDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
maxIntraOpParallelismDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
maxPool :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
maxPool' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
maxPool3D :: forall v'1 t. OneOf '[Word16, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
maxPool3D' :: forall v'1 t. OneOf '[Word16, Float] t => OpParams -> ByteString -> 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) => ByteString -> 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 -> ByteString -> 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 '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> 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 '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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) => ByteString -> 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 -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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) => ByteString -> 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 -> ByteString -> 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 => ByteString -> 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 -> ByteString -> 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) => ByteString -> 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 -> ByteString -> Tensor v'1 t -> (Tensor Build t, Tensor Build targmax)
maximum :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
maximum' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Word16, Word8, 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 '[Int16, Int32, Int64, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
minimum' :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Word16, Word8, 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) => ByteString -> 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 -> ByteString -> 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) => ByteString -> 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 -> ByteString -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
mlirPassthroughOp :: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs) => ByteString -> TensorList v'1 tinputs -> TensorList Build toutputs
mlirPassthroughOp' :: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs) => OpParams -> ByteString -> TensorList v'1 tinputs -> TensorList Build toutputs
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
modelDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
modelDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
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
mulNoNan :: 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
mulNoNan' :: 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
multiDeviceIterator :: forall m'. MonadBuild m' => ByteString -> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
multiDeviceIterator' :: forall m'. MonadBuild m' => OpParams -> ByteString -> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
multiDeviceIteratorFromStringHandle :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
multiDeviceIteratorFromStringHandle' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
multiDeviceIteratorGetNextFromShard :: forall v'1 v'2 v'3 output_types m'. (MonadBuild m', TensorTypes output_types) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 Int64 -> m' (TensorList Value output_types)
multiDeviceIteratorGetNextFromShard' :: forall v'1 v'2 v'3 output_types m'. (MonadBuild m', TensorTypes output_types) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 Int64 -> m' (TensorList Value output_types)
multiDeviceIteratorInit :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> Tensor v'3 Int64 -> m' (Tensor Value Int64)
multiDeviceIteratorInit' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> Tensor v'3 Int64 -> m' (Tensor Value Int64)
multiDeviceIteratorToStringHandle :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
multiDeviceIteratorToStringHandle' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
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 v'2 key_dtype m'. (MonadBuild m', TensorType key_dtype) => DataType -> Tensor v'1 key_dtype -> Tensor v'2 key_dtype -> m' (Tensor Value ResourceHandle)
mutableDenseHashTableV2' :: forall v'1 v'2 key_dtype m'. (MonadBuild m', TensorType key_dtype) => OpParams -> DataType -> Tensor v'1 key_dtype -> Tensor v'2 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)
ncclAllReduce :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> ByteString -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
ncclAllReduce' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> ByteString -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
ncclBroadcast :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Shape -> Tensor v'1 t -> m' (Tensor Value t)
ncclBroadcast' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
ncclReduce :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => ByteString -> [Tensor v'1 t] -> m' (Tensor Value t)
ncclReduce' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> ByteString -> [Tensor v'1 t] -> m' (Tensor Value t)
ndtri :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
ndtri' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
nearestNeighbors :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int64 -> (Tensor Build Int64, Tensor Build Float)
nearestNeighbors' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int64 -> (Tensor Build Int64, Tensor Build Float)
neg :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
neg' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, 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
nextAfter :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
nextAfter' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
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
nonDeterministicInts :: forall v'1 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 shape_dtype -> m' (Tensor Value dtype)
nonDeterministicInts' :: forall v'1 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => OpParams -> Tensor v'1 shape_dtype -> m' (Tensor Value dtype)
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 :: forall v'1 v'2 v'3 v'4 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor v'4 t_threshold -> Tensor Build Int32
nonMaxSuppressionV2' :: forall v'1 v'2 v'3 v'4 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor v'4 t_threshold -> Tensor Build Int32
nonMaxSuppressionV3 :: forall v'1 v'2 v'3 v'4 v'5 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor v'4 t_threshold -> Tensor v'5 t_threshold -> Tensor Build Int32
nonMaxSuppressionV3' :: forall v'1 v'2 v'3 v'4 v'5 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor v'4 t_threshold -> Tensor v'5 t_threshold -> Tensor Build Int32
nonMaxSuppressionV4 :: forall v'1 v'2 v'3 v'4 v'5 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor v'4 t_threshold -> Tensor v'5 t_threshold -> (Tensor Build Int32, Tensor Build Int32)
nonMaxSuppressionV4' :: forall v'1 v'2 v'3 v'4 v'5 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor v'4 t_threshold -> Tensor v'5 t_threshold -> (Tensor Build Int32, Tensor Build Int32)
nonMaxSuppressionV5 :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> (Tensor Build Int32, Tensor Build t, Tensor Build Int32)
nonMaxSuppressionV5' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor v'4 t -> Tensor v'5 t -> Tensor v'6 t -> (Tensor Build Int32, Tensor Build t, Tensor Build Int32)
nonMaxSuppressionWithOverlaps :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor Build Int32
nonMaxSuppressionWithOverlaps' :: OpParams -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor Build Int32
nonSerializableDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
nonSerializableDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
notEqual :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word32, Word64, 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, Word32, Word64, 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
optimizeDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
optimizeDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
optionalFromValue :: forall v'1 toutput_types. TensorTypes toutput_types => TensorList v'1 toutput_types -> Tensor Build Variant
optionalFromValue' :: forall v'1 toutput_types. TensorTypes toutput_types => OpParams -> TensorList v'1 toutput_types -> Tensor Build Variant
optionalGetValue :: forall v'1 output_types. TensorTypes output_types => Tensor v'1 Variant -> TensorList Build output_types
optionalGetValue' :: forall v'1 output_types. TensorTypes output_types => OpParams -> Tensor v'1 Variant -> TensorList Build output_types
optionalHasValue :: Tensor v'1 Variant -> Tensor Build Bool
optionalHasValue' :: OpParams -> Tensor v'1 Variant -> Tensor Build Bool
optionalNone :: Tensor Build Variant
optionalNone' :: OpParams -> Tensor Build Variant
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
paddedBatchDatasetV2 :: forall v'1 v'2 v'3 v'4 v'5 toutput_types. TensorTypes toutput_types => Tensor v'1 Variant -> Tensor v'2 Int64 -> [Tensor v'3 Int64] -> TensorList v'4 toutput_types -> Tensor v'5 Bool -> Tensor Build Variant
paddedBatchDatasetV2' :: forall v'1 v'2 v'3 v'4 v'5 toutput_types. TensorTypes toutput_types => OpParams -> Tensor v'1 Variant -> Tensor v'2 Int64 -> [Tensor v'3 Int64] -> TensorList v'4 toutput_types -> Tensor v'5 Bool -> 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)
parseExampleDataset :: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense => [DataType] -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> TensorList v'3 tdense -> Tensor Build Variant
parseExampleDataset' :: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense => OpParams -> [DataType] -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> TensorList v'3 tdense -> Tensor Build Variant
parseExampleDatasetV2 :: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense => [DataType] -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> TensorList v'3 tdense -> Tensor Build Variant
parseExampleDatasetV2' :: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense => OpParams -> [DataType] -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> TensorList v'3 tdense -> Tensor Build Variant
parseExampleV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 tdense sparse_types ragged_value_types ragged_split_types. (OneOfs '[ByteString, Int64, Float] tdense, OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] ragged_value_types, OneOfs '[Int32, Int64] ragged_split_types) => Int64 -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> Tensor v'5 ByteString -> TensorList v'6 tdense -> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense, TensorList Build ragged_value_types, TensorList Build ragged_split_types)
parseExampleV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 tdense sparse_types ragged_value_types ragged_split_types. (OneOfs '[ByteString, Int64, Float] tdense, OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] ragged_value_types, OneOfs '[Int32, Int64] ragged_split_types) => OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> Tensor v'5 ByteString -> TensorList v'6 tdense -> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense, TensorList Build ragged_value_types, TensorList Build ragged_split_types)
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)
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
prelinearize :: forall v'1 dtype. TensorType dtype => Tensor v'1 dtype -> Tensor Build Variant
prelinearize' :: forall v'1 dtype. TensorType dtype => OpParams -> Tensor v'1 dtype -> Tensor Build Variant
prelinearizeTuple :: forall v'1 dtypes. TensorTypes dtypes => TensorList v'1 dtypes -> Tensor Build Variant
prelinearizeTuple' :: forall v'1 dtypes. TensorTypes dtypes => OpParams -> TensorList v'1 dtypes -> 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)
printV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' ControlNode
printV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' ControlNode
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)
privateThreadPoolDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
privateThreadPoolDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
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
pyFunc :: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout) => ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
pyFunc' :: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout) => OpParams -> ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
pyFuncStateless :: forall v'1 tin tout. (TensorTypes tin, TensorTypes tout) => ByteString -> TensorList v'1 tin -> TensorList Build tout
pyFuncStateless' :: forall v'1 tin tout. (TensorTypes tin, TensorTypes tout) => OpParams -> ByteString -> TensorList v'1 tin -> TensorList Build tout
qr :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t)
qr' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, 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 => ByteString -> 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 -> ByteString -> 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) => ByteString -> 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 -> ByteString -> 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)
quantizedConv2DAndRelu :: 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) => ByteString -> 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)
quantizedConv2DAndRelu' :: 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 -> ByteString -> 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)
quantizedConv2DAndReluAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> 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 v'7 Float -> Tensor v'8 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndReluAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> 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 v'7 Float -> Tensor v'8 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> 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 v'7 Float -> Tensor v'8 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> 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 v'7 Float -> Tensor v'8 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DPerChannel :: 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) => ByteString -> 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)
quantizedConv2DPerChannel' :: 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 -> ByteString -> 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)
quantizedConv2DWithBias :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> 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 v'7 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBias' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> 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 v'7 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndRelu :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> 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 v'7 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndRelu' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> 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 v'7 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndReluAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndReluAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSignedSumAndReluAndRequantize :: 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 tinput tfilter tbias tsummand out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] tsummand, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> Tensor v'10 tsummand -> Tensor v'11 Float -> Tensor v'12 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSignedSumAndReluAndRequantize' :: 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 tinput tfilter tbias tsummand out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] tsummand, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> Tensor v'10 tsummand -> Tensor v'11 Float -> Tensor v'12 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSumAndRelu :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> 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 v'7 Float -> Tensor v'8 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSumAndRelu' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> 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 v'7 Float -> Tensor v'8 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSumAndReluAndRequantize :: 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 tinput tfilter tbias tsummand out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] tsummand, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> Tensor v'10 tsummand -> Tensor v'11 Float -> Tensor v'12 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSumAndReluAndRequantize' :: 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 tinput tfilter tbias tsummand out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] tsummand, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> Tensor v'10 tsummand -> Tensor v'11 Float -> Tensor v'12 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2D :: 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) => ByteString -> 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)
quantizedDepthwiseConv2D' :: 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 -> ByteString -> 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)
quantizedDepthwiseConv2DWithBias :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> 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 v'7 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBias' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> 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 v'7 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBiasAndRelu :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> 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 v'7 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBiasAndRelu' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> 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 v'7 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBiasAndReluAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type) => ByteString -> Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBiasAndReluAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> ByteString -> Tensor v'1 tinput -> Tensor v'2 tfilter -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 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)
quantizedMatMulWithBias :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput) => Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBias' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput) => OpParams -> Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndDequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Float] toutput) => Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> Tensor Build toutput
quantizedMatMulWithBiasAndDequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Float] toutput) => OpParams -> Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> Tensor Build toutput
quantizedMatMulWithBiasAndRelu :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 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 v'7 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndRelu' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 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 v'7 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndReluAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput) => Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndReluAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput) => OpParams -> Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput) => Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 Float -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput) => OpParams -> Tensor v'1 t1 -> Tensor v'2 t2 -> Tensor v'3 tbias -> Tensor v'4 Float -> Tensor v'5 Float -> Tensor v'6 Float -> Tensor v'7 Float -> Tensor v'8 Float -> Tensor v'9 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 => ByteString -> 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 -> ByteString -> 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 :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT' :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => OpParams -> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT2D :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT2D' :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => OpParams -> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT3D :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT3D' :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => OpParams -> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
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
raggedBincount :: forall v'1 v'2 v'3 v'4 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t) => Tensor v'1 Int64 -> Tensor v'2 tidx -> Tensor v'3 tidx -> Tensor v'4 t -> Tensor Build t
raggedBincount' :: forall v'1 v'2 v'3 v'4 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 tidx -> Tensor v'3 tidx -> Tensor v'4 t -> Tensor Build t
raggedCountSparseOutput :: forall v'1 v'2 v'3 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type) => Bool -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 output_type -> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)
raggedCountSparseOutput' :: forall v'1 v'2 v'3 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type) => OpParams -> Bool -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 output_type -> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)
raggedCross :: forall v'1 v'2 v'3 v'4 v'5 v'6 ragged_values_types ragged_splits_types sparse_values_types dense_types out_values_type out_row_splits_type. (OneOfs '[ByteString, Int64] ragged_values_types, OneOfs '[Int32, Int64] ragged_splits_types, OneOfs '[ByteString, Int64] sparse_values_types, OneOfs '[ByteString, Int64] dense_types, OneOf '[ByteString, Int64] out_values_type, OneOf '[Int32, Int64] out_row_splits_type) => Int64 -> Bool -> ByteString -> Int64 -> TensorList v'1 ragged_values_types -> TensorList v'2 ragged_splits_types -> [Tensor v'3 Int64] -> TensorList v'4 sparse_values_types -> [Tensor v'5 Int64] -> TensorList v'6 dense_types -> (Tensor Build out_values_type, Tensor Build out_row_splits_type)
raggedCross' :: forall v'1 v'2 v'3 v'4 v'5 v'6 ragged_values_types ragged_splits_types sparse_values_types dense_types out_values_type out_row_splits_type. (OneOfs '[ByteString, Int64] ragged_values_types, OneOfs '[Int32, Int64] ragged_splits_types, OneOfs '[ByteString, Int64] sparse_values_types, OneOfs '[ByteString, Int64] dense_types, OneOf '[ByteString, Int64] out_values_type, OneOf '[Int32, Int64] out_row_splits_type) => OpParams -> Int64 -> Bool -> ByteString -> Int64 -> TensorList v'1 ragged_values_types -> TensorList v'2 ragged_splits_types -> [Tensor v'3 Int64] -> TensorList v'4 sparse_values_types -> [Tensor v'5 Int64] -> TensorList v'6 dense_types -> (Tensor Build out_values_type, Tensor Build out_row_splits_type)
raggedGather :: forall v'1 v'2 v'3 tvalues tindices tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tsplits) => Int64 -> [Tensor v'1 tsplits] -> Tensor v'2 tvalues -> Tensor v'3 tindices -> ([Tensor Build tsplits], Tensor Build tvalues)
raggedGather' :: forall v'1 v'2 v'3 tvalues tindices tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tsplits) => OpParams -> Int64 -> [Tensor v'1 tsplits] -> Tensor v'2 tvalues -> Tensor v'3 tindices -> ([Tensor Build tsplits], Tensor Build tvalues)
raggedRange :: forall v'1 v'2 v'3 t tsplits. (OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tsplits) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> (Tensor Build tsplits, Tensor Build t)
raggedRange' :: forall v'1 v'2 v'3 t tsplits. (OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tsplits) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> (Tensor Build tsplits, Tensor Build t)
raggedTensorFromVariant :: forall v'1 tvalues tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tsplits) => Int64 -> Int64 -> Tensor v'1 Variant -> ([Tensor Build tsplits], Tensor Build tvalues)
raggedTensorFromVariant' :: forall v'1 tvalues tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tsplits) => OpParams -> Int64 -> Int64 -> Tensor v'1 Variant -> ([Tensor Build tsplits], Tensor Build tvalues)
raggedTensorToSparse :: forall v'1 v'2 t tsplits. (TensorType t, OneOf '[Int32, Int64] tsplits) => [Tensor v'1 tsplits] -> Tensor v'2 t -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
raggedTensorToSparse' :: forall v'1 v'2 t tsplits. (TensorType t, OneOf '[Int32, Int64] tsplits) => OpParams -> [Tensor v'1 tsplits] -> Tensor v'2 t -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
raggedTensorToTensor :: forall v'1 v'2 v'3 v'4 t tindex tshape. (TensorType t, OneOf '[Int32, Int64] tindex, OneOf '[Int32, Int64] tshape) => Tensor v'1 tshape -> Tensor v'2 t -> Tensor v'3 t -> [Tensor v'4 tindex] -> Tensor Build t
raggedTensorToTensor' :: forall v'1 v'2 v'3 v'4 t tindex tshape. (TensorType t, OneOf '[Int32, Int64] tindex, OneOf '[Int32, Int64] tshape) => OpParams -> Tensor v'1 tshape -> Tensor v'2 t -> Tensor v'3 t -> [Tensor v'4 tindex] -> Tensor Build t
raggedTensorToVariant :: forall v'1 v'2 tvalues tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tsplits) => Bool -> [Tensor v'1 tsplits] -> Tensor v'2 tvalues -> Tensor Build Variant
raggedTensorToVariant' :: forall v'1 v'2 tvalues tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tsplits) => OpParams -> Bool -> [Tensor v'1 tsplits] -> Tensor v'2 tvalues -> Tensor Build Variant
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)
randomGammaGrad :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
randomGammaGrad' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build 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
rebatchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
rebatchDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
reciprocal :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
reciprocal' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, 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' => ByteString -> m' (Tensor Value ByteString)
recordInput' :: forall m'. MonadBuild m' => OpParams -> ByteString -> m' (Tensor Value ByteString)
recv :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => ByteString -> ByteString -> Int64 -> ByteString -> m' (Tensor Value tensor_type)
recv' :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> m' (Tensor Value tensor_type)
recvTPUEmbeddingActivations :: forall m'. MonadBuild m' => ByteString -> Int64 -> m' [Tensor Value Float]
recvTPUEmbeddingActivations' :: forall m'. MonadBuild m' => OpParams -> ByteString -> Int64 -> m' [Tensor Value Float]
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) => ByteString -> Tensor Ref t -> m' (Tensor Ref t)
refEnter' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> 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
registerDataset :: Int64 -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> Tensor Build Int64
registerDataset' :: OpParams -> Int64 -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> Tensor Build Int64
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) => ByteString -> TensorList v'1 tinputs -> TensorList Build toutputs
remoteFusedGraphExecute' :: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs) => OpParams -> ByteString -> 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)
requantizationRangePerChannel :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => Float -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build Float, Tensor Build Float)
requantizationRangePerChannel' :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => OpParams -> Float -> Tensor v'1 t -> 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)
requantizePerChannel :: forall v'1 v'2 v'3 v'4 v'5 t out_type. (OneOf '[Int16, Int32, Word16, Word8] t, OneOf '[Int16, Int32, Word16, Word8] out_type) => Tensor v'1 t -> 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)
requantizePerChannel' :: forall v'1 v'2 v'3 v'4 v'5 t out_type. (OneOf '[Int16, Int32, Word16, Word8] t, OneOf '[Int16, Int32, Word16, Word8] out_type) => OpParams -> Tensor v'1 t -> 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
resourceAccumulatorApplyGradient :: forall v'1 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 v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 dtype -> m' ControlNode
resourceAccumulatorApplyGradient' :: forall v'1 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 v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 dtype -> m' ControlNode
resourceAccumulatorNumAccumulated :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
resourceAccumulatorNumAccumulated' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
resourceAccumulatorSetGlobalStep :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
resourceAccumulatorSetGlobalStep' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
resourceAccumulatorTakeGradient :: forall v'1 v'2 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> m' (Tensor Value dtype)
resourceAccumulatorTakeGradient' :: forall v'1 v'2 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> m' (Tensor Value dtype)
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
resourceApplyAdagradV2 :: 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
resourceApplyAdagradV2' :: 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
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
resourceApplyAdamWithAmsgrad :: 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 '[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 -> Tensor v'10 t -> Tensor v'11 t -> m' ControlNode
resourceApplyAdamWithAmsgrad' :: 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 '[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 -> Tensor v'10 t -> Tensor v'11 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
resourceApplyKerasMomentum :: 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
resourceApplyKerasMomentum' :: 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
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
resourceConditionalAccumulator :: forall m'. MonadBuild m' => DataType -> Shape -> m' (Tensor Value ResourceHandle)
resourceConditionalAccumulator' :: forall m'. MonadBuild m' => OpParams -> DataType -> Shape -> m' (Tensor Value ResourceHandle)
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)
resourceGatherNd :: 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)
resourceGatherNd' :: 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
resourceScatterNdAdd :: 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
resourceScatterNdAdd' :: 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
resourceScatterNdMax :: 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
resourceScatterNdMax' :: 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
resourceScatterNdMin :: 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
resourceScatterNdMin' :: 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
resourceScatterNdSub :: 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
resourceScatterNdSub' :: 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
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
resourceSparseApplyAdagradV2 :: 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 t -> Tensor v'6 tindices -> m' ControlNode
resourceSparseApplyAdagradV2' :: 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 t -> Tensor v'6 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) => 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
resourceSparseApplyKerasMomentum :: 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
resourceSparseApplyKerasMomentum' :: 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
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)
retrieveTPUEmbeddingADAMParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingADAMParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingADAMParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingADAMParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdadeltaParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdadeltaParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdagradParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdagradParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdagradParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdagradParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingCenteredRMSPropParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingCenteredRMSPropParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingFTRLParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingFTRLParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingFTRLParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingFTRLParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMDLAdagradLightParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMDLAdagradLightParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMomentumParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMomentumParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMomentumParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMomentumParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalAdagradParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalAdagradParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalYogiParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalYogiParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingRMSPropParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingRMSPropParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingRMSPropParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingRMSPropParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingStochasticGradientDescentParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float)
retrieveTPUEmbeddingStochasticGradientDescentParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float)
retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
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
rngSkip :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> m' ControlNode
rngSkip' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> m' ControlNode
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, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
round' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, 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)
samplingDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Float -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build Variant
samplingDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Float -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build Variant
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
scaleAndTranslate :: 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 Int32 -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor Build Float
scaleAndTranslate' :: 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 Int32 -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor Build Float
scaleAndTranslateGrad :: forall v'1 v'2 v'3 v'4 t. OneOf '[Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor Build t
scaleAndTranslateGrad' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor Build 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) => 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)
scatterNdMax :: 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)
scatterNdMax' :: 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)
scatterNdMin :: 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)
scatterNdMin' :: 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, Bool, 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, Bool, 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 -> ByteString -> 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 -> ByteString -> 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])
sdcaOptimizerV2 :: Float -> Float -> ByteString -> 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])
sdcaOptimizerV2' :: OpParams -> Float -> Float -> ByteString -> 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
selectV2 :: 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
selectV2' :: 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 '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
selfAdjointEig' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
selfAdjointEigV2 :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t)
selfAdjointEigV2' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, 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
send :: forall v'1 t m'. (MonadBuild m', TensorType t) => ByteString -> ByteString -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
send' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
sendTPUEmbeddingGradients :: forall v'1 v'2 m'. MonadBuild m' => ByteString -> [Tensor v'1 Float] -> [Tensor v'2 Float] -> m' ControlNode
sendTPUEmbeddingGradients' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> ByteString -> [Tensor v'1 Float] -> [Tensor v'2 Float] -> m' ControlNode
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 v'3 v'4 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> m' (Tensor Value Variant)
setStatsAggregatorDataset' :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> 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]
shardDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
shardDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build Variant
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
shuffleAndRepeatDatasetV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> Tensor v'6 ResourceHandle -> m' (Tensor Value Variant)
shuffleAndRepeatDatasetV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> Tensor v'6 ResourceHandle -> m' (Tensor Value 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
shuffleDatasetV2 :: forall v'1 v'2 v'3 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 ResourceHandle -> m' (Tensor Value Variant)
shuffleDatasetV2' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 ResourceHandle -> m' (Tensor Value Variant)
shuffleDatasetV3 :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 ResourceHandle -> m' (Tensor Value Variant)
shuffleDatasetV3' :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 ResourceHandle -> m' (Tensor Value 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 -> ByteString -> 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 -> ByteString -> m' (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64, Tensor Value Int32, Tensor Value Int64, Tensor Value Int32, Tensor Value Int32)
sleepDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
sleepDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
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
slidingWindowDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build Variant
slidingWindowDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 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
snapshotDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
snapshotDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor Build Variant
sobolSample :: forall v'1 v'2 v'3 dtype. OneOf '[Double, Float] dtype => Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build dtype
sobolSample' :: forall v'1 v'2 v'3 dtype. OneOf '[Double, Float] dtype => OpParams -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build dtype
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 '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
softplus' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
softplusGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
softplusGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
softsign :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
softsign' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
softsignGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
softsignGrad' :: forall v'1 v'2 t. OneOf '[Word16, 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)
sparseApplyAdagradV2 :: 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 t -> Tensor v'6 tindices -> m' (Tensor Ref t)
sparseApplyAdagradV2' :: 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 t -> Tensor v'6 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) => 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)
sparseBincount :: forall v'1 v'2 v'3 v'4 v'5 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t) => Tensor v'1 Int64 -> Tensor v'2 tidx -> Tensor v'3 Int64 -> Tensor v'4 tidx -> Tensor v'5 t -> Tensor Build t
sparseBincount' :: forall v'1 v'2 v'3 v'4 v'5 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 tidx -> Tensor v'3 Int64 -> Tensor v'4 tidx -> Tensor v'5 t -> Tensor Build 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)
sparseCountSparseOutput :: forall v'1 v'2 v'3 v'4 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type) => Bool -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 output_type -> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)
sparseCountSparseOutput' :: forall v'1 v'2 v'3 v'4 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type) => OpParams -> Bool -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 output_type -> (Tensor Build Int64, Tensor Build output_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) => 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)
sparseCrossHashed :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 sparse_types dense_types. (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types) => [Tensor v'1 Int64] -> TensorList v'2 sparse_types -> [Tensor v'3 Int64] -> TensorList v'4 dense_types -> Tensor v'5 Int64 -> Tensor v'6 Bool -> Tensor v'7 Int64 -> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64)
sparseCrossHashed' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 sparse_types dense_types. (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types) => OpParams -> [Tensor v'1 Int64] -> TensorList v'2 sparse_types -> [Tensor v'3 Int64] -> TensorList v'4 dense_types -> Tensor v'5 Int64 -> Tensor v'6 Bool -> Tensor v'7 Int64 -> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64)
sparseCrossV2 :: forall v'1 v'2 v'3 v'4 v'5 sparse_types dense_types. (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types) => [Tensor v'1 Int64] -> TensorList v'2 sparse_types -> [Tensor v'3 Int64] -> TensorList v'4 dense_types -> Tensor v'5 ByteString -> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64)
sparseCrossV2' :: forall v'1 v'2 v'3 v'4 v'5 sparse_types dense_types. (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types) => OpParams -> [Tensor v'1 Int64] -> TensorList v'2 sparse_types -> [Tensor v'3 Int64] -> TensorList v'4 dense_types -> Tensor v'5 ByteString -> (Tensor Build Int64, Tensor Build ByteString, 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
sparseMatrixAdd :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build Variant
sparseMatrixAdd' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build Variant
sparseMatrixMatMul :: forall v'1 v'2 t. TensorType t => Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build t
sparseMatrixMatMul' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build t
sparseMatrixMul :: forall v'1 v'2 t. TensorType t => Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build Variant
sparseMatrixMul' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build Variant
sparseMatrixNNZ :: Tensor v'1 Variant -> Tensor Build Int32
sparseMatrixNNZ' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int32
sparseMatrixOrderingAMD :: Tensor v'1 Variant -> Tensor Build Int32
sparseMatrixOrderingAMD' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int32
sparseMatrixSoftmax :: DataType -> Tensor v'1 Variant -> Tensor Build Variant
sparseMatrixSoftmax' :: OpParams -> DataType -> Tensor v'1 Variant -> Tensor Build Variant
sparseMatrixSoftmaxGrad :: DataType -> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
sparseMatrixSoftmaxGrad' :: OpParams -> DataType -> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
sparseMatrixSparseCholesky :: DataType -> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build Variant
sparseMatrixSparseCholesky' :: OpParams -> DataType -> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build Variant
sparseMatrixSparseMatMul :: DataType -> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
sparseMatrixSparseMatMul' :: OpParams -> DataType -> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
sparseMatrixTranspose :: DataType -> Tensor v'1 Variant -> Tensor Build Variant
sparseMatrixTranspose' :: OpParams -> DataType -> Tensor v'1 Variant -> Tensor Build Variant
sparseMatrixZeros :: DataType -> Tensor v'1 Int64 -> Tensor Build Variant
sparseMatrixZeros' :: OpParams -> DataType -> Tensor v'1 Int64 -> Tensor Build Variant
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 tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
sparseSegmentMean' :: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
sparseSegmentMeanGrad :: forall v'1 v'2 v'3 v'4 t tidx tsegmentids. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 Int32 -> Tensor Build t
sparseSegmentMeanGrad' :: forall v'1 v'2 v'3 v'4 t tidx tsegmentids. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 Int32 -> Tensor Build t
sparseSegmentMeanWithNumSegments :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 tnumsegments -> Tensor Build t
sparseSegmentMeanWithNumSegments' :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 tnumsegments -> Tensor Build t
sparseSegmentSqrtN :: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
sparseSegmentSqrtN' :: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
sparseSegmentSqrtNGrad :: forall v'1 v'2 v'3 v'4 t tidx tsegmentids. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 Int32 -> Tensor Build t
sparseSegmentSqrtNGrad' :: forall v'1 v'2 v'3 v'4 t tidx tsegmentids. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 Int32 -> Tensor Build t
sparseSegmentSqrtNWithNumSegments :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 tnumsegments -> Tensor Build t
sparseSegmentSqrtNWithNumSegments' :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 tnumsegments -> Tensor Build t
sparseSegmentSum :: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
sparseSegmentSum' :: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
sparseSegmentSumWithNumSegments :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 tnumsegments -> Tensor Build t
sparseSegmentSumWithNumSegments' :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids) => OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> 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)
sparseSliceGrad :: 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
sparseSliceGrad' :: 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
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)
sparseTensorToCSRSparseMatrix :: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build Variant
sparseTensorToCSRSparseMatrix' :: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build 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 => ByteString -> 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 -> ByteString -> 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)
spence :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
spence' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
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, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
square' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, 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)
statefulRandomBinomial :: forall v'1 v'2 v'3 v'4 v'5 s t dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64, Word16, Double, Float] dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 s -> Tensor v'4 t -> Tensor v'5 t -> m' (Tensor Value dtype)
statefulRandomBinomial' :: forall v'1 v'2 v'3 v'4 v'5 s t dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64, Word16, Double, Float] dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 s -> Tensor v'4 t -> Tensor v'5 t -> m' (Tensor Value dtype)
statefulStandardNormal :: forall v'1 v'2 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 shape_dtype -> m' (Tensor Value dtype)
statefulStandardNormal' :: forall v'1 v'2 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 shape_dtype -> m' (Tensor Value dtype)
statefulStandardNormalV2 :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
statefulStandardNormalV2' :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
statefulTruncatedNormal :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
statefulTruncatedNormal' :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
statefulUniform :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
statefulUniform' :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
statefulUniformFullInt :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
statefulUniformFullInt' :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
statefulUniformInt :: forall v'1 v'2 v'3 v'4 v'5 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> Tensor v'4 dtype -> Tensor v'5 dtype -> m' (Tensor Value dtype)
statefulUniformInt' :: forall v'1 v'2 v'3 v'4 v'5 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> Tensor v'4 dtype -> Tensor v'5 dtype -> m' (Tensor Value 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) => 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
statelessParameterizedTruncatedNormal :: forall v'1 v'2 v'3 v'4 v'5 v'6 s tseed dtype. (OneOf '[Int32, Int64] s, OneOf '[Int32, Int64] tseed, OneOf '[Word16, Double, Float] dtype) => Tensor v'1 s -> Tensor v'2 tseed -> Tensor v'3 dtype -> Tensor v'4 dtype -> Tensor v'5 dtype -> Tensor v'6 dtype -> Tensor Build dtype
statelessParameterizedTruncatedNormal' :: forall v'1 v'2 v'3 v'4 v'5 v'6 s tseed dtype. (OneOf '[Int32, Int64] s, OneOf '[Int32, Int64] tseed, OneOf '[Word16, Double, Float] dtype) => OpParams -> Tensor v'1 s -> Tensor v'2 tseed -> Tensor v'3 dtype -> Tensor v'4 dtype -> Tensor v'5 dtype -> Tensor v'6 dtype -> Tensor Build dtype
statelessRandomBinomial :: forall v'1 v'2 v'3 v'4 s tseed t dtype. (OneOf '[Int32, Int64] s, OneOf '[Int32, Int64] tseed, OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64, Word16, Double, Float] dtype) => Tensor v'1 s -> Tensor v'2 tseed -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build dtype
statelessRandomBinomial' :: forall v'1 v'2 v'3 v'4 s tseed t dtype. (OneOf '[Int32, Int64] s, OneOf '[Int32, Int64] tseed, OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64, Word16, Double, Float] dtype) => OpParams -> Tensor v'1 s -> Tensor v'2 tseed -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build dtype
statelessRandomGammaV2 :: forall v'1 v'2 v'3 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 v'3 dtype -> Tensor Build dtype
statelessRandomGammaV2' :: forall v'1 v'2 v'3 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 v'3 dtype -> 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) => 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
statelessRandomPoisson :: forall v'1 v'2 v'3 rtype dtype t tseed. (OneOf '[Int32, Int64, Word16, Double, Float] rtype, OneOf '[Int32, Int64, Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor v'3 rtype -> Tensor Build dtype
statelessRandomPoisson' :: forall v'1 v'2 v'3 rtype dtype t tseed. (OneOf '[Int32, Int64, Word16, Double, Float] rtype, OneOf '[Int32, Int64, Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor v'3 rtype -> 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
statelessRandomUniformFullInt :: forall v'1 v'2 dtype t tseed. (OneOf '[Int32, Int64, Word32, Word64] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Word32, Word64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessRandomUniformFullInt' :: forall v'1 v'2 dtype t tseed. (OneOf '[Int32, Int64, Word32, Word64] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Word32, Word64] tseed) => OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessRandomUniformInt :: forall v'1 v'2 v'3 v'4 dtype t tseed. (OneOf '[Int32, Int64] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor v'3 dtype -> Tensor v'4 dtype -> Tensor Build dtype
statelessRandomUniformInt' :: forall v'1 v'2 v'3 v'4 dtype t tseed. (OneOf '[Int32, Int64] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor v'3 dtype -> Tensor v'4 dtype -> 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
staticRegexFullMatch :: ByteString -> Tensor v'1 ByteString -> Tensor Build Bool
staticRegexFullMatch' :: OpParams -> ByteString -> Tensor v'1 ByteString -> Tensor Build Bool
staticRegexReplace :: ByteString -> ByteString -> Tensor v'1 ByteString -> Tensor Build ByteString
staticRegexReplace' :: OpParams -> ByteString -> ByteString -> Tensor v'1 ByteString -> Tensor Build ByteString
statsAggregatorHandle :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
statsAggregatorHandle' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
statsAggregatorHandleV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
statsAggregatorHandleV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
statsAggregatorSetSummaryWriter :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> m' ControlNode
statsAggregatorSetSummaryWriter' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> m' ControlNode
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
stringFormat :: forall v'1 t. TensorTypes t => TensorList v'1 t -> Tensor Build ByteString
stringFormat' :: forall v'1 t. TensorTypes t => OpParams -> TensorList v'1 t -> Tensor Build ByteString
stringJoin :: [Tensor v'1 ByteString] -> Tensor Build ByteString
stringJoin' :: OpParams -> [Tensor v'1 ByteString] -> Tensor Build ByteString
stringLength :: Tensor v'1 ByteString -> Tensor Build Int32
stringLength' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Int32
stringLower :: Tensor v'1 ByteString -> Tensor Build ByteString
stringLower' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
stringNGrams :: forall v'1 v'2 tsplits. OneOf '[Int32, Int64] tsplits => ByteString -> Int64 -> Bool -> ByteString -> ByteString -> Tensor v'1 ByteString -> Tensor v'2 tsplits -> (Tensor Build ByteString, Tensor Build tsplits)
stringNGrams' :: forall v'1 v'2 tsplits. OneOf '[Int32, Int64] tsplits => OpParams -> ByteString -> Int64 -> Bool -> ByteString -> ByteString -> Tensor v'1 ByteString -> Tensor v'2 tsplits -> (Tensor Build ByteString, Tensor Build tsplits)
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)
stringSplitV2 :: Tensor v'1 ByteString -> Tensor v'2 ByteString -> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64)
stringSplitV2' :: 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
stringUpper :: Tensor v'1 ByteString -> Tensor Build ByteString
stringUpper' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
sub :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, 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, Word32, 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, Word16, 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, Word16, 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
tPUOrdinalSelector :: forall m'. MonadBuild m' => m' (Tensor Value Int32)
tPUOrdinalSelector' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value Int32)
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, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
tan' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, 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' => ByteString -> Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Ref ByteString)
tensorArrayGrad' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> ByteString -> Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Ref ByteString)
tensorArrayGradV2 :: forall v'1 v'2 m'. MonadBuild m' => ByteString -> Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Value ByteString)
tensorArrayGradV2' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> ByteString -> Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Value ByteString)
tensorArrayGradV3 :: forall v'1 v'2 m'. MonadBuild m' => ByteString -> 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 -> ByteString -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value ResourceHandle, Tensor Value Float)
tensorArrayGradWithShape :: forall v'1 v'2 v'3 m'. MonadBuild m' => ByteString -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> Tensor v'3 Int32 -> m' (Tensor Value ResourceHandle, Tensor Value Float)
tensorArrayGradWithShape' :: forall v'1 v'2 v'3 m'. MonadBuild m' => OpParams -> ByteString -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> Tensor v'3 Int32 -> 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)
tensorForestCreateTreeVariable :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
tensorForestCreateTreeVariable' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
tensorForestTreeDeserialize :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
tensorForestTreeDeserialize' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
tensorForestTreeIsInitializedOp :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
tensorForestTreeIsInitializedOp' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
tensorForestTreePredict :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value Float)
tensorForestTreePredict' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value Float)
tensorForestTreeResourceHandleOp :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
tensorForestTreeResourceHandleOp' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
tensorForestTreeSerialize :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
tensorForestTreeSerialize' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
tensorForestTreeSize :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
tensorForestTreeSize' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
tensorListConcat :: forall v'1 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> (Tensor Build element_dtype, Tensor Build Int64)
tensorListConcat' :: forall v'1 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> (Tensor Build element_dtype, Tensor Build Int64)
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
tensorListConcatV2 :: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => Tensor v'1 Variant -> Tensor v'2 shape_type -> Tensor v'3 Int64 -> (Tensor Build element_dtype, Tensor Build Int64)
tensorListConcatV2' :: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => OpParams -> Tensor v'1 Variant -> Tensor v'2 shape_type -> Tensor v'3 Int64 -> (Tensor Build element_dtype, Tensor Build Int64)
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
tensorListGather :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build element_dtype
tensorListGather' :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build element_dtype
tensorListGetItem :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build element_dtype
tensorListGetItem' :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor v'3 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 v'2 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 Int32 -> (Tensor Build Variant, Tensor Build element_dtype)
tensorListPopBack' :: forall v'1 v'2 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 Int32 -> (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
tensorListResize :: Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build Variant
tensorListResize' :: OpParams -> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build Variant
tensorListScatter :: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => Tensor v'1 element_dtype -> Tensor v'2 Int32 -> Tensor v'3 shape_type -> Tensor Build Variant
tensorListScatter' :: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => OpParams -> Tensor v'1 element_dtype -> Tensor v'2 Int32 -> Tensor v'3 shape_type -> Tensor Build Variant
tensorListScatterIntoExistingList :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 element_dtype -> Tensor v'3 Int32 -> Tensor Build Variant
tensorListScatterIntoExistingList' :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 element_dtype -> Tensor v'3 Int32 -> Tensor Build Variant
tensorListScatterV2 :: forall v'1 v'2 v'3 v'4 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => Tensor v'1 element_dtype -> Tensor v'2 Int32 -> Tensor v'3 shape_type -> Tensor v'4 Int32 -> Tensor Build Variant
tensorListScatterV2' :: forall v'1 v'2 v'3 v'4 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => OpParams -> Tensor v'1 element_dtype -> Tensor v'2 Int32 -> Tensor v'3 shape_type -> Tensor v'4 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
tensorListSplit :: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => Tensor v'1 element_dtype -> Tensor v'2 shape_type -> Tensor v'3 Int64 -> Tensor Build Variant
tensorListSplit' :: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => OpParams -> Tensor v'1 element_dtype -> Tensor v'2 shape_type -> Tensor v'3 Int64 -> Tensor Build Variant
tensorListStack :: forall v'1 v'2 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build element_dtype
tensorListStack' :: forall v'1 v'2 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build element_dtype
tensorScatterAdd :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterAdd' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterMax :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterMax' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterMin :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterMin' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterSub :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterSub' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterUpdate :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterUpdate' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
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)
tensorStridedSliceUpdate :: forall v'1 v'2 v'3 v'4 v'5 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 v'5 t -> Tensor Build t
tensorStridedSliceUpdate' :: forall v'1 v'2 v'3 v'4 v'5 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 v'5 t -> Tensor Build t
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)
threadPoolDataset :: forall v'1 v'2 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' (Tensor Value Variant)
threadPoolDataset' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' (Tensor Value Variant)
threadPoolHandle :: forall m'. MonadBuild m' => ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
threadPoolHandle' :: forall m'. MonadBuild m' => OpParams -> ByteString -> Int64 -> 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)
toBool :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build Bool
toBool' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build Bool
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
tridiagonalMatMul :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build t
tridiagonalMatMul' :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build t
tridiagonalSolve :: 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
tridiagonalSolve' :: 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
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
uncompressElement :: forall v'1 output_types. TensorTypes output_types => Tensor v'1 Variant -> TensorList Build output_types
uncompressElement' :: forall v'1 output_types. TensorTypes output_types => OpParams -> Tensor v'1 Variant -> TensorList Build output_types
unicodeDecode :: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits => ByteString -> Tensor v'1 ByteString -> (Tensor Build tsplits, Tensor Build Int32)
unicodeDecode' :: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits => OpParams -> ByteString -> Tensor v'1 ByteString -> (Tensor Build tsplits, Tensor Build Int32)
unicodeDecodeWithOffsets :: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits => ByteString -> Tensor v'1 ByteString -> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)
unicodeDecodeWithOffsets' :: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits => OpParams -> ByteString -> Tensor v'1 ByteString -> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)
unicodeEncode :: forall v'1 v'2 tsplits. OneOf '[Int32, Int64] tsplits => ByteString -> Tensor v'1 Int32 -> Tensor v'2 tsplits -> Tensor Build ByteString
unicodeEncode' :: forall v'1 v'2 tsplits. OneOf '[Int32, Int64] tsplits => OpParams -> ByteString -> Tensor v'1 Int32 -> Tensor v'2 tsplits -> Tensor Build ByteString
unicodeScript :: Tensor v'1 Int32 -> Tensor Build Int32
unicodeScript' :: OpParams -> Tensor v'1 Int32 -> Tensor Build Int32
unicodeTranscode :: ByteString -> ByteString -> Tensor v'1 ByteString -> Tensor Build ByteString
unicodeTranscode' :: OpParams -> ByteString -> ByteString -> Tensor v'1 ByteString -> Tensor Build ByteString
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)
uniqueDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
uniqueDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
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
unsortedSegmentJoin :: forall v'1 v'2 v'3 tindices tnumsegments. (OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => Tensor v'1 ByteString -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build ByteString
unsortedSegmentJoin' :: forall v'1 v'2 v'3 tindices tnumsegments. (OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build ByteString
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 '[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
unsortedSegmentProd' :: 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
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)
unwrapDatasetVariant :: Tensor v'1 Variant -> Tensor Build Variant
unwrapDatasetVariant' :: OpParams -> Tensor v'1 Variant -> Tensor Build Variant
upperBound :: forall v'1 v'2 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
upperBound' :: forall v'1 v'2 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
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)
windowDataset :: [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Bool -> Tensor Build Variant
windowDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor v'5 Bool -> Tensor Build Variant
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)
wrapDatasetVariant :: Tensor v'1 Variant -> Tensor Build Variant
wrapDatasetVariant' :: OpParams -> Tensor v'1 Variant -> Tensor Build Variant
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
writeRawProtoSummary :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
writeRawProtoSummary' :: 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
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
xdivy :: 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
xdivy' :: 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
xlaBroadcastHelper :: 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 t -> Tensor v'3 tindices -> (Tensor Build t, Tensor Build t)
xlaBroadcastHelper' :: 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 t -> Tensor v'3 tindices -> (Tensor Build t, Tensor Build t)
xlaClusterOutput :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
xlaClusterOutput' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
xlaConv :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => ByteString -> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor v'4 tindices -> Tensor v'5 tindices -> Tensor v'6 tindices -> Tensor v'7 tindices -> Tensor Build t
xlaConv' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => OpParams -> ByteString -> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor v'4 tindices -> Tensor v'5 tindices -> Tensor v'6 tindices -> Tensor v'7 tindices -> Tensor Build t
xlaDequantize :: forall v'1 m'. MonadBuild m' => Float -> Float -> ByteString -> Bool -> Tensor v'1 Word32 -> m' (Tensor Value Word16)
xlaDequantize' :: forall v'1 m'. MonadBuild m' => OpParams -> Float -> Float -> ByteString -> Bool -> Tensor v'1 Word32 -> m' (Tensor Value Word16)
xlaDot :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlaDot' :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> ByteString -> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlaDynamicSlice :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tindices -> Tensor Build t
xlaDynamicSlice' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tindices -> Tensor Build t
xlaDynamicUpdateSlice :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor Build t
xlaDynamicUpdateSlice' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor Build t
xlaEinsum :: forall v'1 v'2 t. OneOf '[Complex Float, Word16, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlaEinsum' :: forall v'1 v'2 t. OneOf '[Complex Float, Word16, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlaGather :: 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) => ByteString -> Bool -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tindices -> Tensor Build t
xlaGather' :: 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 -> ByteString -> Bool -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tindices -> Tensor Build t
xlaKeyValueSort :: forall v'1 v'2 k v. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] k, TensorType v) => Tensor v'1 k -> Tensor v'2 v -> (Tensor Build k, Tensor Build v)
xlaKeyValueSort' :: forall v'1 v'2 k v. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] k, TensorType v) => OpParams -> Tensor v'1 k -> Tensor v'2 v -> (Tensor Build k, Tensor Build v)
xlaPad :: forall v'1 v'2 v'3 v'4 v'5 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor v'4 tindices -> Tensor v'5 tindices -> Tensor Build t
xlaPad' :: forall v'1 v'2 v'3 v'4 v'5 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor v'4 tindices -> Tensor v'5 tindices -> Tensor Build t
xlaRecv :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> ByteString -> m' (Tensor Value dtype)
xlaRecv' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> ByteString -> m' (Tensor Value dtype)
xlaReplicaId :: Tensor Build Int32
xlaReplicaId' :: OpParams -> Tensor Build Int32
xlaSelfAdjointEig :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Float -> Bool -> Int64 -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
xlaSelfAdjointEig' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Float -> Bool -> Int64 -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
xlaSend :: forall v'1 t m'. (MonadBuild m', TensorType t) => ByteString -> Tensor v'1 t -> m' ControlNode
xlaSend' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> Tensor v'1 t -> m' ControlNode
xlaSharding :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
xlaSharding' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
xlaSort :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
xlaSort' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
xlaSpmdFullToShardShape :: forall v'1 t. TensorType t => ByteString -> Tensor v'1 t -> Tensor Build t
xlaSpmdFullToShardShape' :: forall v'1 t. TensorType t => OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
xlaSpmdShardToFullShape :: forall v'1 t. TensorType t => Shape -> ByteString -> Tensor v'1 t -> Tensor Build t
xlaSpmdShardToFullShape' :: forall v'1 t. TensorType t => OpParams -> Shape -> ByteString -> Tensor v'1 t -> Tensor Build t
xlaSvd :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Float -> Int64 -> ByteString -> Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
xlaSvd' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Float -> Int64 -> ByteString -> Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
xlog1py :: 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
xlog1py' :: 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
xlogy :: 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
xlogy' :: 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
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)
_DeviceArg :: forall t m'. (MonadBuild m', TensorType t) => Int64 -> m' (Tensor Value t)
_DeviceArg' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> m' (Tensor Value t)
_DeviceRetval :: forall v'1 t m'. (MonadBuild m', TensorType t) => Int64 -> Tensor v'1 t -> m' ControlNode
_DeviceRetval' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
_DisconnectHostFromDistributedTPUSystem :: forall m'. MonadBuild m' => m' (Tensor Value Int32)
_DisconnectHostFromDistributedTPUSystem' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value Int32)
_FusedBatchNormEx :: forall v'1 v'2 v'3 v'4 v'5 v'6 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 v'6 t] -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u)
_FusedBatchNormEx' :: forall v'1 v'2 v'3 v'4 v'5 v'6 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 v'6 t] -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u)
_FusedConv2D :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
_FusedConv2D' :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
_FusedDepthwiseConv2dNative :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
_FusedDepthwiseConv2dNative' :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
_FusedMatMul :: 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
_FusedMatMul' :: 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
_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) => ByteString -> ByteString -> Int64 -> ByteString -> m' (Tensor Value tensor_type)
_HostRecv' :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> m' (Tensor Value tensor_type)
_HostSend :: forall v'1 t m'. (MonadBuild m', TensorType t) => ByteString -> ByteString -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
_HostSend' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> 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]
_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)
_NcclBroadcastRecv :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> ByteString -> Tensor v'1 Int32 -> m' (Tensor Value t)
_NcclBroadcastRecv' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> ByteString -> Tensor v'1 Int32 -> m' (Tensor Value t)
_NcclBroadcastSend :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
_NcclBroadcastSend' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
_NcclReduceRecv :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> ByteString -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
_NcclReduceRecv' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> ByteString -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
_NcclReduceSend :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> ByteString -> ByteString -> Tensor v'1 t -> m' ControlNode
_NcclReduceSend' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> ByteString -> ByteString -> Tensor v'1 t -> m' ControlNode
_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
_ReadVariablesOp :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => [Tensor v'1 ResourceHandle] -> m' (TensorList Value dtypes)
_ReadVariablesOp' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> [Tensor v'1 ResourceHandle] -> m' (TensorList Value dtypes)
_Recv :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => ByteString -> ByteString -> Int64 -> ByteString -> m' (Tensor Value tensor_type)
_Recv' :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> 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 -> ByteString -> Shape -> m' (Tensor Value t)
_ScopedAllocator' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Int64 -> ByteString -> Shape -> m' (Tensor Value t)
_ScopedAllocatorConcat :: forall v'1 v'2 t m'. (MonadBuild m', TensorType t) => Int64 -> ByteString -> 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 -> ByteString -> 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 -> ByteString -> 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 -> ByteString -> Tensor v'1 t -> [Tensor v'2 t] -> m' [Tensor Value t]
_Send :: forall v'1 t m'. (MonadBuild m', TensorType t) => ByteString -> ByteString -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
_Send' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> 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
_SwitchN :: forall v'1 v'2 t. TensorType t => Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
_SwitchN' :: forall v'1 v'2 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
_UnaryOpsComposition :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
_UnaryOpsComposition' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
_VarHandlesOp :: forall m'. MonadBuild m' => Int64 -> [DataType] -> m' [Tensor Value ResourceHandle]
_VarHandlesOp' :: forall m'. MonadBuild m' => OpParams -> Int64 -> [DataType] -> m' [Tensor Value ResourceHandle]
_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)
_XlaMerge :: forall v'1 v'2 t. TensorType t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
_XlaMerge' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
_XlaRecvAtHost :: forall v'1 toutputs m'. (MonadBuild m', TensorTypes toutputs) => Int64 -> ByteString -> Tensor v'1 ByteString -> m' (TensorList Value toutputs)
_XlaRecvAtHost' :: forall v'1 toutputs m'. (MonadBuild m', TensorTypes toutputs) => OpParams -> Int64 -> ByteString -> Tensor v'1 ByteString -> m' (TensorList Value toutputs)
_XlaRun :: forall v'1 v'2 targs tresults m'. (MonadBuild m', TensorTypes targs, TensorTypes tresults) => TensorList v'1 targs -> Tensor v'2 ByteString -> m' (TensorList Value tresults)
_XlaRun' :: forall v'1 v'2 targs tresults m'. (MonadBuild m', TensorTypes targs, TensorTypes tresults) => OpParams -> TensorList v'1 targs -> Tensor v'2 ByteString -> m' (TensorList Value tresults)
_XlaSendFromHost :: forall v'1 v'2 tinputs m'. (MonadBuild m', TensorTypes tinputs) => Int64 -> ByteString -> TensorList v'1 tinputs -> Tensor v'2 ByteString -> m' ControlNode
_XlaSendFromHost' :: forall v'1 v'2 tinputs m'. (MonadBuild m', TensorTypes tinputs) => OpParams -> Int64 -> ByteString -> 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 #

abs Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

abs' Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

accumulateNV2 Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Shape	shape
-> [Tensor v'1 t]	inputs
-> Tensor Build t	sum

accumulateNV2' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Shape	shape
-> [Tensor v'1 t]	inputs
-> Tensor Build t	sum

accumulatorApplyGradient Source #

Arguments

:: 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	handle
-> Tensor v'2 Int64	local_step
-> Tensor v'3 dtype	gradient
-> m' ControlNode

accumulatorApplyGradient' Source #

Arguments

:: 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	handle
-> Tensor v'2 Int64	local_step
-> Tensor v'3 dtype	gradient
-> m' ControlNode

accumulatorNumAccumulated Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	handle
-> m' (Tensor Value Int32)	num_accumulated

accumulatorNumAccumulated' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	handle
-> m' (Tensor Value Int32)	num_accumulated

accumulatorSetGlobalStep Source #

Arguments

:: forall v'2 m'. MonadBuild m'
=> Tensor Ref ByteString	handle
-> Tensor v'2 Int64	new_global_step
-> m' ControlNode

accumulatorSetGlobalStep' Source #

Arguments

:: forall v'2 m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	handle
-> Tensor v'2 Int64	new_global_step
-> m' ControlNode

accumulatorTakeGradient Source #

Arguments

:: 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	handle
-> Tensor v'2 Int32	num_required
-> m' (Tensor Value dtype)	average

accumulatorTakeGradient' Source #

Arguments

:: 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	handle
-> Tensor v'2 Int32	num_required
-> m' (Tensor Value dtype)	average

acos Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

acos' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

acosh Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

acosh' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

add Source #

Arguments

:: 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	x
-> Tensor v'2 t	y
-> Tensor Build t	z

add' Source #

Arguments

:: 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	x
-> Tensor v'2 t	y
-> Tensor Build t	z

addManySparseToTensorsMap Source #

Arguments

:: forall v'1 v'2 v'3 t m'. (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

addManySparseToTensorsMap' Source #

Arguments

:: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor v'1 Int64	sparse_indices
-> Tensor v'2 t	sparse_values
-> Tensor v'3 Int64	sparse_shape
-> m' (Tensor Value Int64)	sparse_handles

addN Source #

Arguments

:: 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]	inputs
-> Tensor Build t	sum

addN' Source #

Arguments

:: 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]	inputs
-> Tensor Build t	sum

addSparseToTensorsMap Source #

Arguments

:: forall v'1 v'2 v'3 t m'. (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

addSparseToTensorsMap' Source #

Arguments

:: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor v'1 Int64	sparse_indices
-> Tensor v'2 t	sparse_values
-> Tensor v'3 Int64	sparse_shape
-> m' (Tensor Value Int64)	sparse_handle

addV2 Source #

Arguments

:: forall v'1 v'2 t. 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 Build t	z

addV2' Source #

Arguments

:: 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	x
-> Tensor v'2 t	y
-> Tensor Build t	z

adjustContrast Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Float	contrast_factor
-> Tensor v'3 Float	min_value
-> Tensor v'4 Float	max_value
-> Tensor Build Float	output

adjustContrast' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor v'2 Float	contrast_factor
-> Tensor v'3 Float	min_value
-> Tensor v'4 Float	max_value
-> Tensor Build Float	output

adjustContrastv2 Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Float	contrast_factor
-> Tensor Build t	output

adjustContrastv2' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor v'2 Float	contrast_factor
-> Tensor Build t	output

adjustHue Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Float	delta
-> Tensor Build t	output

adjustHue' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor v'2 Float	delta
-> Tensor Build t	output

adjustSaturation Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Float	scale
-> Tensor Build t	output

adjustSaturation' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor v'2 Float	scale
-> Tensor Build t	output

all Source #

Arguments

:: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx
=> Tensor v'1 Bool	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build Bool	output

all' Source #

Arguments

:: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx
=> OpParams
-> Tensor v'1 Bool	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build Bool	output

allCandidateSampler Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Int64	num_sampled
-> Int64	num_true
-> 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) sampled_candidates true_expected_count sampled_expected_count

allCandidateSampler' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Int64	num_sampled
-> Int64	num_true
-> 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) sampled_candidates true_expected_count sampled_expected_count

allToAll Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Int64	concat_dimension
-> Int64	split_count
-> Int64	split_dimension
-> Tensor v'1 t	input
-> Tensor v'2 Int32	group_assignment
-> Tensor Build t	output

allToAll' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Int64	concat_dimension
-> Int64	split_count
-> Int64	split_dimension
-> Tensor v'1 t	input
-> Tensor v'2 Int32	group_assignment
-> Tensor Build t	output

angle Source #

Arguments

:: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout)
=> Tensor v'1 t	input
-> Tensor Build tout	output

angle' Source #

Arguments

:: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout)
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build tout	output

anonymousIterator Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	output_types
-> m' (Tensor Value ResourceHandle)	handle

anonymousIterator' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> m' (Tensor Value ResourceHandle)	handle

anonymousIteratorV2 Source #

Arguments

:: forall m'. MonadBuild m'

=> [DataType]

output_types

-> m' (Tensor Value ResourceHandle, Tensor Value Variant)

(handle, deleter)

handle
deleter

anonymousIteratorV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)	(handle, deleter) handle deleter

anonymousMemoryCache Source #

Arguments

:: forall m'. MonadBuild m'

=> m' (Tensor Value ResourceHandle, Tensor Value Variant)

(handle, deleter)

handle
deleter

anonymousMemoryCache' Source #

Arguments

:: forall m'. MonadBuild m'

=> OpParams

-> m' (Tensor Value ResourceHandle, Tensor Value Variant)

(handle, deleter)

handle
deleter

anonymousMultiDeviceIterator Source #

Arguments

:: forall m'. MonadBuild m'

=> [DataType]

output_types

-> m' (Tensor Value ResourceHandle, Tensor Value Variant)

(handle, deleter)

handle
deleter

anonymousMultiDeviceIterator' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)	(handle, deleter) handle deleter

anonymousRandomSeedGenerator Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 Int64	seed
-> Tensor v'2 Int64	seed2
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)	(handle, deleter) handle deleter

anonymousRandomSeedGenerator' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 Int64	seed
-> Tensor v'2 Int64	seed2
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)	(handle, deleter) handle deleter

anonymousSeedGenerator Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 Int64	seed
-> Tensor v'2 Int64	seed2
-> Tensor v'3 Bool	reshuffle
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)	(handle, deleter) handle deleter

anonymousSeedGenerator' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 Int64	seed
-> Tensor v'2 Int64	seed2
-> Tensor v'3 Bool	reshuffle
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)	(handle, deleter) handle deleter

any Source #

Arguments

:: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx
=> Tensor v'1 Bool	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build Bool	output

any' Source #

Arguments

:: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx
=> OpParams
-> Tensor v'1 Bool	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build Bool	output

applyAdaMax Source #

Arguments

:: 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	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

applyAdaMax' Source #

Arguments

:: 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	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

applyAdadelta Source #

Arguments

:: 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	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

applyAdadelta' Source #

Arguments

:: 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	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

applyAdagrad Source #

Arguments

:: 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	var
-> Tensor Ref t	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	grad
-> m' (Tensor Ref t)	out

applyAdagrad' Source #

Arguments

:: 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	var
-> Tensor Ref t	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	grad
-> m' (Tensor Ref t)	out

applyAdagradDA Source #

Arguments

:: 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	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

applyAdagradDA' Source #

Arguments

:: 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	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

applyAdagradV2 Source #

Arguments

:: 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	var
-> Tensor Ref t	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	epsilon
-> Tensor v'5 t	grad
-> m' (Tensor Ref t)	out

applyAdagradV2' Source #

Arguments

:: 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	var
-> Tensor Ref t	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	epsilon
-> Tensor v'5 t	grad
-> m' (Tensor Ref t)	out

applyAdam Source #

Arguments

:: 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	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

applyAdam' Source #

Arguments

:: 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	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

applyAddSign Source #

Arguments

:: 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	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

applyAddSign' Source #

Arguments

:: 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	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

applyCenteredRMSProp Source #

Arguments

:: 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	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 #

Arguments

:: 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	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

applyFtrl Source #

Arguments

:: 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	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

applyFtrl' Source #

Arguments

:: 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	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

applyFtrlV2 Source #

Arguments

:: 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	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

applyFtrlV2' Source #

Arguments

:: 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	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

applyGradientDescent Source #

Arguments

:: 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	var
-> Tensor v'2 t	alpha
-> Tensor v'3 t	delta
-> m' (Tensor Ref t)	out

applyGradientDescent' Source #

Arguments

:: 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	var
-> Tensor v'2 t	alpha
-> Tensor v'3 t	delta
-> m' (Tensor Ref t)	out

applyMomentum Source #

Arguments

:: 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	var
-> Tensor Ref t	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	grad
-> Tensor v'5 t	momentum
-> m' (Tensor Ref t)	out

applyMomentum' Source #

Arguments

:: 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	var
-> Tensor Ref t	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	grad
-> Tensor v'5 t	momentum
-> m' (Tensor Ref t)	out

applyPowerSign Source #

Arguments

:: 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	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

applyPowerSign' Source #

Arguments

:: 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	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

applyProximalAdagrad Source #

Arguments

:: 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	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
-> m' (Tensor Ref t)	out

applyProximalAdagrad' Source #

Arguments

:: 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	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
-> m' (Tensor Ref t)	out

applyProximalGradientDescent Source #

Arguments

:: 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	var
-> Tensor v'2 t	alpha
-> Tensor v'3 t	l1
-> Tensor v'4 t	l2
-> Tensor v'5 t	delta
-> m' (Tensor Ref t)	out

applyProximalGradientDescent' Source #

Arguments

:: 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	var
-> Tensor v'2 t	alpha
-> Tensor v'3 t	l1
-> Tensor v'4 t	l2
-> Tensor v'5 t	delta
-> m' (Tensor Ref t)	out

applyRMSProp Source #

Arguments

:: 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	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

applyRMSProp' Source #

Arguments

:: 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	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

approximateEqual Source #

Arguments

:: 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	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

approximateEqual' Source #

Arguments

:: 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	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

argMax Source #

Arguments

:: forall v'1 v'2 t tidx output_type. (OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] output_type)
=> Tensor v'1 t	input
-> Tensor v'2 tidx	dimension
-> Tensor Build output_type	output

argMax' Source #

Arguments

:: forall v'1 v'2 t tidx output_type. (OneOf '[Complex Double, Complex Float, Bool, 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	input
-> Tensor v'2 tidx	dimension
-> Tensor Build output_type	output

argMin Source #

Arguments

:: forall v'1 v'2 t tidx output_type. (OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] output_type)
=> Tensor v'1 t	input
-> Tensor v'2 tidx	dimension
-> Tensor Build output_type	output

argMin' Source #

Arguments

:: forall v'1 v'2 t tidx output_type. (OneOf '[Complex Double, Complex Float, Bool, 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	input
-> Tensor v'2 tidx	dimension
-> Tensor Build output_type	output

asString Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build ByteString	output

asString' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build ByteString	output

asin Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

asin' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

asinh Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

asinh' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

assert Source #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', TensorTypes t)
=> Tensor v'1 Bool	condition
-> TensorList v'2 t	data
-> m' ControlNode

assert' Source #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', TensorTypes t)
=> OpParams
-> Tensor v'1 Bool	condition
-> TensorList v'2 t	data
-> m' ControlNode

assertCardinalityDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	cardinality
-> Tensor Build Variant	handle

assertCardinalityDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	cardinality
-> Tensor Build Variant	handle

assertNextDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	transformations
-> Tensor Build Variant	handle

assertNextDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	transformations
-> Tensor Build Variant	handle

assign Source #

Arguments

:: forall v'2 t m'. (MonadBuild m', TensorType t)
=> Tensor Ref t	ref
-> Tensor v'2 t	value
-> m' (Tensor Ref t)	output_ref

assign' Source #

Arguments

:: forall v'2 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor Ref t	ref
-> Tensor v'2 t	value
-> m' (Tensor Ref t)	output_ref

assignAdd Source #

Arguments

:: 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	ref
-> Tensor v'2 t	value
-> m' (Tensor Ref t)	output_ref

assignAdd' Source #

Arguments

:: 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	ref
-> Tensor v'2 t	value
-> m' (Tensor Ref t)	output_ref

assignAddVariableOp Source #

Arguments

:: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 dtype	value
-> m' ControlNode

assignAddVariableOp' Source #

Arguments

:: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 dtype	value
-> m' ControlNode

assignSub Source #

Arguments

:: 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	ref
-> Tensor v'2 t	value
-> m' (Tensor Ref t)	output_ref

assignSub' Source #

Arguments

:: 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	ref
-> Tensor v'2 t	value
-> m' (Tensor Ref t)	output_ref

assignSubVariableOp Source #

Arguments

:: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 dtype	value
-> m' ControlNode

assignSubVariableOp' Source #

Arguments

:: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 dtype	value
-> m' ControlNode

assignVariableOp Source #

Arguments

:: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 dtype	value
-> m' ControlNode

assignVariableOp' Source #

Arguments

:: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 dtype	value
-> m' ControlNode

atan Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

atan' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

atan2 Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	y
-> Tensor v'2 t	x
-> Tensor Build t	z

atan2' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	y
-> Tensor v'2 t	x
-> Tensor Build t	z

atanh Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

atanh' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

audioSpectrogram Source #

Arguments

:: Int64	stride
-> Int64	window_size
-> Tensor v'1 Float	input
-> Tensor Build Float	spectrogram

audioSpectrogram' Source #

Arguments

:: OpParams
-> Int64	stride
-> Int64	window_size
-> Tensor v'1 Float	input
-> Tensor Build Float	spectrogram

audioSummary Source #

Arguments

:: Float	sample_rate
-> Tensor v'1 ByteString	tag
-> Tensor v'2 Float	tensor
-> Tensor Build ByteString	summary

audioSummary' Source #

Arguments

:: OpParams
-> Float	sample_rate
-> Tensor v'1 ByteString	tag
-> Tensor v'2 Float	tensor
-> Tensor Build ByteString	summary

audioSummaryV2 Source #

Arguments

:: Tensor v'1 ByteString	tag
-> Tensor v'2 Float	tensor
-> Tensor v'3 Float	sample_rate
-> Tensor Build ByteString	summary

audioSummaryV2' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	tag
-> Tensor v'2 Float	tensor
-> Tensor v'3 Float	sample_rate
-> Tensor Build ByteString	summary

autoShardDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_workers
-> Tensor v'3 Int64	index
-> Tensor Build Variant	handle

autoShardDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_workers
-> Tensor v'3 Int64	index
-> Tensor Build Variant	handle

avgPool Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	value
-> Tensor Build t	output

avgPool' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	value
-> Tensor Build t	output

avgPool3D Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor Build t	output

avgPool3D' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor Build t	output

avgPool3DGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 Int32	orig_input_shape
-> Tensor v'2 t	grad
-> Tensor Build t	output

avgPool3DGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 Int32	orig_input_shape
-> Tensor v'2 t	grad
-> Tensor Build t	output

avgPoolGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 Int32	orig_input_shape
-> Tensor v'2 t	grad
-> Tensor Build t	output

avgPoolGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 Int32	orig_input_shape
-> Tensor v'2 t	grad
-> Tensor Build t	output

bandedTriangularSolve Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor Build t	output

bandedTriangularSolve' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor Build t	output

barrier Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	component_types
-> m' (Tensor Ref ByteString)	handle

barrier' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	component_types
-> m' (Tensor Ref ByteString)	handle

barrierClose Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	handle
-> m' ControlNode

barrierClose' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	handle
-> m' ControlNode

barrierIncompleteSize Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	handle
-> m' (Tensor Value Int32)	size

barrierIncompleteSize' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	handle
-> m' (Tensor Value Int32)	size

barrierInsertMany Source #

Arguments

:: forall v'2 v'3 t m'. (MonadBuild m', TensorType t)
=> Int64	component_index
-> Tensor Ref ByteString	handle
-> Tensor v'2 ByteString	keys
-> Tensor v'3 t	values
-> m' ControlNode

barrierInsertMany' Source #

Arguments

:: forall v'2 v'3 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Int64	component_index
-> Tensor Ref ByteString	handle
-> Tensor v'2 ByteString	keys
-> Tensor v'3 t	values
-> m' ControlNode

barrierReadySize Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	handle
-> m' (Tensor Value Int32)	size

barrierReadySize' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	handle
-> m' (Tensor Value Int32)	size

barrierTakeMany Source #

Arguments

:: forall v'2 component_types m'. (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) indices keys values

barrierTakeMany' Source #

Arguments

:: forall v'2 component_types m'. (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) indices keys values

batch Source #

Arguments

:: forall v'1 t. 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) batched_tensors batch_index id

batch' Source #

Arguments

:: forall v'1 t. 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) batched_tensors batch_index id

batchCholesky Source #

Arguments

:: forall v'1 t. OneOf '[Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

batchCholesky' Source #

Arguments

:: forall v'1 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

batchCholeskyGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 t	l
-> Tensor v'2 t	grad
-> Tensor Build t	output

batchCholeskyGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	l
-> Tensor v'2 t	grad
-> Tensor Build t	output

batchDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> Tensor Build Variant	handle

batchDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> Tensor Build Variant	handle

batchDatasetV2 Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> Tensor v'3 Bool	drop_remainder
-> Tensor Build Variant	handle

batchDatasetV2' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> Tensor v'3 Bool	drop_remainder
-> Tensor Build Variant	handle

batchFFT Source #

Arguments

:: Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchFFT' Source #

Arguments

:: OpParams
-> Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchFFT2D Source #

Arguments

:: Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchFFT2D' Source #

Arguments

:: OpParams
-> Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchFFT3D Source #

Arguments

:: Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchFFT3D' Source #

Arguments

:: OpParams
-> Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchIFFT Source #

Arguments

:: Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchIFFT' Source #

Arguments

:: OpParams
-> Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchIFFT2D Source #

Arguments

:: Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchIFFT2D' Source #

Arguments

:: OpParams
-> Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchIFFT3D Source #

Arguments

:: Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchIFFT3D' Source #

Arguments

:: OpParams
-> Tensor v'1 (Complex Float)	input
-> Tensor Build (Complex Float)	output

batchMatMul Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	output

batchMatMul' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	output

batchMatMulV2 Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	output

batchMatMulV2' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	output

batchMatrixBandPart Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 t	input
-> Tensor v'2 Int64	num_lower
-> Tensor v'3 Int64	num_upper
-> Tensor Build t	band

batchMatrixBandPart' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 Int64	num_lower
-> Tensor v'3 Int64	num_upper
-> Tensor Build t	band

batchMatrixDeterminant Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

batchMatrixDeterminant' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

batchMatrixDiag Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	diagonal
-> Tensor Build t	output

batchMatrixDiag' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	diagonal
-> Tensor Build t	output

batchMatrixDiagPart Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	diagonal

batchMatrixDiagPart' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	diagonal

batchMatrixInverse Source #

Arguments

:: forall v'1 t. OneOf '[Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

batchMatrixInverse' Source #

Arguments

:: forall v'1 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

batchMatrixSetDiag Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Tensor v'1 t	input
-> Tensor v'2 t	diagonal
-> Tensor Build t	output

batchMatrixSetDiag' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 t	diagonal
-> Tensor Build t	output

batchMatrixSolve Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor Build t	output

batchMatrixSolve' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor Build t	output

batchMatrixSolveLs Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t
=> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor v'3 Double	l2_regularizer
-> Tensor Build t	output

batchMatrixSolveLs' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor v'3 Double	l2_regularizer
-> Tensor Build t	output

batchMatrixTriangularSolve Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor Build t	output

batchMatrixTriangularSolve' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor Build t	output

batchNormWithGlobalNormalization Source #

Arguments

:: 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	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	beta
-> Tensor v'5 t	gamma
-> Tensor Build t	result

batchNormWithGlobalNormalization' Source #

Arguments

:: 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	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	beta
-> Tensor v'5 t	gamma
-> Tensor Build t	result

batchNormWithGlobalNormalizationGrad Source #

Arguments

:: 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	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) dx dm dv db dg

batchNormWithGlobalNormalizationGrad' Source #

Arguments

:: 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	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) dx dm dv db dg

batchSelfAdjointEig Source #

Arguments

:: forall v'1 t. OneOf '[Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

batchSelfAdjointEig' Source #

Arguments

:: forall v'1 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

batchSelfAdjointEigV2 Source #

Arguments

:: forall v'1 t. OneOf '[Double, Float] t

=> Tensor v'1 t

input

-> (Tensor Build t, Tensor Build t)

(e, v)

e
v

batchSelfAdjointEigV2' Source #

Arguments

:: forall v'1 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> (Tensor Build t, Tensor Build t)	(e, v) e v

batchSvd Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t

=> Tensor v'1 t

input

-> (Tensor Build t, Tensor Build t, Tensor Build t)

(s, u, v)

s
u
v

batchSvd' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> (Tensor Build t, Tensor Build t, Tensor Build t)	(s, u, v) s u v

batchToSpace Source #

Arguments

:: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx)
=> Int64	block_size
-> Tensor v'1 t	input
-> Tensor v'2 tidx	crops
-> Tensor Build t	output

batchToSpace' Source #

Arguments

:: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx)
=> OpParams
-> Int64	block_size
-> Tensor v'1 t	input
-> Tensor v'2 tidx	crops
-> Tensor Build t	output

batchToSpaceND Source #

Arguments

:: 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	input
-> Tensor v'2 tblock_shape	block_shape
-> Tensor v'3 tcrops	crops
-> Tensor Build t	output

batchToSpaceND' Source #

Arguments

:: 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	input
-> Tensor v'2 tblock_shape	block_shape
-> Tensor v'3 tcrops	crops
-> Tensor Build t	output

besselI0 Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselI0' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

besselI0e Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselI0e' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

besselI1 Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselI1' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

besselI1e Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselI1e' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

besselJ0 Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselJ0' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

besselJ1 Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselJ1' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

besselK0 Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselK0' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

besselK0e Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselK0e' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

besselK1 Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselK1' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

besselK1e Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselK1e' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

besselY0 Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselY0' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

besselY1 Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

besselY1' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

betainc Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t
=> Tensor v'1 t	a
-> Tensor v'2 t	b
-> Tensor v'3 t	x
-> Tensor Build t	z

betainc' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	a
-> Tensor v'2 t	b
-> Tensor v'3 t	x
-> Tensor Build t	z

biasAdd Source #

Arguments

:: 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	value
-> Tensor v'2 t	bias
-> Tensor Build t	output

biasAdd' Source #

Arguments

:: 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	value
-> Tensor v'2 t	bias
-> Tensor Build t	output

biasAddGrad Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	out_backprop
-> Tensor Build t	output

biasAddGrad' Source #

Arguments

:: 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	out_backprop
-> Tensor Build t	output

biasAddV1 Source #

Arguments

:: 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	value
-> Tensor v'2 t	bias
-> Tensor Build t	output

biasAddV1' Source #

Arguments

:: 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	value
-> Tensor v'2 t	bias
-> Tensor Build t	output

bincount Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64, Double, Float] t
=> Tensor v'1 Int32	arr
-> Tensor v'2 Int32	size
-> Tensor v'3 t	weights
-> Tensor Build t	bins

bincount' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64, Double, Float] t
=> OpParams
-> Tensor v'1 Int32	arr
-> Tensor v'2 Int32	size
-> Tensor v'3 t	weights
-> Tensor Build t	bins

bitcast Source #

Arguments

:: forall v'1 t type'. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] type')
=> Tensor v'1 t	input
-> Tensor Build type'	output

bitcast' Source #

Arguments

:: forall v'1 t type'. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] type')
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build type'	output

bitwiseAnd Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

bitwiseAnd' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

bitwiseOr Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

bitwiseOr' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

bitwiseXor Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

bitwiseXor' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

blockLSTM Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t. OneOf '[Word16, Float] t
=> Tensor v'1 Int64	seq_len_max
-> Tensor v'2 t	x
-> Tensor v'3 t	cs_prev
-> Tensor v'4 t	h_prev
-> Tensor v'5 t	w
-> Tensor v'6 t	wci
-> Tensor v'7 t	wcf
-> Tensor v'8 t	wco
-> Tensor v'9 t	b
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(i, cs, f, o, ci, co, h) i cs f o ci co h

blockLSTM' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t. OneOf '[Word16, Float] t
=> OpParams
-> Tensor v'1 Int64	seq_len_max
-> Tensor v'2 t	x
-> Tensor v'3 t	cs_prev
-> Tensor v'4 t	h_prev
-> Tensor v'5 t	w
-> Tensor v'6 t	wci
-> Tensor v'7 t	wcf
-> Tensor v'8 t	wco
-> Tensor v'9 t	b
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(i, cs, f, o, ci, co, h) i cs f o ci co h

blockLSTMGrad Source #

Arguments

:: 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 v'16 v'17 v'18 t. OneOf '[Word16, Float] t
=> Bool	use_peephole
-> Tensor v'1 Int64	seq_len_max
-> Tensor v'2 t	x
-> Tensor v'3 t	cs_prev
-> Tensor v'4 t	h_prev
-> Tensor v'5 t	w
-> Tensor v'6 t	wci
-> Tensor v'7 t	wcf
-> Tensor v'8 t	wco
-> Tensor v'9 t	b
-> Tensor v'10 t	i
-> Tensor v'11 t	cs
-> Tensor v'12 t	f
-> Tensor v'13 t	o
-> Tensor v'14 t	ci
-> Tensor v'15 t	co
-> Tensor v'16 t	h
-> Tensor v'17 t	cs_grad
-> Tensor v'18 t	h_grad
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(x_grad, cs_prev_grad, h_prev_grad, w_grad, wci_grad, wcf_grad, wco_grad, b_grad) x_grad cs_prev_grad h_prev_grad w_grad wci_grad wcf_grad wco_grad b_grad

blockLSTMGrad' Source #

Arguments

:: 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 v'16 v'17 v'18 t. OneOf '[Word16, Float] t
=> OpParams
-> Bool	use_peephole
-> Tensor v'1 Int64	seq_len_max
-> Tensor v'2 t	x
-> Tensor v'3 t	cs_prev
-> Tensor v'4 t	h_prev
-> Tensor v'5 t	w
-> Tensor v'6 t	wci
-> Tensor v'7 t	wcf
-> Tensor v'8 t	wco
-> Tensor v'9 t	b
-> Tensor v'10 t	i
-> Tensor v'11 t	cs
-> Tensor v'12 t	f
-> Tensor v'13 t	o
-> Tensor v'14 t	ci
-> Tensor v'15 t	co
-> Tensor v'16 t	h
-> Tensor v'17 t	cs_grad
-> Tensor v'18 t	h_grad
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(x_grad, cs_prev_grad, h_prev_grad, w_grad, wci_grad, wcf_grad, wco_grad, b_grad) x_grad cs_prev_grad h_prev_grad w_grad wci_grad wcf_grad wco_grad b_grad

blockLSTMGradV2 Source #

Arguments

:: 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 v'16 v'17 v'18 t. OneOf '[Word16, Float] t
=> Bool	use_peephole
-> Tensor v'1 Int64	seq_len_max
-> Tensor v'2 t	x
-> Tensor v'3 t	cs_prev
-> Tensor v'4 t	h_prev
-> Tensor v'5 t	w
-> Tensor v'6 t	wci
-> Tensor v'7 t	wcf
-> Tensor v'8 t	wco
-> Tensor v'9 t	b
-> Tensor v'10 t	i
-> Tensor v'11 t	cs
-> Tensor v'12 t	f
-> Tensor v'13 t	o
-> Tensor v'14 t	ci
-> Tensor v'15 t	co
-> Tensor v'16 t	h
-> Tensor v'17 t	cs_grad
-> Tensor v'18 t	h_grad
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(x_grad, cs_prev_grad, h_prev_grad, w_grad, wci_grad, wcf_grad, wco_grad, b_grad) x_grad cs_prev_grad h_prev_grad w_grad wci_grad wcf_grad wco_grad b_grad

blockLSTMGradV2' Source #

Arguments

:: 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 v'16 v'17 v'18 t. OneOf '[Word16, Float] t
=> OpParams
-> Bool	use_peephole
-> Tensor v'1 Int64	seq_len_max
-> Tensor v'2 t	x
-> Tensor v'3 t	cs_prev
-> Tensor v'4 t	h_prev
-> Tensor v'5 t	w
-> Tensor v'6 t	wci
-> Tensor v'7 t	wcf
-> Tensor v'8 t	wco
-> Tensor v'9 t	b
-> Tensor v'10 t	i
-> Tensor v'11 t	cs
-> Tensor v'12 t	f
-> Tensor v'13 t	o
-> Tensor v'14 t	ci
-> Tensor v'15 t	co
-> Tensor v'16 t	h
-> Tensor v'17 t	cs_grad
-> Tensor v'18 t	h_grad
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(x_grad, cs_prev_grad, h_prev_grad, w_grad, wci_grad, wcf_grad, wco_grad, b_grad) x_grad cs_prev_grad h_prev_grad w_grad wci_grad wcf_grad wco_grad b_grad

blockLSTMV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t. OneOf '[Word16, Float] t
=> Tensor v'1 Int64	seq_len_max
-> Tensor v'2 t	x
-> Tensor v'3 t	cs_prev
-> Tensor v'4 t	h_prev
-> Tensor v'5 t	w
-> Tensor v'6 t	wci
-> Tensor v'7 t	wcf
-> Tensor v'8 t	wco
-> Tensor v'9 t	b
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(i, cs, f, o, ci, co, h) i cs f o ci co h

blockLSTMV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t. OneOf '[Word16, Float] t
=> OpParams
-> Tensor v'1 Int64	seq_len_max
-> Tensor v'2 t	x
-> Tensor v'3 t	cs_prev
-> Tensor v'4 t	h_prev
-> Tensor v'5 t	w
-> Tensor v'6 t	wci
-> Tensor v'7 t	wcf
-> Tensor v'8 t	wco
-> Tensor v'9 t	b
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(i, cs, f, o, ci, co, h) i cs f o ci co h

boostedTreesAggregateStats Source #

Arguments

:: Int64	max_splits
-> Int64	num_buckets
-> Tensor v'1 Int32	node_ids
-> Tensor v'2 Float	gradients
-> Tensor v'3 Float	hessians
-> Tensor v'4 Int32	feature
-> Tensor Build Float	stats_summary

boostedTreesAggregateStats' Source #

Arguments

:: OpParams
-> Int64	max_splits
-> Int64	num_buckets
-> Tensor v'1 Int32	node_ids
-> Tensor v'2 Float	gradients
-> Tensor v'3 Float	hessians
-> Tensor v'4 Int32	feature
-> Tensor Build Float	stats_summary

boostedTreesBucketize Source #

Arguments

:: [Tensor v'1 Float]	float_values
-> [Tensor v'2 Float]	bucket_boundaries
-> [Tensor Build Int32]	buckets

boostedTreesBucketize' Source #

Arguments

:: OpParams
-> [Tensor v'1 Float]	float_values
-> [Tensor v'2 Float]	bucket_boundaries
-> [Tensor Build Int32]	buckets

boostedTreesCalculateBestFeatureSplit Source #

Arguments

:: Int64	logits_dimension
-> Tensor v'1 Int32	node_id_range
-> Tensor v'2 Float	stats_summary
-> 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 Int32, Tensor Build Float, Tensor Build Float, Tensor Build ByteString)	(node_ids, gains, feature_dimensions, thresholds, left_node_contribs, right_node_contribs, split_with_default_directions) node_ids gains feature_dimensions thresholds left_node_contribs right_node_contribs split_with_default_directions

boostedTreesCalculateBestFeatureSplit' Source #

Arguments

:: OpParams
-> Int64	logits_dimension
-> Tensor v'1 Int32	node_id_range
-> Tensor v'2 Float	stats_summary
-> 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 Int32, Tensor Build Float, Tensor Build Float, Tensor Build ByteString)	(node_ids, gains, feature_dimensions, thresholds, left_node_contribs, right_node_contribs, split_with_default_directions) node_ids gains feature_dimensions thresholds left_node_contribs right_node_contribs split_with_default_directions

boostedTreesCalculateBestFeatureSplitV2 Source #

Arguments

:: Int64	logits_dimension
-> Tensor v'1 Int32	node_id_range
-> [Tensor v'2 Float]	stats_summaries_list
-> Tensor v'3 ByteString	split_types
-> Tensor v'4 Int32	candidate_feature_ids
-> Tensor v'5 Float	l1
-> Tensor v'6 Float	l2
-> Tensor v'7 Float	tree_complexity
-> Tensor v'8 Float	min_node_weight
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32, Tensor Build Int32, Tensor Build Int32, Tensor Build Float, Tensor Build Float, Tensor Build ByteString)	(node_ids, gains, feature_ids, feature_dimensions, thresholds, left_node_contribs, right_node_contribs, split_with_default_directions) node_ids gains feature_ids feature_dimensions thresholds left_node_contribs right_node_contribs split_with_default_directions

boostedTreesCalculateBestFeatureSplitV2' Source #

Arguments

:: OpParams
-> Int64	logits_dimension
-> Tensor v'1 Int32	node_id_range
-> [Tensor v'2 Float]	stats_summaries_list
-> Tensor v'3 ByteString	split_types
-> Tensor v'4 Int32	candidate_feature_ids
-> Tensor v'5 Float	l1
-> Tensor v'6 Float	l2
-> Tensor v'7 Float	tree_complexity
-> Tensor v'8 Float	min_node_weight
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32, Tensor Build Int32, Tensor Build Int32, Tensor Build Float, Tensor Build Float, Tensor Build ByteString)	(node_ids, gains, feature_ids, feature_dimensions, thresholds, left_node_contribs, right_node_contribs, split_with_default_directions) node_ids gains feature_ids feature_dimensions thresholds left_node_contribs right_node_contribs split_with_default_directions

boostedTreesCalculateBestGainsPerFeature Source #

Arguments

:: 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) node_ids_list gains_list thresholds_list left_node_contribs_list right_node_contribs_list

boostedTreesCalculateBestGainsPerFeature' Source #

Arguments

:: 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) node_ids_list gains_list thresholds_list left_node_contribs_list right_node_contribs_list

boostedTreesCenterBias Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	tree_ensemble_handle
-> Tensor v'2 Float	mean_gradients
-> Tensor v'3 Float	mean_hessians
-> Tensor v'4 Float	l1
-> Tensor v'5 Float	l2
-> m' (Tensor Value Bool)	continue_centering

boostedTreesCenterBias' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	tree_ensemble_handle
-> Tensor v'2 Float	mean_gradients
-> Tensor v'3 Float	mean_hessians
-> Tensor v'4 Float	l1
-> Tensor v'5 Float	l2
-> m' (Tensor Value Bool)	continue_centering

boostedTreesCreateEnsemble Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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 #

Arguments

:: forall v'1 v'2 v'3 m'. 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

boostedTreesCreateQuantileStreamResource Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> Tensor v'2 Float	epsilon
-> Tensor v'3 Int64	num_streams
-> m' ControlNode

boostedTreesCreateQuantileStreamResource' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> Tensor v'2 Float	epsilon
-> Tensor v'3 Int64	num_streams
-> m' ControlNode

boostedTreesDeserializeEnsemble Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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 #

Arguments

:: forall v'1 v'2 v'3 m'. 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 #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	resource

boostedTreesEnsembleResourceHandleOp' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	resource

boostedTreesExampleDebugOutputs Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Int64	logits_dimension
-> Tensor v'1 ResourceHandle	tree_ensemble_handle
-> [Tensor v'2 Int32]	bucketized_features
-> m' (Tensor Value ByteString)	examples_debug_outputs_serialized

boostedTreesExampleDebugOutputs' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Int64	logits_dimension
-> Tensor v'1 ResourceHandle	tree_ensemble_handle
-> [Tensor v'2 Int32]	bucketized_features
-> m' (Tensor Value ByteString)	examples_debug_outputs_serialized

boostedTreesFlushQuantileSummaries Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Int64	num_features
-> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> m' [Tensor Value Float]	summaries

boostedTreesFlushQuantileSummaries' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Int64	num_features
-> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> m' [Tensor Value Float]	summaries

boostedTreesGetEnsembleStates Source #

Arguments

:: forall v'1 m'. 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)

stamp_token
num_trees
num_finalized_trees
num_attempted_layers
last_layer_nodes_range

boostedTreesGetEnsembleStates' Source #

Arguments

:: forall v'1 m'. 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) stamp_token num_trees num_finalized_trees num_attempted_layers last_layer_nodes_range

boostedTreesMakeQuantileSummaries Source #

Arguments

:: [Tensor v'1 Float]	float_values
-> Tensor v'2 Float	example_weights
-> Tensor v'3 Float	epsilon
-> [Tensor Build Float]	summaries

boostedTreesMakeQuantileSummaries' Source #

Arguments

:: OpParams
-> [Tensor v'1 Float]	float_values
-> Tensor v'2 Float	example_weights
-> Tensor v'3 Float	epsilon
-> [Tensor Build Float]	summaries

boostedTreesMakeStatsSummary Source #

Arguments

:: Int64	max_splits
-> Int64	num_buckets
-> Tensor v'1 Int32	node_ids
-> Tensor v'2 Float	gradients
-> Tensor v'3 Float	hessians
-> [Tensor v'4 Int32]	bucketized_features_list
-> Tensor Build Float	stats_summary

boostedTreesMakeStatsSummary' Source #

Arguments

:: OpParams
-> Int64	max_splits
-> Int64	num_buckets
-> Tensor v'1 Int32	node_ids
-> Tensor v'2 Float	gradients
-> Tensor v'3 Float	hessians
-> [Tensor v'4 Int32]	bucketized_features_list
-> Tensor Build Float	stats_summary

boostedTreesPredict Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Int64	logits_dimension
-> Tensor v'1 ResourceHandle	tree_ensemble_handle
-> [Tensor v'2 Int32]	bucketized_features
-> m' (Tensor Value Float)	logits

boostedTreesPredict' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Int64	logits_dimension
-> Tensor v'1 ResourceHandle	tree_ensemble_handle
-> [Tensor v'2 Int32]	bucketized_features
-> m' (Tensor Value Float)	logits

boostedTreesQuantileStreamResourceAddSummaries Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> [Tensor v'2 Float]	summaries
-> m' ControlNode

boostedTreesQuantileStreamResourceAddSummaries' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> [Tensor v'2 Float]	summaries
-> m' ControlNode

boostedTreesQuantileStreamResourceDeserialize Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> [Tensor v'2 Float]	bucket_boundaries
-> m' ControlNode

boostedTreesQuantileStreamResourceDeserialize' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> [Tensor v'2 Float]	bucket_boundaries
-> m' ControlNode

boostedTreesQuantileStreamResourceFlush Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> Tensor v'2 Int64	num_buckets
-> m' ControlNode

boostedTreesQuantileStreamResourceFlush' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> Tensor v'2 Int64	num_buckets
-> m' ControlNode

boostedTreesQuantileStreamResourceGetBucketBoundaries Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Int64	num_features
-> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> m' [Tensor Value Float]	bucket_boundaries

boostedTreesQuantileStreamResourceGetBucketBoundaries' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Int64	num_features
-> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> m' [Tensor Value Float]	bucket_boundaries

boostedTreesQuantileStreamResourceHandleOp Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	resource

boostedTreesQuantileStreamResourceHandleOp' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	resource

boostedTreesSerializeEnsemble Source #

Arguments

:: forall v'1 m'. MonadBuild m'

=> Tensor v'1 ResourceHandle

tree_ensemble_handle

-> m' (Tensor Value Int64, Tensor Value ByteString)

(stamp_token, tree_ensemble_serialized)

stamp_token
tree_ensemble_serialized

boostedTreesSerializeEnsemble' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	tree_ensemble_handle
-> m' (Tensor Value Int64, Tensor Value ByteString)	(stamp_token, tree_ensemble_serialized) stamp_token tree_ensemble_serialized

boostedTreesSparseAggregateStats Source #

Arguments

:: Int64	max_splits
-> Int64	num_buckets
-> Tensor v'1 Int32	node_ids
-> Tensor v'2 Float	gradients
-> Tensor v'3 Float	hessians
-> Tensor v'4 Int32	feature_indices
-> Tensor v'5 Int32	feature_values
-> Tensor v'6 Int32	feature_shape
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32)	(stats_summary_indices, stats_summary_values, stats_summary_shape) stats_summary_indices stats_summary_values stats_summary_shape

boostedTreesSparseAggregateStats' Source #

Arguments

:: OpParams
-> Int64	max_splits
-> Int64	num_buckets
-> Tensor v'1 Int32	node_ids
-> Tensor v'2 Float	gradients
-> Tensor v'3 Float	hessians
-> Tensor v'4 Int32	feature_indices
-> Tensor v'5 Int32	feature_values
-> Tensor v'6 Int32	feature_shape
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32)	(stats_summary_indices, stats_summary_values, stats_summary_shape) stats_summary_indices stats_summary_values stats_summary_shape

boostedTreesSparseCalculateBestFeatureSplit Source #

Arguments

:: Int64	logits_dimension
-> Tensor v'1 Int32	node_id_range
-> Tensor v'2 Int32	stats_summary_indices
-> Tensor v'3 Float	stats_summary_values
-> Tensor v'4 Int32	stats_summary_shape
-> Tensor v'5 Float	l1
-> Tensor v'6 Float	l2
-> Tensor v'7 Float	tree_complexity
-> Tensor v'8 Float	min_node_weight
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32, Tensor Build Int32, Tensor Build Float, Tensor Build Float, Tensor Build ByteString)	(node_ids, gains, feature_dimensions, thresholds, left_node_contribs, right_node_contribs, split_with_default_directions) node_ids gains feature_dimensions thresholds left_node_contribs right_node_contribs split_with_default_directions

boostedTreesSparseCalculateBestFeatureSplit' Source #

Arguments

:: OpParams
-> Int64	logits_dimension
-> Tensor v'1 Int32	node_id_range
-> Tensor v'2 Int32	stats_summary_indices
-> Tensor v'3 Float	stats_summary_values
-> Tensor v'4 Int32	stats_summary_shape
-> Tensor v'5 Float	l1
-> Tensor v'6 Float	l2
-> Tensor v'7 Float	tree_complexity
-> Tensor v'8 Float	min_node_weight
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32, Tensor Build Int32, Tensor Build Float, Tensor Build Float, Tensor Build ByteString)	(node_ids, gains, feature_dimensions, thresholds, left_node_contribs, right_node_contribs, split_with_default_directions) node_ids gains feature_dimensions thresholds left_node_contribs right_node_contribs split_with_default_directions

boostedTreesTrainingPredict Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. 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) partial_logits tree_ids node_ids

boostedTreesTrainingPredict' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. 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) partial_logits tree_ids node_ids

boostedTreesUpdateEnsemble Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 m'. 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 #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 m'. 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

boostedTreesUpdateEnsembleV2 Source #

Arguments

:: 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 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	tree_ensemble_handle
-> [Tensor v'2 Int32]	feature_ids
-> [Tensor v'3 Int32]	dimension_ids
-> [Tensor v'4 Int32]	node_ids
-> [Tensor v'5 Float]	gains
-> [Tensor v'6 Int32]	thresholds
-> [Tensor v'7 Float]	left_node_contribs
-> [Tensor v'8 Float]	right_node_contribs
-> [Tensor v'9 ByteString]	split_types
-> Tensor v'10 Int32	max_depth
-> Tensor v'11 Float	learning_rate
-> Tensor v'12 Int32	pruning_mode
-> m' ControlNode

boostedTreesUpdateEnsembleV2' Source #

Arguments

:: 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 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	tree_ensemble_handle
-> [Tensor v'2 Int32]	feature_ids
-> [Tensor v'3 Int32]	dimension_ids
-> [Tensor v'4 Int32]	node_ids
-> [Tensor v'5 Float]	gains
-> [Tensor v'6 Int32]	thresholds
-> [Tensor v'7 Float]	left_node_contribs
-> [Tensor v'8 Float]	right_node_contribs
-> [Tensor v'9 ByteString]	split_types
-> Tensor v'10 Int32	max_depth
-> Tensor v'11 Float	learning_rate
-> Tensor v'12 Int32	pruning_mode
-> m' ControlNode

broadcastArgs Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64] t
=> Tensor v'1 t	s0
-> Tensor v'2 t	s1
-> Tensor Build t	r0

broadcastArgs' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64] t
=> OpParams
-> Tensor v'1 t	s0
-> Tensor v'2 t	s1
-> Tensor Build t	r0

broadcastGradientArgs Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64] t
=> Tensor v'1 t	s0
-> Tensor v'2 t	s1
-> (Tensor Build t, Tensor Build t)	(r0, r1) r0 r1

broadcastGradientArgs' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64] t
=> OpParams
-> Tensor v'1 t	s0
-> Tensor v'2 t	s1
-> (Tensor Build t, Tensor Build t)	(r0, r1) r0 r1

broadcastTo Source #

Arguments

:: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx)
=> Tensor v'1 t	input
-> Tensor v'2 tidx	shape
-> Tensor Build t	output

broadcastTo' Source #

Arguments

:: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx)
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 tidx	shape
-> Tensor Build t	output

bucketize Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build Int32	output

bucketize' Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build Int32	output

bytesProducedStatsDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	tag
-> Tensor Build Variant	handle

bytesProducedStatsDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	tag
-> Tensor Build Variant	handle

cSRSparseMatrixComponents Source #

Arguments

:: forall v'1 v'2 type'. OneOf '[Complex Double, Complex Float, Double, Float] type'
=> Tensor v'1 Variant	csr_sparse_matrix
-> Tensor v'2 Int32	index
-> (Tensor Build Int32, Tensor Build Int32, Tensor Build type')	(row_ptrs, col_inds, values) row_ptrs col_inds values

cSRSparseMatrixComponents' Source #

Arguments

:: forall v'1 v'2 type'. OneOf '[Complex Double, Complex Float, Double, Float] type'
=> OpParams
-> Tensor v'1 Variant	csr_sparse_matrix
-> Tensor v'2 Int32	index
-> (Tensor Build Int32, Tensor Build Int32, Tensor Build type')	(row_ptrs, col_inds, values) row_ptrs col_inds values

cSRSparseMatrixToDense Source #

Arguments

:: forall v'1 type'. OneOf '[Complex Double, Complex Float, Double, Float] type'
=> Tensor v'1 Variant	sparse_input
-> Tensor Build type'	dense_output

cSRSparseMatrixToDense' Source #

Arguments

:: forall v'1 type'. OneOf '[Complex Double, Complex Float, Double, Float] type'
=> OpParams
-> Tensor v'1 Variant	sparse_input
-> Tensor Build type'	dense_output

cSRSparseMatrixToSparseTensor Source #

Arguments

:: forall v'1 type'. OneOf '[Complex Double, Complex Float, Double, Float] type'

=> Tensor v'1 Variant

sparse_matrix

-> (Tensor Build Int64, Tensor Build type', Tensor Build Int64)

(indices, values, dense_shape)

indices
values
dense_shape

cSRSparseMatrixToSparseTensor' Source #

Arguments

:: forall v'1 type'. OneOf '[Complex Double, Complex Float, Double, Float] type'
=> OpParams
-> Tensor v'1 Variant	sparse_matrix
-> (Tensor Build Int64, Tensor Build type', Tensor Build Int64)	(indices, values, dense_shape) indices values dense_shape

cSVDataset Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 output_types m'. (MonadBuild m', OneOfs '[ByteString, Int32, Int64, Double, Float] output_types)
=> Tensor v'1 ByteString	filenames
-> Tensor v'2 ByteString	compression_type
-> Tensor v'3 Int64	buffer_size
-> Tensor v'4 Bool	header
-> Tensor v'5 ByteString	field_delim
-> Tensor v'6 Bool	use_quote_delim
-> Tensor v'7 ByteString	na_value
-> Tensor v'8 Int64	select_cols
-> TensorList v'9 output_types	record_defaults
-> m' (Tensor Value Variant)	handle

cSVDataset' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 output_types m'. (MonadBuild m', OneOfs '[ByteString, Int32, Int64, Double, Float] output_types)
=> OpParams
-> Tensor v'1 ByteString	filenames
-> Tensor v'2 ByteString	compression_type
-> Tensor v'3 Int64	buffer_size
-> Tensor v'4 Bool	header
-> Tensor v'5 ByteString	field_delim
-> Tensor v'6 Bool	use_quote_delim
-> Tensor v'7 ByteString	na_value
-> Tensor v'8 Int64	select_cols
-> TensorList v'9 output_types	record_defaults
-> m' (Tensor Value Variant)	handle

cTCBeamSearchDecoder Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Int64	beam_width
-> Int64	top_paths
-> Tensor v'1 t	inputs
-> Tensor v'2 Int32	sequence_length
-> ([Tensor Build Int64], [Tensor Build Int64], [Tensor Build Int64], Tensor Build t)	(decoded_indices, decoded_values, decoded_shape, log_probability) decoded_indices decoded_values decoded_shape log_probability

cTCBeamSearchDecoder' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Int64	beam_width
-> Int64	top_paths
-> Tensor v'1 t	inputs
-> Tensor v'2 Int32	sequence_length
-> ([Tensor Build Int64], [Tensor Build Int64], [Tensor Build Int64], Tensor Build t)	(decoded_indices, decoded_values, decoded_shape, log_probability) decoded_indices decoded_values decoded_shape log_probability

cTCGreedyDecoder Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 t	inputs
-> Tensor v'2 Int32	sequence_length
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64, Tensor Build t)	(decoded_indices, decoded_values, decoded_shape, log_probability) decoded_indices decoded_values decoded_shape log_probability

cTCGreedyDecoder' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	inputs
-> Tensor v'2 Int32	sequence_length
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64, Tensor Build t)	(decoded_indices, decoded_values, decoded_shape, log_probability) decoded_indices decoded_values decoded_shape log_probability

cTCLoss Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Double, Float] t
=> Tensor v'1 t	inputs
-> Tensor v'2 Int64	labels_indices
-> Tensor v'3 Int32	labels_values
-> Tensor v'4 Int32	sequence_length
-> (Tensor Build t, Tensor Build t)	(loss, gradient) loss gradient

cTCLoss' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	inputs
-> Tensor v'2 Int64	labels_indices
-> Tensor v'3 Int32	labels_values
-> Tensor v'4 Int32	sequence_length
-> (Tensor Build t, Tensor Build t)	(loss, gradient) loss gradient

cTCLossV2 Source #

Arguments

:: Tensor v'1 Float	inputs
-> Tensor v'2 Int64	labels_indices
-> Tensor v'3 Int32	labels_values
-> Tensor v'4 Int32	sequence_length
-> (Tensor Build Float, Tensor Build Float)	(loss, gradient) loss gradient

cTCLossV2' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	inputs
-> Tensor v'2 Int64	labels_indices
-> Tensor v'3 Int32	labels_values
-> Tensor v'4 Int32	sequence_length
-> (Tensor Build Float, Tensor Build Float)	(loss, gradient) loss gradient

cacheDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	filename
-> Tensor Build Variant	handle

cacheDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	filename
-> Tensor Build Variant	handle

cacheDatasetV2 Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	filename
-> Tensor v'3 ResourceHandle	cache
-> m' (Tensor Value Variant)	handle

cacheDatasetV2' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	filename
-> Tensor v'3 ResourceHandle	cache
-> m' (Tensor Value Variant)	handle

cast Source #

Arguments

:: forall v'1 srcT dstT. (TensorType srcT, TensorType dstT)
=> Tensor v'1 srcT	x
-> Tensor Build dstT	y

cast' Source #

Arguments

:: forall v'1 srcT dstT. (TensorType srcT, TensorType dstT)
=> OpParams
-> Tensor v'1 srcT	x
-> Tensor Build dstT	y

ceil Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

ceil' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

checkNumerics Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t)
=> ByteString	message
-> Tensor v'1 t	tensor
-> m' (Tensor Value t)	output

checkNumerics' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t)
=> OpParams
-> ByteString	message
-> Tensor v'1 t	tensor
-> m' (Tensor Value t)	output

checkNumericsV2 Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t)
=> ByteString	message
-> Tensor v'1 t	tensor
-> m' (Tensor Value t)	output

checkNumericsV2' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t)
=> OpParams
-> ByteString	message
-> Tensor v'1 t	tensor
-> m' (Tensor Value t)	output

cholesky Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

cholesky' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

choleskyGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	l
-> Tensor v'2 t	grad
-> Tensor Build t	output

choleskyGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	l
-> Tensor v'2 t	grad
-> Tensor Build t	output

chooseFastestDataset Source #

Arguments

:: Int64	num_experiments
-> [DataType]	output_types
-> [Tensor v'1 Variant]	input_datasets
-> Tensor Build Variant	handle

chooseFastestDataset' Source #

Arguments

:: OpParams
-> Int64	num_experiments
-> [DataType]	output_types
-> [Tensor v'1 Variant]	input_datasets
-> Tensor Build Variant	handle

clipByValue Source #

Arguments

:: 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	t
-> Tensor v'2 t	clip_value_min
-> Tensor v'3 t	clip_value_max
-> Tensor Build t	output

clipByValue' Source #

Arguments

:: 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	t
-> Tensor v'2 t	clip_value_min
-> Tensor v'3 t	clip_value_max
-> Tensor Build t	output

closeSummaryWriter Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	writer
-> m' ControlNode

closeSummaryWriter' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	writer
-> m' ControlNode

collectiveBcastRecv Source #

Arguments

:: forall t m'. (MonadBuild m', OneOf '[Bool, Int32, Int64, Word16, Double, Float] t)
=> Int64	group_key
-> Int64	group_size
-> Int64	instance_key
-> Shape	shape
-> m' (Tensor Value t)	data

collectiveBcastRecv' Source #

Arguments

:: forall t m'. (MonadBuild m', OneOf '[Bool, Int32, Int64, Word16, Double, Float] t)
=> OpParams
-> Int64	group_key
-> Int64	group_size
-> Int64	instance_key
-> Shape	shape
-> m' (Tensor Value t)	data

collectiveBcastSend Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Bool, Int32, Int64, Word16, Double, Float] t)
=> Int64	group_key
-> Int64	group_size
-> Int64	instance_key
-> Shape	shape
-> Tensor v'1 t	input
-> m' (Tensor Value t)	data

collectiveBcastSend' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Bool, Int32, Int64, Word16, Double, Float] t)
=> OpParams
-> Int64	group_key
-> Int64	group_size
-> Int64	instance_key
-> Shape	shape
-> Tensor v'1 t	input
-> m' (Tensor Value t)	data

collectiveGather Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> Int64	group_key
-> Int64	group_size
-> Int64	instance_key
-> Shape	shape
-> Tensor v'1 t	input
-> m' (Tensor Value t)	data

collectiveGather' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> OpParams
-> Int64	group_key
-> Int64	group_size
-> Int64	instance_key
-> Shape	shape
-> Tensor v'1 t	input
-> m' (Tensor Value t)	data

collectivePermute Source #

Arguments

:: 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	input
-> Tensor v'2 Int32	source_target_pairs
-> Tensor Build t	output

collectivePermute' Source #

Arguments

:: 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	input
-> Tensor v'2 Int32	source_target_pairs
-> Tensor Build t	output

collectiveReduce Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> ByteString	final_op
-> Int64	group_key
-> Int64	group_size
-> Int64	instance_key
-> ByteString	merge_op
-> Tensor v'1 t	input
-> m' (Tensor Value t)	data

collectiveReduce' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> OpParams
-> ByteString	final_op
-> Int64	group_key
-> Int64	group_size
-> Int64	instance_key
-> ByteString	merge_op
-> Tensor v'1 t	input
-> m' (Tensor Value t)	data

combinedNonMaxSuppression Source #

Arguments

:: Tensor v'1 Float	boxes
-> Tensor v'2 Float	scores
-> Tensor v'3 Int32	max_output_size_per_class
-> Tensor v'4 Int32	max_total_size
-> Tensor v'5 Float	iou_threshold
-> Tensor v'6 Float	score_threshold
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float, Tensor Build Int32)	(nmsed_boxes, nmsed_scores, nmsed_classes, valid_detections) nmsed_boxes nmsed_scores nmsed_classes valid_detections

combinedNonMaxSuppression' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	boxes
-> Tensor v'2 Float	scores
-> Tensor v'3 Int32	max_output_size_per_class
-> Tensor v'4 Int32	max_total_size
-> Tensor v'5 Float	iou_threshold
-> Tensor v'6 Float	score_threshold
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float, Tensor Build Int32)	(nmsed_boxes, nmsed_scores, nmsed_classes, valid_detections) nmsed_boxes nmsed_scores nmsed_classes valid_detections

compareAndBitpack Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Bool, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> Tensor v'1 t	input
-> Tensor v'2 t	threshold
-> Tensor Build Word8	output

compareAndBitpack' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Bool, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 t	threshold
-> Tensor Build Word8	output

complex Source #

Arguments

:: forall v'1 v'2 t tout. (OneOf '[Double, Float] t, OneOf '[Complex Double, Complex Float] tout)
=> Tensor v'1 t	real
-> Tensor v'2 t	imag
-> Tensor Build tout	out

complex' Source #

Arguments

:: forall v'1 v'2 t tout. (OneOf '[Double, Float] t, OneOf '[Complex Double, Complex Float] tout)
=> OpParams
-> Tensor v'1 t	real
-> Tensor v'2 t	imag
-> Tensor Build tout	out

complexAbs Source #

Arguments

:: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout)
=> Tensor v'1 t	x
-> Tensor Build tout	y

complexAbs' Source #

Arguments

:: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout)
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build tout	y

compressElement Source #

Arguments

:: forall v'1 input_types. TensorTypes input_types
=> TensorList v'1 input_types	components
-> Tensor Build Variant	compressed

compressElement' Source #

Arguments

:: forall v'1 input_types. TensorTypes input_types
=> OpParams
-> TensorList v'1 input_types	components
-> Tensor Build Variant	compressed

computeAccidentalHits Source #

Arguments

:: Int64	num_true
-> Tensor v'1 Int64	true_classes
-> Tensor v'2 Int64	sampled_candidates
-> (Tensor Build Int32, Tensor Build Int64, Tensor Build Float)	(indices, ids, weights) indices ids weights

computeAccidentalHits' Source #

Arguments

:: OpParams
-> Int64	num_true
-> Tensor v'1 Int64	true_classes
-> Tensor v'2 Int64	sampled_candidates
-> (Tensor Build Int32, Tensor Build Int64, Tensor Build Float)	(indices, ids, weights) indices ids weights

concat Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Tensor v'1 Int32	concat_dim
-> [Tensor v'2 t]	values
-> Tensor Build t	output

concat' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Tensor v'1 Int32	concat_dim
-> [Tensor v'2 t]	values
-> Tensor Build t	output

concatOffset Source #

Arguments

:: Tensor v'1 Int32	concat_dim
-> [Tensor v'2 Int32]	shape
-> [Tensor Build Int32]	offset

concatOffset' Source #

Arguments

:: OpParams
-> Tensor v'1 Int32	concat_dim
-> [Tensor v'2 Int32]	shape
-> [Tensor Build Int32]	offset

concatV2 Source #

Arguments

:: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx)
=> [Tensor v'1 t]	values
-> Tensor v'2 tidx	axis
-> Tensor Build t	output

concatV2' Source #

Arguments

:: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx)
=> OpParams
-> [Tensor v'1 t]	values
-> Tensor v'2 tidx	axis
-> Tensor Build t	output

concatenateDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Variant	another_dataset
-> Tensor Build Variant	handle

concatenateDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Variant	another_dataset
-> Tensor Build Variant	handle

conditionalAccumulator Source #

Arguments

:: forall m'. MonadBuild m'
=> DataType	dtype
-> Shape	shape
-> m' (Tensor Ref ByteString)	handle

conditionalAccumulator' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> DataType	dtype
-> Shape	shape
-> m' (Tensor Ref ByteString)	handle

configureDistributedTPU Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ByteString)	topology

configureDistributedTPU' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ByteString)	topology

configureTPUEmbedding Source #

Arguments

:: forall m'. MonadBuild m'
=> ByteString	config
-> m' ControlNode

configureTPUEmbedding' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> ByteString	config
-> m' ControlNode

conj Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Variant] t
=> Tensor v'1 t	input
-> Tensor Build t	output

conj' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Variant] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

conjugateTranspose Source #

Arguments

:: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm)
=> Tensor v'1 t	x
-> Tensor v'2 tperm	perm
-> Tensor Build t	y

conjugateTranspose' Source #

Arguments

:: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm)
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 tperm	perm
-> Tensor Build t	y

const Source #

Arguments

:: forall dtype. TensorType dtype
=> Tensor Build dtype	output

const' Source #

Arguments

:: forall dtype. TensorType dtype
=> OpParams
-> Tensor Build dtype	output

consumeMutexLock Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 Variant	mutex_lock
-> m' ControlNode

consumeMutexLock' Source #

Arguments

:: forall v'1 m'. 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 #

conv2D Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor Build t	output

conv2D' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor Build t	output

conv2DBackpropFilter Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	filter_sizes
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

conv2DBackpropFilter' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	filter_sizes
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

conv2DBackpropInput Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int32, Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 Int32	input_sizes
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

conv2DBackpropInput' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int32, Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 Int32	input_sizes
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

conv3D Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor Build t	output

conv3D' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor Build t	output

conv3DBackpropFilter Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

conv3DBackpropFilter' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

conv3DBackpropFilterV2 Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	filter_sizes
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

conv3DBackpropFilterV2' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	filter_sizes
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

conv3DBackpropInput Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

conv3DBackpropInput' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

conv3DBackpropInputV2 Source #

Arguments

:: forall v'1 v'2 v'3 t tshape. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tshape)
=> ByteString	padding
-> Tensor v'1 tshape	input_sizes
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

conv3DBackpropInputV2' Source #

Arguments

:: forall v'1 v'2 v'3 t tshape. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tshape)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tshape	input_sizes
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

copy Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	output

copy' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

copyHost Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	output

copyHost' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

cos Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

cos' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

cosh Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

cosh' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

countUpTo Source #

Arguments

:: forall t m'. (MonadBuild m', OneOf '[Int32, Int64] t)
=> Int64	limit
-> Tensor Ref t	ref
-> m' (Tensor Value t)	output

countUpTo' Source #

Arguments

:: forall t m'. (MonadBuild m', OneOf '[Int32, Int64] t)
=> OpParams
-> Int64	limit
-> Tensor Ref t	ref
-> m' (Tensor Value t)	output

createSummaryDbWriter Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. 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 #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. 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 #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. 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 #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. 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

cropAndResize Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> Tensor v'1 t	image
-> Tensor v'2 Float	boxes
-> Tensor v'3 Int32	box_ind
-> Tensor v'4 Int32	crop_size
-> Tensor Build Float	crops

cropAndResize' Source #

Arguments

:: 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	image
-> Tensor v'2 Float	boxes
-> Tensor v'3 Int32	box_ind
-> Tensor v'4 Int32	crop_size
-> Tensor Build Float	crops

cropAndResizeGradBoxes Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> Tensor v'1 Float	grads
-> Tensor v'2 t	image
-> Tensor v'3 Float	boxes
-> Tensor v'4 Int32	box_ind
-> Tensor Build Float	output

cropAndResizeGradBoxes' Source #

Arguments

:: 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	grads
-> Tensor v'2 t	image
-> Tensor v'3 Float	boxes
-> Tensor v'4 Int32	box_ind
-> Tensor Build Float	output

cropAndResizeGradImage Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 Float	grads
-> Tensor v'2 Float	boxes
-> Tensor v'3 Int32	box_ind
-> Tensor v'4 Int32	image_size
-> Tensor Build t	output

cropAndResizeGradImage' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 Float	grads
-> Tensor v'2 Float	boxes
-> Tensor v'3 Int32	box_ind
-> Tensor v'4 Int32	image_size
-> Tensor Build t	output

cross Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	a
-> Tensor v'2 t	b
-> Tensor Build t	product

cross' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	a
-> Tensor v'2 t	b
-> Tensor Build t	product

crossReplicaSum Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Word16, Word32, Float] t
=> Tensor v'1 t	input
-> Tensor v'2 Int32	group_assignment
-> Tensor Build t	output

crossReplicaSum' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Word16, Word32, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 Int32	group_assignment
-> Tensor Build t	output

cudnnRNN Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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)	(output, output_h, output_c, reserve_space) output output_h output_c reserve_space

cudnnRNN' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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)	(output, output_h, output_c, reserve_space) output output_h output_c reserve_space

cudnnRNNBackprop Source #

Arguments

:: 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	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) input_backprop input_h_backprop input_c_backprop params_backprop

cudnnRNNBackprop' Source #

Arguments

:: 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	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) input_backprop input_h_backprop input_c_backprop params_backprop

cudnnRNNBackpropV2 Source #

Arguments

:: 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	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) input_backprop input_h_backprop input_c_backprop params_backprop

cudnnRNNBackpropV2' Source #

Arguments

:: 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	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) input_backprop input_h_backprop input_c_backprop params_backprop

cudnnRNNBackpropV3 Source #

Arguments

:: 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 t m'. (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 Int32	sequence_lengths
-> Tensor v'6 t	output
-> Tensor v'7 t	output_h
-> Tensor v'8 t	output_c
-> Tensor v'9 t	output_backprop
-> Tensor v'10 t	output_h_backprop
-> Tensor v'11 t	output_c_backprop
-> Tensor v'12 t	reserve_space
-> Tensor v'13 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) input_backprop input_h_backprop input_c_backprop params_backprop

cudnnRNNBackpropV3' Source #

Arguments

:: 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 t m'. (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 Int32	sequence_lengths
-> Tensor v'6 t	output
-> Tensor v'7 t	output_h
-> Tensor v'8 t	output_c
-> Tensor v'9 t	output_backprop
-> Tensor v'10 t	output_h_backprop
-> Tensor v'11 t	output_c_backprop
-> Tensor v'12 t	reserve_space
-> Tensor v'13 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) input_backprop input_h_backprop input_c_backprop params_backprop

cudnnRNNCanonicalToParams Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 Int32	num_layers
-> Tensor v'2 Int32	num_units
-> Tensor v'3 Int32	input_size
-> [Tensor v'4 t]	weights
-> [Tensor v'5 t]	biases
-> Tensor Build t	params

cudnnRNNCanonicalToParams' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 Int32	num_layers
-> Tensor v'2 Int32	num_units
-> Tensor v'3 Int32	input_size
-> [Tensor v'4 t]	weights
-> [Tensor v'5 t]	biases
-> Tensor Build t	params

cudnnRNNCanonicalToParamsV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 Int32	num_layers
-> Tensor v'2 Int32	num_units
-> Tensor v'3 Int32	input_size
-> [Tensor v'4 t]	weights
-> [Tensor v'5 t]	biases
-> Tensor Build t	params

cudnnRNNCanonicalToParamsV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 Int32	num_layers
-> Tensor v'2 Int32	num_units
-> Tensor v'3 Int32	input_size
-> [Tensor v'4 t]	weights
-> [Tensor v'5 t]	biases
-> Tensor Build t	params

cudnnRNNParamsSize Source #

Arguments

:: forall v'1 v'2 v'3 s. OneOf '[Int32, Int64] s
=> DataType	T
-> Tensor v'1 Int32	num_layers
-> Tensor v'2 Int32	num_units
-> Tensor v'3 Int32	input_size
-> Tensor Build s	params_size

cudnnRNNParamsSize' Source #

Arguments

:: forall v'1 v'2 v'3 s. OneOf '[Int32, Int64] s
=> OpParams
-> DataType	T
-> Tensor v'1 Int32	num_layers
-> Tensor v'2 Int32	num_units
-> Tensor v'3 Int32	input_size
-> Tensor Build s	params_size

cudnnRNNParamsToCanonical Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t
=> Int64	num_params
-> Tensor v'1 Int32	num_layers
-> Tensor v'2 Int32	num_units
-> Tensor v'3 Int32	input_size
-> Tensor v'4 t	params
-> ([Tensor Build t], [Tensor Build t])	(weights, biases) weights biases

cudnnRNNParamsToCanonical' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Int64	num_params
-> Tensor v'1 Int32	num_layers
-> Tensor v'2 Int32	num_units
-> Tensor v'3 Int32	input_size
-> Tensor v'4 t	params
-> ([Tensor Build t], [Tensor Build t])	(weights, biases) weights biases

cudnnRNNParamsToCanonicalV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t
=> Int64	num_params_biases
-> Int64	num_params_weights
-> Tensor v'1 Int32	num_layers
-> Tensor v'2 Int32	num_units
-> Tensor v'3 Int32	input_size
-> Tensor v'4 t	params
-> ([Tensor Build t], [Tensor Build t])	(weights, biases) weights biases

cudnnRNNParamsToCanonicalV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Int64	num_params_biases
-> Int64	num_params_weights
-> Tensor v'1 Int32	num_layers
-> Tensor v'2 Int32	num_units
-> Tensor v'3 Int32	input_size
-> Tensor v'4 t	params
-> ([Tensor Build t], [Tensor Build t])	(weights, biases) weights biases

cudnnRNNV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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) output output_h output_c reserve_space host_reserved

cudnnRNNV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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) output output_h output_c reserve_space host_reserved

cudnnRNNV3 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t m'. (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 Int32	sequence_lengths
-> 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) output output_h output_c reserve_space host_reserved

cudnnRNNV3' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t m'. (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 Int32	sequence_lengths
-> 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) output output_h output_c reserve_space host_reserved

cumprod Source #

Arguments

:: 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	x
-> Tensor v'2 tidx	axis
-> Tensor Build t	out

cumprod' Source #

Arguments

:: 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	x
-> Tensor v'2 tidx	axis
-> Tensor Build t	out

cumsum Source #

Arguments

:: 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	x
-> Tensor v'2 tidx	axis
-> Tensor Build t	out

cumsum' Source #

Arguments

:: 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	x
-> Tensor v'2 tidx	axis
-> Tensor Build t	out

cumulativeLogsumexp Source #

Arguments

:: forall v'1 v'2 t tidx. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx)
=> Tensor v'1 t	x
-> Tensor v'2 tidx	axis
-> Tensor Build t	out

cumulativeLogsumexp' Source #

Arguments

:: forall v'1 v'2 t tidx. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx)
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 tidx	axis
-> Tensor Build t	out

dataFormatDimMap Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64] t
=> Tensor v'1 t	x
-> Tensor Build t	y

dataFormatDimMap' Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

dataFormatVecPermute Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64] t
=> Tensor v'1 t	x
-> Tensor Build t	y

dataFormatVecPermute' Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

dataServiceDataset Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Int64	dataset_id
-> Tensor v'2 ByteString	processing_mode
-> Tensor v'3 ByteString	address
-> Tensor v'4 ByteString	protocol
-> Tensor v'5 ByteString	job_name
-> Tensor v'6 Int64	max_outstanding_requests
-> Tensor v'7 ResourceHandle	iteration_counter
-> m' (Tensor Value Variant)	handle

dataServiceDataset' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Int64	dataset_id
-> Tensor v'2 ByteString	processing_mode
-> Tensor v'3 ByteString	address
-> Tensor v'4 ByteString	protocol
-> Tensor v'5 ByteString	job_name
-> Tensor v'6 Int64	max_outstanding_requests
-> Tensor v'7 ResourceHandle	iteration_counter
-> m' (Tensor Value Variant)	handle

datasetCardinality Source #

Arguments

:: Tensor v'1 Variant	input_dataset
-> Tensor Build Int64	cardinality

datasetCardinality' Source #

Arguments

:: OpParams
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Int64	cardinality

datasetFromGraph Source #

Arguments

:: Tensor v'1 ByteString	graph_def
-> Tensor Build Variant	handle

datasetFromGraph' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	graph_def
-> Tensor Build Variant	handle

datasetToGraph Source #

Arguments

:: Tensor v'1 Variant	input_dataset
-> Tensor Build ByteString	graph

datasetToGraph' Source #

Arguments

:: OpParams
-> Tensor v'1 Variant	input_dataset
-> Tensor Build ByteString	graph

datasetToGraphV2 Source #

Arguments

:: Tensor v'1 Variant	input_dataset
-> Tensor Build ByteString	graph

datasetToGraphV2' Source #

Arguments

:: OpParams
-> Tensor v'1 Variant	input_dataset
-> Tensor Build ByteString	graph

datasetToSingleElement Source #

Arguments

:: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types)
=> Tensor v'1 Variant	dataset
-> m' (TensorList Value output_types)	components

datasetToSingleElement' Source #

Arguments

:: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types)
=> OpParams
-> Tensor v'1 Variant	dataset
-> m' (TensorList Value output_types)	components

datasetToTFRecord Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	filename
-> Tensor v'3 ByteString	compression_type
-> m' ControlNode

datasetToTFRecord' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	filename
-> Tensor v'3 ByteString	compression_type
-> m' ControlNode

dawsn Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

dawsn' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

debugGradientIdentity Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	output

debugGradientIdentity' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

debugGradientRefIdentity Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> Tensor Ref t	input
-> m' (Tensor Ref t)	output

debugGradientRefIdentity' Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor Ref t	input
-> m' (Tensor Ref t)	output

debugIdentity Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	output

debugIdentity' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

debugIdentityV2 Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> Tensor v'1 t	input
-> m' (Tensor Value t)	output

debugIdentityV2' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor v'1 t	input
-> m' (Tensor Value t)	output

debugNanCount Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build Int64	output

debugNanCount' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build Int64	output

debugNumericSummary Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build Double	output

debugNumericSummary' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build Double	output

debugNumericSummaryV2 Source #

Arguments

:: forall v'1 output_dtype t. (OneOf '[Double, Float] output_dtype, TensorType t)
=> Tensor v'1 t	input
-> Tensor Build output_dtype	output

debugNumericSummaryV2' Source #

Arguments

:: forall v'1 output_dtype t. (OneOf '[Double, Float] output_dtype, TensorType t)
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build output_dtype	output

decodeAndCropJpeg Source #

Arguments

:: Tensor v'1 ByteString	contents
-> Tensor v'2 Int32	crop_window
-> Tensor Build Word8	image

decodeAndCropJpeg' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	contents
-> Tensor v'2 Int32	crop_window
-> Tensor Build Word8	image

decodeBase64 Source #

Arguments

:: Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

decodeBase64' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

decodeBmp Source #

Arguments

:: Tensor v'1 ByteString	contents
-> Tensor Build Word8	image

decodeBmp' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	contents
-> Tensor Build Word8	image

decodeCSV Source #

Arguments

:: forall v'1 v'2 oUT_TYPE. 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

decodeCSV' Source #

Arguments

:: forall v'1 v'2 oUT_TYPE. 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

decodeCompressed Source #

Arguments

:: Tensor v'1 ByteString	bytes
-> Tensor Build ByteString	output

decodeCompressed' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	bytes
-> Tensor Build ByteString	output

decodeGif Source #

Arguments

:: Tensor v'1 ByteString	contents
-> Tensor Build Word8	image

decodeGif' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	contents
-> Tensor Build Word8	image

decodeJSONExample Source #

Arguments

:: Tensor v'1 ByteString	json_examples
-> Tensor Build ByteString	binary_examples

decodeJSONExample' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	json_examples
-> Tensor Build ByteString	binary_examples

decodeJpeg Source #

Arguments

:: Tensor v'1 ByteString	contents
-> Tensor Build Word8	image

decodeJpeg' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	contents
-> Tensor Build Word8	image

decodePaddedRaw Source #

Arguments

:: forall v'1 v'2 out_type. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] out_type
=> Tensor v'1 ByteString	input_bytes
-> Tensor v'2 Int32	fixed_length
-> Tensor Build out_type	output

decodePaddedRaw' Source #

Arguments

:: forall v'1 v'2 out_type. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] out_type
=> OpParams
-> Tensor v'1 ByteString	input_bytes
-> Tensor v'2 Int32	fixed_length
-> Tensor Build out_type	output

decodePng Source #

Arguments

:: forall v'1 dtype. OneOf '[Word16, Word8] dtype
=> Tensor v'1 ByteString	contents
-> Tensor Build dtype	image

decodePng' Source #

Arguments

:: forall v'1 dtype. OneOf '[Word16, Word8] dtype
=> OpParams
-> Tensor v'1 ByteString	contents
-> Tensor Build dtype	image

decodeProtoV2 Source #

Arguments

:: forall v'1 output_types. TensorTypes output_types
=> ByteString	message_type
-> Tensor v'1 ByteString	bytes
-> (Tensor Build Int32, TensorList Build output_types)	(sizes, values) sizes values

decodeProtoV2' Source #

Arguments

:: forall v'1 output_types. TensorTypes output_types
=> OpParams
-> ByteString	message_type
-> Tensor v'1 ByteString	bytes
-> (Tensor Build Int32, TensorList Build output_types)	(sizes, values) sizes values

decodeRaw Source #

Arguments

:: forall v'1 out_type. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] out_type
=> Tensor v'1 ByteString	bytes
-> Tensor Build out_type	output

decodeRaw' Source #

Arguments

:: forall v'1 out_type. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] out_type
=> OpParams
-> Tensor v'1 ByteString	bytes
-> Tensor Build out_type	output

decodeWav Source #

Arguments

:: Tensor v'1 ByteString

contents

-> (Tensor Build Float, Tensor Build Int32)

(audio, sample_rate)

audio
sample_rate

decodeWav' Source #

Arguments

:: OpParams

-> Tensor v'1 ByteString

contents

-> (Tensor Build Float, Tensor Build Int32)

(audio, sample_rate)

audio
sample_rate

deepCopy Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> Tensor v'1 t	x
-> m' (Tensor Value t)	y

deepCopy' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor v'1 t	x
-> m' (Tensor Value t)	y

deleteIterator Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Variant	deleter
-> m' ControlNode

deleteIterator' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Variant	deleter
-> m' ControlNode

deleteMemoryCache Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Variant	deleter
-> m' ControlNode

deleteMemoryCache' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Variant	deleter
-> m' ControlNode

deleteMultiDeviceIterator Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	multi_device_iterator
-> [Tensor v'2 ResourceHandle]	iterators
-> Tensor v'3 Variant	deleter
-> m' ControlNode

deleteMultiDeviceIterator' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	multi_device_iterator
-> [Tensor v'2 ResourceHandle]	iterators
-> Tensor v'3 Variant	deleter
-> m' ControlNode

deleteRandomSeedGenerator Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Variant	deleter
-> m' ControlNode

deleteRandomSeedGenerator' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Variant	deleter
-> m' ControlNode

deleteSeedGenerator Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Variant	deleter
-> m' ControlNode

deleteSeedGenerator' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Variant	deleter
-> m' ControlNode

deleteSessionTensor Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ByteString	handle
-> m' ControlNode

deleteSessionTensor' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	handle
-> m' ControlNode

denseBincount Source #

Arguments

:: forall v'1 v'2 v'3 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t)
=> Tensor v'1 tidx	input
-> Tensor v'2 tidx	size
-> Tensor v'3 t	weights
-> Tensor Build t	output

denseBincount' Source #

Arguments

:: forall v'1 v'2 v'3 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t)
=> OpParams
-> Tensor v'1 tidx	input
-> Tensor v'2 tidx	size
-> Tensor v'3 t	weights
-> Tensor Build t	output

denseCountSparseOutput Source #

Arguments

:: forall v'1 v'2 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type)
=> Bool	binary_output
-> Tensor v'1 t	values
-> Tensor v'2 output_type	weights
-> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)	(output_indices, output_values, output_dense_shape) output_indices output_values output_dense_shape

denseCountSparseOutput' Source #

Arguments

:: forall v'1 v'2 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type)
=> OpParams
-> Bool	binary_output
-> Tensor v'1 t	values
-> Tensor v'2 output_type	weights
-> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)	(output_indices, output_values, output_dense_shape) output_indices output_values output_dense_shape

denseToCSRSparseMatrix Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> Tensor v'1 t	dense_input
-> Tensor v'2 Int64	indices
-> Tensor Build Variant	sparse_output

denseToCSRSparseMatrix' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> OpParams
-> Tensor v'1 t	dense_input
-> Tensor v'2 Int64	indices
-> Tensor Build Variant	sparse_output

denseToDenseSetOperation Source #

Arguments

:: forall v'1 v'2 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t
=> ByteString	set_operation
-> Tensor v'1 t	set1
-> Tensor v'2 t	set2
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)	(result_indices, result_values, result_shape) result_indices result_values result_shape

denseToDenseSetOperation' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t
=> OpParams
-> ByteString	set_operation
-> Tensor v'1 t	set1
-> Tensor v'2 t	set2
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)	(result_indices, result_values, result_shape) result_indices result_values result_shape

denseToSparseBatchDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> Tensor v'3 Int64	row_shape
-> Tensor Build Variant	handle

denseToSparseBatchDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> Tensor v'3 Int64	row_shape
-> Tensor Build Variant	handle

denseToSparseSetOperation Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t
=> ByteString	set_operation
-> 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) result_indices result_values result_shape

denseToSparseSetOperation' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t
=> OpParams
-> ByteString	set_operation
-> 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) result_indices result_values result_shape

depthToSpace Source #

Arguments

:: forall v'1 t. TensorType t
=> Int64	block_size
-> Tensor v'1 t	input
-> Tensor Build t	output

depthToSpace' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Int64	block_size
-> Tensor v'1 t	input
-> Tensor Build t	output

depthwiseConv2dNative Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor Build t	output

depthwiseConv2dNative' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor Build t	output

depthwiseConv2dNativeBackpropFilter Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	filter_sizes
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

depthwiseConv2dNativeBackpropFilter' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	filter_sizes
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

depthwiseConv2dNativeBackpropInput Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 Int32	input_sizes
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

depthwiseConv2dNativeBackpropInput' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 Int32	input_sizes
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	output

dequantize Source #

Arguments

:: forall v'1 v'2 v'3 t dtype. (OneOf '[Int16, Int32, Word16, Word8] t, OneOf '[Word16, Float] dtype)
=> Tensor v'1 t	input
-> Tensor v'2 Float	min_range
-> Tensor v'3 Float	max_range
-> Tensor Build dtype	output

dequantize' Source #

Arguments

:: forall v'1 v'2 v'3 t dtype. (OneOf '[Int16, Int32, Word16, Word8] t, OneOf '[Word16, Float] dtype)
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 Float	min_range
-> Tensor v'3 Float	max_range
-> Tensor Build dtype	output

deserializeIterator Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	resource_handle
-> Tensor v'2 Variant	serialized
-> m' ControlNode

deserializeIterator' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	resource_handle
-> Tensor v'2 Variant	serialized
-> m' ControlNode

deserializeManySparse Source #

Arguments

:: forall v'1 dtype. TensorType dtype

=> Tensor v'1 ByteString

serialized_sparse

-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)

(sparse_indices, sparse_values, sparse_shape)

sparse_indices
sparse_values
sparse_shape

deserializeManySparse' Source #

Arguments

:: forall v'1 dtype. TensorType dtype
=> OpParams
-> Tensor v'1 ByteString	serialized_sparse
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)	(sparse_indices, sparse_values, sparse_shape) sparse_indices sparse_values sparse_shape

deserializeSparse Source #

Arguments

:: forall v'1 dtype tserialized. (TensorType dtype, OneOf '[ByteString, Variant] tserialized)

=> Tensor v'1 tserialized

serialized_sparse

-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)

(sparse_indices, sparse_values, sparse_shape)

sparse_indices
sparse_values
sparse_shape

deserializeSparse' Source #

Arguments

:: forall v'1 dtype tserialized. (TensorType dtype, OneOf '[ByteString, Variant] tserialized)
=> OpParams
-> Tensor v'1 tserialized	serialized_sparse
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)	(sparse_indices, sparse_values, sparse_shape) sparse_indices sparse_values sparse_shape

destroyResourceOp Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	resource
-> m' ControlNode

destroyResourceOp' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> m' ControlNode

destroyTemporaryVariable Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> ByteString	var_name
-> Tensor Ref t	ref
-> m' (Tensor Value t)	value

destroyTemporaryVariable' Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> OpParams
-> ByteString	var_name
-> Tensor Ref t	ref
-> m' (Tensor Value t)	value

deviceIndex Source #

Arguments

:: Tensor Build Int32

index

deviceIndex' Source #

Arguments

:: OpParams
-> Tensor Build Int32	index

diag Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> Tensor v'1 t	diagonal
-> Tensor Build t	output

diag' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	diagonal
-> Tensor Build t	output

diagPart Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	diagonal

diagPart' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	diagonal

digamma Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

digamma' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

dilation2D Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor Build t	output

dilation2D' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor Build t	output

dilation2DBackpropFilter Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	filter_backprop

dilation2DBackpropFilter' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	filter_backprop

dilation2DBackpropInput Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	in_backprop

dilation2DBackpropInput' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> Tensor v'3 t	out_backprop
-> Tensor Build t	in_backprop

directedInterleaveDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	selector_input_dataset
-> [Tensor v'2 Variant]	data_input_datasets
-> Tensor Build Variant	handle

directedInterleaveDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	selector_input_dataset
-> [Tensor v'2 Variant]	data_input_datasets
-> Tensor Build Variant	handle

div Source #

Arguments

:: forall v'1 v'2 t. 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 Build t	z

div' Source #

Arguments

:: 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	x
-> Tensor v'2 t	y
-> Tensor Build t	z

divNoNan Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

divNoNan' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

drawBoundingBoxes Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Float	boxes
-> Tensor Build t	output

drawBoundingBoxes' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor v'2 Float	boxes
-> Tensor Build t	output

drawBoundingBoxesV2 Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Float	boxes
-> Tensor v'3 Float	colors
-> Tensor Build t	output

drawBoundingBoxesV2' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor v'2 Float	boxes
-> Tensor v'3 Float	colors
-> Tensor Build t	output

dummyIterationCounter Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	handle

dummyIterationCounter' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	handle

dummyMemoryCache Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	handle

dummyMemoryCache' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	handle

dummySeedGenerator Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	handle

dummySeedGenerator' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	handle

dynamicPartition Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Int64	num_partitions
-> Tensor v'1 t	data
-> Tensor v'2 Int32	partitions
-> [Tensor Build t]	outputs

dynamicPartition' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Int64	num_partitions
-> Tensor v'1 t	data
-> Tensor v'2 Int32	partitions
-> [Tensor Build t]	outputs

dynamicStitch Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> [Tensor v'1 Int32]	indices
-> [Tensor v'2 t]	data
-> Tensor Build t	merged

dynamicStitch' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> [Tensor v'1 Int32]	indices
-> [Tensor v'2 t]	data
-> Tensor Build t	merged

eagerPyFunc Source #

Arguments

:: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout)
=> ByteString	token
-> TensorList v'1 tin	input
-> m' (TensorList Value tout)	output

eagerPyFunc' Source #

Arguments

:: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout)
=> OpParams
-> ByteString	token
-> TensorList v'1 tin	input
-> m' (TensorList Value tout)	output

editDistance Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t. TensorType t
=> Tensor v'1 Int64	hypothesis_indices
-> Tensor v'2 t	hypothesis_values
-> Tensor v'3 Int64	hypothesis_shape
-> Tensor v'4 Int64	truth_indices
-> Tensor v'5 t	truth_values
-> Tensor v'6 Int64	truth_shape
-> Tensor Build Float	output

editDistance' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t. TensorType t
=> OpParams
-> Tensor v'1 Int64	hypothesis_indices
-> Tensor v'2 t	hypothesis_values
-> Tensor v'3 Int64	hypothesis_shape
-> Tensor v'4 Int64	truth_indices
-> Tensor v'5 t	truth_values
-> Tensor v'6 Int64	truth_shape
-> Tensor Build Float	output

eig Source #

Arguments

:: forall v'1 t tout. (OneOf '[Complex Double, Complex Float, Double, Float] t, OneOf '[Complex Double, Complex Float] tout)

=> Tensor v'1 t

input

-> (Tensor Build tout, Tensor Build tout)

(e, v)

e
v

eig' Source #

Arguments

:: forall v'1 t tout. (OneOf '[Complex Double, Complex Float, Double, Float] t, OneOf '[Complex Double, Complex Float] tout)
=> OpParams
-> Tensor v'1 t	input
-> (Tensor Build tout, Tensor Build tout)	(e, v) e v

einsum Source #

Arguments

:: forall v'1 t. TensorType t
=> ByteString	equation
-> [Tensor v'1 t]	inputs
-> Tensor Build t	output

einsum' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> ByteString	equation
-> [Tensor v'1 t]	inputs
-> Tensor Build t	output

elu Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	features
-> Tensor Build t	activations

elu' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	features
-> Tensor Build t	activations

eluGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	gradients
-> Tensor v'2 t	outputs
-> Tensor Build t	backprops

eluGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	gradients
-> Tensor v'2 t	outputs
-> Tensor Build t	backprops

empty Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor v'1 Int32	shape
-> m' (Tensor Value dtype)	output

empty' Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor v'1 Int32	shape
-> m' (Tensor Value dtype)	output

emptyTensorList Source #

Arguments

:: forall v'1 v'2 shape_type. OneOf '[Int32, Int64] shape_type
=> DataType	element_dtype
-> Tensor v'1 shape_type	element_shape
-> Tensor v'2 Int32	max_num_elements
-> Tensor Build Variant	handle

emptyTensorList' Source #

Arguments

:: forall v'1 v'2 shape_type. OneOf '[Int32, Int64] shape_type
=> OpParams
-> DataType	element_dtype
-> Tensor v'1 shape_type	element_shape
-> Tensor v'2 Int32	max_num_elements
-> Tensor Build Variant	handle

encodeBase64 Source #

Arguments

:: Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

encodeBase64' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

encodeJpeg Source #

Arguments

:: Tensor v'1 Word8	image
-> Tensor Build ByteString	contents

encodeJpeg' Source #

Arguments

:: OpParams
-> Tensor v'1 Word8	image
-> Tensor Build ByteString	contents

encodeJpegVariableQuality Source #

Arguments

:: Tensor v'1 Word8	images
-> Tensor v'2 Int32	quality
-> Tensor Build ByteString	contents

encodeJpegVariableQuality' Source #

Arguments

:: OpParams
-> Tensor v'1 Word8	images
-> Tensor v'2 Int32	quality
-> Tensor Build ByteString	contents

encodePng Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Word8] t
=> Tensor v'1 t	image
-> Tensor Build ByteString	contents

encodePng' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Word8] t
=> OpParams
-> Tensor v'1 t	image
-> Tensor Build ByteString	contents

encodeProto Source #

Arguments

:: forall v'1 v'2 tinput_types. TensorTypes tinput_types
=> ByteString	message_type
-> Tensor v'1 Int32	sizes
-> TensorList v'2 tinput_types	values
-> Tensor Build ByteString	bytes

encodeProto' Source #

Arguments

:: forall v'1 v'2 tinput_types. TensorTypes tinput_types
=> OpParams
-> ByteString	message_type
-> Tensor v'1 Int32	sizes
-> TensorList v'2 tinput_types	values
-> Tensor Build ByteString	bytes

encodeWav Source #

Arguments

:: Tensor v'1 Float	audio
-> Tensor v'2 Int32	sample_rate
-> Tensor Build ByteString	contents

encodeWav' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	audio
-> Tensor v'2 Int32	sample_rate
-> Tensor Build ByteString	contents

enqueueTPUEmbeddingIntegerBatch Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> [Tensor v'1 Int32]	batch
-> Tensor v'2 ByteString	mode_override
-> m' ControlNode

enqueueTPUEmbeddingIntegerBatch' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> [Tensor v'1 Int32]	batch
-> Tensor v'2 ByteString	mode_override
-> m' ControlNode

enqueueTPUEmbeddingRaggedTensorBatch Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3)
=> [Tensor v'1 t1]	sample_splits
-> [Tensor v'2 t2]	embedding_indices
-> [Tensor v'3 t3]	aggregation_weights
-> Tensor v'4 ByteString	mode_override
-> m' ControlNode

enqueueTPUEmbeddingRaggedTensorBatch' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3)
=> OpParams
-> [Tensor v'1 t1]	sample_splits
-> [Tensor v'2 t2]	embedding_indices
-> [Tensor v'3 t3]	aggregation_weights
-> Tensor v'4 ByteString	mode_override
-> m' ControlNode

enqueueTPUEmbeddingSparseBatch Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3)
=> [Tensor v'1 t1]	sample_indices
-> [Tensor v'2 t2]	embedding_indices
-> [Tensor v'3 t3]	aggregation_weights
-> Tensor v'4 ByteString	mode_override
-> m' ControlNode

enqueueTPUEmbeddingSparseBatch' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3)
=> OpParams
-> [Tensor v'1 t1]	sample_indices
-> [Tensor v'2 t2]	embedding_indices
-> [Tensor v'3 t3]	aggregation_weights
-> Tensor v'4 ByteString	mode_override
-> m' ControlNode

enqueueTPUEmbeddingSparseTensorBatch Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3)
=> [Tensor v'1 t1]	sample_indices
-> [Tensor v'2 t2]	embedding_indices
-> [Tensor v'3 t3]	aggregation_weights
-> Tensor v'4 ByteString	mode_override
-> m' ControlNode

enqueueTPUEmbeddingSparseTensorBatch' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t1 t2 t3 m'. (MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2, OneOf '[Double, Float] t3)
=> OpParams
-> [Tensor v'1 t1]	sample_indices
-> [Tensor v'2 t2]	embedding_indices
-> [Tensor v'3 t3]	aggregation_weights
-> Tensor v'4 ByteString	mode_override
-> m' ControlNode

ensureShape Source #

Arguments

:: forall v'1 t. TensorType t
=> Shape	shape
-> Tensor v'1 t	input
-> Tensor Build t	output

ensureShape' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Shape	shape
-> Tensor v'1 t	input
-> Tensor Build t	output

enter Source #

Arguments

:: forall v'1 t. TensorType t
=> ByteString	frame_name
-> Tensor v'1 t	data
-> Tensor Build t	output

enter' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> ByteString	frame_name
-> Tensor v'1 t	data
-> Tensor Build t	output

equal Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

equal' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

erf Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

erf' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

erfc Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

erfc' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

erfinv Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

erfinv' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

euclideanNorm Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

euclideanNorm' Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

exit Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	data
-> Tensor Build t	output

exit' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	data
-> Tensor Build t	output

exp Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

exp' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

expandDims Source #

Arguments

:: forall v'1 v'2 t tdim. (TensorType t, OneOf '[Int32, Int64] tdim)
=> Tensor v'1 t	input
-> Tensor v'2 tdim	dim
-> Tensor Build t	output

expandDims' Source #

Arguments

:: forall v'1 v'2 t tdim. (TensorType t, OneOf '[Int32, Int64] tdim)
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 tdim	dim
-> Tensor Build t	output

experimentalAssertNextDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	transformations
-> Tensor Build Variant	handle

experimentalAssertNextDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	transformations
-> Tensor Build Variant	handle

experimentalAutoShardDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_workers
-> Tensor v'3 Int64	index
-> Tensor Build Variant	handle

experimentalAutoShardDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_workers
-> Tensor v'3 Int64	index
-> Tensor Build Variant	handle

experimentalBytesProducedStatsDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	tag
-> Tensor Build Variant	handle

experimentalBytesProducedStatsDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	tag
-> Tensor Build Variant	handle

experimentalCSVDataset Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 output_types m'. (MonadBuild m', OneOfs '[ByteString, Int32, Int64, Double, Float] output_types)
=> Tensor v'1 ByteString	filenames
-> Tensor v'2 ByteString	compression_type
-> Tensor v'3 Int64	buffer_size
-> Tensor v'4 Bool	header
-> Tensor v'5 ByteString	field_delim
-> Tensor v'6 Bool	use_quote_delim
-> Tensor v'7 ByteString	na_value
-> Tensor v'8 Int64	select_cols
-> TensorList v'9 output_types	record_defaults
-> m' (Tensor Value Variant)	handle

experimentalCSVDataset' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 output_types m'. (MonadBuild m', OneOfs '[ByteString, Int32, Int64, Double, Float] output_types)
=> OpParams
-> Tensor v'1 ByteString	filenames
-> Tensor v'2 ByteString	compression_type
-> Tensor v'3 Int64	buffer_size
-> Tensor v'4 Bool	header
-> Tensor v'5 ByteString	field_delim
-> Tensor v'6 Bool	use_quote_delim
-> Tensor v'7 ByteString	na_value
-> Tensor v'8 Int64	select_cols
-> TensorList v'9 output_types	record_defaults
-> m' (Tensor Value Variant)	handle

experimentalChooseFastestDataset Source #

Arguments

:: Int64	num_experiments
-> [DataType]	output_types
-> [Tensor v'1 Variant]	input_datasets
-> Tensor Build Variant	handle

experimentalChooseFastestDataset' Source #

Arguments

:: OpParams
-> Int64	num_experiments
-> [DataType]	output_types
-> [Tensor v'1 Variant]	input_datasets
-> Tensor Build Variant	handle

experimentalDatasetCardinality Source #

Arguments

:: Tensor v'1 Variant	input_dataset
-> Tensor Build Int64	cardinality

experimentalDatasetCardinality' Source #

Arguments

:: OpParams
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Int64	cardinality

experimentalDatasetToTFRecord Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	filename
-> Tensor v'3 ByteString	compression_type
-> m' ControlNode

experimentalDatasetToTFRecord' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	filename
-> Tensor v'3 ByteString	compression_type
-> m' ControlNode

experimentalDenseToSparseBatchDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> Tensor v'3 Int64	row_shape
-> Tensor Build Variant	handle

experimentalDenseToSparseBatchDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> Tensor v'3 Int64	row_shape
-> Tensor Build Variant	handle

experimentalDirectedInterleaveDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	selector_input_dataset
-> [Tensor v'2 Variant]	data_input_datasets
-> Tensor Build Variant	handle

experimentalDirectedInterleaveDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	selector_input_dataset
-> [Tensor v'2 Variant]	data_input_datasets
-> Tensor Build Variant	handle

experimentalIgnoreErrorsDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

experimentalIgnoreErrorsDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

experimentalIteratorGetDevice Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Value ByteString)	device

experimentalIteratorGetDevice' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Value ByteString)	device

experimentalLMDBDataset Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 ByteString	filenames
-> m' (Tensor Value Variant)	handle

experimentalLMDBDataset' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 ByteString	filenames
-> m' (Tensor Value Variant)	handle

experimentalLatencyStatsDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	tag
-> Tensor Build Variant	handle

experimentalLatencyStatsDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	tag
-> Tensor Build Variant	handle

experimentalMatchingFilesDataset Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ByteString	patterns
-> m' (Tensor Value Variant)	handle

experimentalMatchingFilesDataset' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	patterns
-> m' (Tensor Value Variant)	handle

experimentalMaxIntraOpParallelismDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	max_intra_op_parallelism
-> Tensor Build Variant	handle

experimentalMaxIntraOpParallelismDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	max_intra_op_parallelism
-> Tensor Build Variant	handle

experimentalNonSerializableDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

experimentalNonSerializableDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

experimentalParseExampleDataset Source #

Arguments

:: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense
=> [DataType]	output_types
-> [DataType]	sparse_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_parallel_calls
-> TensorList v'3 tdense	dense_defaults
-> Tensor Build Variant	handle

experimentalParseExampleDataset' Source #

Arguments

:: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense
=> OpParams
-> [DataType]	output_types
-> [DataType]	sparse_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_parallel_calls
-> TensorList v'3 tdense	dense_defaults
-> Tensor Build Variant	handle

experimentalPrivateThreadPoolDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_threads
-> Tensor Build Variant	handle

experimentalPrivateThreadPoolDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_threads
-> Tensor Build Variant	handle

experimentalRandomDataset Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Int64	seed
-> Tensor v'2 Int64	seed2
-> m' (Tensor Value Variant)	handle

experimentalRandomDataset' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Int64	seed
-> Tensor v'2 Int64	seed2
-> m' (Tensor Value Variant)	handle

experimentalRebatchDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_replicas
-> Tensor Build Variant	handle

experimentalRebatchDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_replicas
-> Tensor Build Variant	handle

experimentalSetStatsAggregatorDataset Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ResourceHandle	stats_aggregator
-> Tensor v'3 ByteString	tag
-> Tensor v'4 ByteString	counter_prefix
-> m' (Tensor Value Variant)	handle

experimentalSetStatsAggregatorDataset' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ResourceHandle	stats_aggregator
-> Tensor v'3 ByteString	tag
-> Tensor v'4 ByteString	counter_prefix
-> m' (Tensor Value Variant)	handle

experimentalSleepDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	sleep_microseconds
-> Tensor Build Variant	handle

experimentalSleepDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	sleep_microseconds
-> Tensor Build Variant	handle

experimentalSlidingWindowDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	window_size
-> Tensor v'3 Int64	window_shift
-> Tensor v'4 Int64	window_stride
-> Tensor Build Variant	handle

experimentalSlidingWindowDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	window_size
-> Tensor v'3 Int64	window_shift
-> Tensor v'4 Int64	window_stride
-> Tensor Build Variant	handle

experimentalSqlDataset Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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

experimentalSqlDataset' Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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

experimentalStatsAggregatorHandle Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	handle

experimentalStatsAggregatorHandle' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	handle

experimentalStatsAggregatorSummary Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	iterator
-> m' (Tensor Value ByteString)	summary

experimentalStatsAggregatorSummary' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	iterator
-> m' (Tensor Value ByteString)	summary

experimentalThreadPoolDataset Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ResourceHandle	thread_pool
-> m' (Tensor Value Variant)	handle

experimentalThreadPoolDataset' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ResourceHandle	thread_pool
-> m' (Tensor Value Variant)	handle

experimentalThreadPoolHandle Source #

Arguments

:: forall m'. MonadBuild m'
=> ByteString	display_name
-> Int64	num_threads
-> m' (Tensor Value ResourceHandle)	handle

experimentalThreadPoolHandle' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> ByteString	display_name
-> Int64	num_threads
-> m' (Tensor Value ResourceHandle)	handle

experimentalUnbatchDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

experimentalUnbatchDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

experimentalUniqueDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

experimentalUniqueDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

expint Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

expint' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

expm1 Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

expm1' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

extractGlimpse Source #

Arguments

:: Tensor v'1 Float	input
-> Tensor v'2 Int32	size
-> Tensor v'3 Float	offsets
-> Tensor Build Float	glimpse

extractGlimpse' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	input
-> Tensor v'2 Int32	size
-> Tensor v'3 Float	offsets
-> Tensor Build Float	glimpse

extractGlimpseV2 Source #

Arguments

:: Tensor v'1 Float	input
-> Tensor v'2 Int32	size
-> Tensor v'3 Float	offsets
-> Tensor Build Float	glimpse

extractGlimpseV2' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	input
-> Tensor v'2 Int32	size
-> Tensor v'3 Float	offsets
-> Tensor Build Float	glimpse

extractImagePatches Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	images
-> Tensor Build t	patches

extractImagePatches' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	images
-> Tensor Build t	patches

extractJpegShape Source #

Arguments

:: forall v'1 output_type. OneOf '[Int32, Int64] output_type
=> Tensor v'1 ByteString	contents
-> Tensor Build output_type	image_shape

extractJpegShape' Source #

Arguments

:: forall v'1 output_type. OneOf '[Int32, Int64] output_type
=> OpParams
-> Tensor v'1 ByteString	contents
-> Tensor Build output_type	image_shape

extractVolumePatches Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor Build t	patches

extractVolumePatches' Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor Build t	patches

fFT Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

fFT' Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> OpParams
-> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

fFT2D Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

fFT2D' Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> OpParams
-> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

fFT3D Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

fFT3D' Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> OpParams
-> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

fIFOQueue Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	component_types
-> m' (Tensor Ref ByteString)	handle

fIFOQueue' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	component_types
-> m' (Tensor Ref ByteString)	handle

fIFOQueueV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	component_types
-> m' (Tensor Value ResourceHandle)	handle

fIFOQueueV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	component_types
-> m' (Tensor Value ResourceHandle)	handle

fact Source #

Arguments

:: Tensor Build ByteString

fact

fact' Source #

Arguments

:: OpParams
-> Tensor Build ByteString	fact

fakeParam Source #

Arguments

:: forall dtype. TensorType dtype
=> Shape	shape
-> Tensor Build dtype	output

fakeParam' Source #

Arguments

:: forall dtype. TensorType dtype
=> OpParams
-> Shape	shape
-> Tensor Build dtype	output

fakeQuantWithMinMaxArgs Source #

Arguments

:: Tensor v'1 Float	inputs
-> Tensor Build Float	outputs

fakeQuantWithMinMaxArgs' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	inputs
-> Tensor Build Float	outputs

fakeQuantWithMinMaxArgsGradient Source #

Arguments

:: Tensor v'1 Float	gradients
-> Tensor v'2 Float	inputs
-> Tensor Build Float	backprops

fakeQuantWithMinMaxArgsGradient' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	gradients
-> Tensor v'2 Float	inputs
-> Tensor Build Float	backprops

fakeQuantWithMinMaxVars Source #

Arguments

:: Tensor v'1 Float	inputs
-> Tensor v'2 Float	min
-> Tensor v'3 Float	max
-> Tensor Build Float	outputs

fakeQuantWithMinMaxVars' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	inputs
-> Tensor v'2 Float	min
-> Tensor v'3 Float	max
-> Tensor Build Float	outputs

fakeQuantWithMinMaxVarsGradient Source #

Arguments

:: Tensor v'1 Float	gradients
-> Tensor v'2 Float	inputs
-> Tensor v'3 Float	min
-> Tensor v'4 Float	max
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)	(backprops_wrt_input, backprop_wrt_min, backprop_wrt_max) backprops_wrt_input backprop_wrt_min backprop_wrt_max

fakeQuantWithMinMaxVarsGradient' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	gradients
-> Tensor v'2 Float	inputs
-> Tensor v'3 Float	min
-> Tensor v'4 Float	max
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)	(backprops_wrt_input, backprop_wrt_min, backprop_wrt_max) backprops_wrt_input backprop_wrt_min backprop_wrt_max

fakeQuantWithMinMaxVarsPerChannel Source #

Arguments

:: Tensor v'1 Float	inputs
-> Tensor v'2 Float	min
-> Tensor v'3 Float	max
-> Tensor Build Float	outputs

fakeQuantWithMinMaxVarsPerChannel' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	inputs
-> Tensor v'2 Float	min
-> Tensor v'3 Float	max
-> Tensor Build Float	outputs

fakeQuantWithMinMaxVarsPerChannelGradient Source #

Arguments

:: Tensor v'1 Float	gradients
-> Tensor v'2 Float	inputs
-> Tensor v'3 Float	min
-> Tensor v'4 Float	max
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)	(backprops_wrt_input, backprop_wrt_min, backprop_wrt_max) backprops_wrt_input backprop_wrt_min backprop_wrt_max

fakeQuantWithMinMaxVarsPerChannelGradient' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	gradients
-> Tensor v'2 Float	inputs
-> Tensor v'3 Float	min
-> Tensor v'4 Float	max
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)	(backprops_wrt_input, backprop_wrt_min, backprop_wrt_max) backprops_wrt_input backprop_wrt_min backprop_wrt_max

fakeQueue Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Ref ByteString)	handle

fakeQueue' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Ref ByteString)	handle

fill Source #

Arguments

:: forall v'1 v'2 t index_type. (TensorType t, OneOf '[Int32, Int64] index_type)
=> Tensor v'1 index_type	dims
-> Tensor v'2 t	value
-> Tensor Build t	output

fill' Source #

Arguments

:: forall v'1 v'2 t index_type. (TensorType t, OneOf '[Int32, Int64] index_type)
=> OpParams
-> Tensor v'1 index_type	dims
-> Tensor v'2 t	value
-> Tensor Build t	output

filterByLastComponentDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	output

filterByLastComponentDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	output

fingerprint Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Tensor v'1 t	data
-> Tensor v'2 ByteString	method
-> Tensor Build Word8	fingerprint

fingerprint' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Tensor v'1 t	data
-> Tensor v'2 ByteString	method
-> Tensor Build Word8	fingerprint

fixedLengthRecordDataset Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m'
=> Tensor v'1 ByteString	filenames
-> Tensor v'2 Int64	header_bytes
-> Tensor v'3 Int64	record_bytes
-> Tensor v'4 Int64	footer_bytes
-> Tensor v'5 Int64	buffer_size
-> m' (Tensor Value Variant)	handle

fixedLengthRecordDataset' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	filenames
-> Tensor v'2 Int64	header_bytes
-> Tensor v'3 Int64	record_bytes
-> Tensor v'4 Int64	footer_bytes
-> Tensor v'5 Int64	buffer_size
-> m' (Tensor Value Variant)	handle

fixedLengthRecordDatasetV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 m'. MonadBuild m'
=> Tensor v'1 ByteString	filenames
-> Tensor v'2 Int64	header_bytes
-> Tensor v'3 Int64	record_bytes
-> Tensor v'4 Int64	footer_bytes
-> Tensor v'5 Int64	buffer_size
-> Tensor v'6 ByteString	compression_type
-> m' (Tensor Value Variant)	handle

fixedLengthRecordDatasetV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	filenames
-> Tensor v'2 Int64	header_bytes
-> Tensor v'3 Int64	record_bytes
-> Tensor v'4 Int64	footer_bytes
-> Tensor v'5 Int64	buffer_size
-> Tensor v'6 ByteString	compression_type
-> m' (Tensor Value Variant)	handle

fixedLengthRecordReader Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	record_bytes
-> m' (Tensor Ref ByteString)	reader_handle

fixedLengthRecordReader' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	record_bytes
-> m' (Tensor Ref ByteString)	reader_handle

fixedLengthRecordReaderV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	record_bytes
-> m' (Tensor Value ResourceHandle)	reader_handle

fixedLengthRecordReaderV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	record_bytes
-> m' (Tensor Value ResourceHandle)	reader_handle

fixedUnigramCandidateSampler Source #

Arguments

:: forall v'1 m'. 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) sampled_candidates true_expected_count sampled_expected_count

fixedUnigramCandidateSampler' Source #

Arguments

:: forall v'1 m'. 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) sampled_candidates true_expected_count sampled_expected_count

floor Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

floor' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

floorDiv Source #

Arguments

:: forall v'1 v'2 t. 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 Build t	z

floorDiv' Source #

Arguments

:: 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	x
-> Tensor v'2 t	y
-> Tensor Build t	z

floorMod Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Word64, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

floorMod' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Word64, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

flushSummaryWriter Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	writer
-> m' ControlNode

flushSummaryWriter' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	writer
-> m' ControlNode

fractionalAvgPool Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t

=> Tensor v'1 t

value

-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)

(output, row_pooling_sequence, col_pooling_sequence)

output
row_pooling_sequence
col_pooling_sequence

fractionalAvgPool' Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t
=> OpParams
-> Tensor v'1 t	value
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)	(output, row_pooling_sequence, col_pooling_sequence) output row_pooling_sequence col_pooling_sequence

fractionalAvgPoolGrad Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Int64, Double, Float] t
=> Tensor v'1 Int64	orig_input_tensor_shape
-> Tensor v'2 t	out_backprop
-> Tensor v'3 Int64	row_pooling_sequence
-> Tensor v'4 Int64	col_pooling_sequence
-> Tensor Build t	output

fractionalAvgPoolGrad' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Int64, Double, Float] t
=> OpParams
-> Tensor v'1 Int64	orig_input_tensor_shape
-> Tensor v'2 t	out_backprop
-> Tensor v'3 Int64	row_pooling_sequence
-> Tensor v'4 Int64	col_pooling_sequence
-> Tensor Build t	output

fractionalMaxPool Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t

=> Tensor v'1 t

value

-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)

(output, row_pooling_sequence, col_pooling_sequence)

output
row_pooling_sequence
col_pooling_sequence

fractionalMaxPool' Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t
=> OpParams
-> Tensor v'1 t	value
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)	(output, row_pooling_sequence, col_pooling_sequence) output row_pooling_sequence col_pooling_sequence

fractionalMaxPoolGrad Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int32, Int64, Double, Float] t
=> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	out_backprop
-> Tensor v'4 Int64	row_pooling_sequence
-> Tensor v'5 Int64	col_pooling_sequence
-> Tensor Build t	output

fractionalMaxPoolGrad' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int32, Int64, Double, Float] t
=> OpParams
-> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	out_backprop
-> Tensor v'4 Int64	row_pooling_sequence
-> Tensor v'5 Int64	col_pooling_sequence
-> Tensor Build t	output

fresnelCos Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

fresnelCos' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

fresnelSin Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

fresnelSin' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

fusedBatchNorm Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. 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) y batch_mean batch_variance reserve_space_1 reserve_space_2

fusedBatchNorm' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. 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) y batch_mean batch_variance reserve_space_1 reserve_space_2

fusedBatchNormGrad Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. 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) x_backprop scale_backprop offset_backprop reserve_space_3 reserve_space_4

fusedBatchNormGrad' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. 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) x_backprop scale_backprop offset_backprop reserve_space_3 reserve_space_4

fusedBatchNormGradV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t u. (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) x_backprop scale_backprop offset_backprop reserve_space_3 reserve_space_4

fusedBatchNormGradV2' Source #

Arguments

:: 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	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) x_backprop scale_backprop offset_backprop reserve_space_3 reserve_space_4

fusedBatchNormGradV3 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t u. (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 v'6 u	reserve_space_3
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u)	(x_backprop, scale_backprop, offset_backprop, reserve_space_4, reserve_space_5) x_backprop scale_backprop offset_backprop reserve_space_4 reserve_space_5

fusedBatchNormGradV3' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t u. (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 v'6 u	reserve_space_3
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u)	(x_backprop, scale_backprop, offset_backprop, reserve_space_4, reserve_space_5) x_backprop scale_backprop offset_backprop reserve_space_4 reserve_space_5

fusedBatchNormV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t u. (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) y batch_mean batch_variance reserve_space_1 reserve_space_2

fusedBatchNormV2' Source #

Arguments

:: 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	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) y batch_mean batch_variance reserve_space_1 reserve_space_2

fusedBatchNormV3 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t u. (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, Tensor Build u)	(y, batch_mean, batch_variance, reserve_space_1, reserve_space_2, reserve_space_3) y batch_mean batch_variance reserve_space_1 reserve_space_2 reserve_space_3

fusedBatchNormV3' Source #

Arguments

:: 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	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, Tensor Build u)	(y, batch_mean, batch_variance, reserve_space_1, reserve_space_2, reserve_space_3) y batch_mean batch_variance reserve_space_1 reserve_space_2 reserve_space_3

fusedPadConv2D Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> ByteString	mode
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	paddings
-> Tensor v'3 t	filter
-> Tensor Build t	output

fusedPadConv2D' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	mode
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	paddings
-> Tensor v'3 t	filter
-> Tensor Build t	output

fusedResizeAndPadConv2D Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t
=> ByteString	mode
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	size
-> Tensor v'3 Int32	paddings
-> Tensor v'4 t	filter
-> Tensor Build t	output

fusedResizeAndPadConv2D' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	mode
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	size
-> Tensor v'3 Int32	paddings
-> Tensor v'4 t	filter
-> Tensor Build t	output

gRUBlockCell Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	h_prev
-> Tensor v'3 t	w_ru
-> Tensor v'4 t	w_c
-> Tensor v'5 t	b_ru
-> Tensor v'6 t	b_c
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(r, u, c, h) r u c h

gRUBlockCell' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	h_prev
-> Tensor v'3 t	w_ru
-> Tensor v'4 t	w_c
-> Tensor v'5 t	b_ru
-> Tensor v'6 t	b_c
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(r, u, c, h) r u c h

gRUBlockCellGrad Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t. OneOf '[Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	h_prev
-> Tensor v'3 t	w_ru
-> Tensor v'4 t	w_c
-> Tensor v'5 t	b_ru
-> Tensor v'6 t	b_c
-> Tensor v'7 t	r
-> Tensor v'8 t	u
-> Tensor v'9 t	c
-> Tensor v'10 t	d_h
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(d_x, d_h_prev, d_c_bar, d_r_bar_u_bar) d_x d_h_prev d_c_bar d_r_bar_u_bar

gRUBlockCellGrad' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t. OneOf '[Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	h_prev
-> Tensor v'3 t	w_ru
-> Tensor v'4 t	w_c
-> Tensor v'5 t	b_ru
-> Tensor v'6 t	b_c
-> Tensor v'7 t	r
-> Tensor v'8 t	u
-> Tensor v'9 t	c
-> Tensor v'10 t	d_h
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(d_x, d_h_prev, d_c_bar, d_r_bar_u_bar) d_x d_h_prev d_c_bar d_r_bar_u_bar

gather Source #

Arguments

:: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 tparams	params
-> Tensor v'2 tindices	indices
-> Tensor Build tparams	output

gather' Source #

Arguments

:: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 tparams	params
-> Tensor v'2 tindices	indices
-> Tensor Build tparams	output

gatherNd Source #

Arguments

:: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 tparams	params
-> Tensor v'2 tindices	indices
-> Tensor Build tparams	output

gatherNd' Source #

Arguments

:: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 tparams	params
-> Tensor v'2 tindices	indices
-> Tensor Build tparams	output

gatherV2 Source #

Arguments

:: forall v'1 v'2 v'3 tparams tindices taxis. (TensorType tparams, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] taxis)
=> Tensor v'1 tparams	params
-> Tensor v'2 tindices	indices
-> Tensor v'3 taxis	axis
-> Tensor Build tparams	output

gatherV2' Source #

Arguments

:: 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	params
-> Tensor v'2 tindices	indices
-> Tensor v'3 taxis	axis
-> Tensor Build tparams	output

generateBoundingBoxProposals Source #

Arguments

:: Tensor v'1 Float	scores
-> Tensor v'2 Float	bbox_deltas
-> Tensor v'3 Float	image_info
-> Tensor v'4 Float	anchors
-> Tensor v'5 Float	nms_threshold
-> Tensor v'6 Int32	pre_nms_topn
-> Tensor v'7 Float	min_size
-> (Tensor Build Float, Tensor Build Float)	(rois, roi_probabilities) rois roi_probabilities

generateBoundingBoxProposals' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	scores
-> Tensor v'2 Float	bbox_deltas
-> Tensor v'3 Float	image_info
-> Tensor v'4 Float	anchors
-> Tensor v'5 Float	nms_threshold
-> Tensor v'6 Int32	pre_nms_topn
-> Tensor v'7 Float	min_size
-> (Tensor Build Float, Tensor Build Float)	(rois, roi_probabilities) rois roi_probabilities

generateVocabRemapping Source #

Arguments

:: Int64	new_vocab_offset
-> Int64	num_new_vocab
-> Tensor v'1 ByteString	new_vocab_file
-> Tensor v'2 ByteString	old_vocab_file
-> (Tensor Build Int64, Tensor Build Int32)	(remapping, num_present) remapping num_present

generateVocabRemapping' Source #

Arguments

:: OpParams
-> Int64	new_vocab_offset
-> Int64	num_new_vocab
-> Tensor v'1 ByteString	new_vocab_file
-> Tensor v'2 ByteString	old_vocab_file
-> (Tensor Build Int64, Tensor Build Int32)	(remapping, num_present) remapping num_present

getSessionHandle Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> Tensor v'1 t	value
-> m' (Tensor Value ByteString)	handle

getSessionHandle' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor v'1 t	value
-> m' (Tensor Value ByteString)	handle

getSessionHandleV2 Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> Tensor v'1 t	value
-> m' (Tensor Value ResourceHandle)	handle

getSessionHandleV2' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor v'1 t	value
-> m' (Tensor Value ResourceHandle)	handle

getSessionTensor Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor v'1 ByteString	handle
-> m' (Tensor Value dtype)	value

getSessionTensor' Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor v'1 ByteString	handle
-> m' (Tensor Value dtype)	value

greater Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

greater' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

greaterEqual Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

greaterEqual' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

guaranteeConst Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> Tensor v'1 t	input
-> m' (Tensor Value t)	output

guaranteeConst' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor v'1 t	input
-> m' (Tensor Value t)	output

hSVToRGB Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	images
-> Tensor Build t	output

hSVToRGB' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor Build t	output

hashTable Source #

Arguments

:: forall m'. MonadBuild m'
=> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Ref ByteString)	table_handle

hashTable' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Ref ByteString)	table_handle

hashTableV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Value ResourceHandle)	table_handle

hashTableV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Value ResourceHandle)	table_handle

histogramFixedWidth Source #

Arguments

:: forall v'1 v'2 v'3 t dtype. (OneOf '[Int32, Int64, Double, Float] t, OneOf '[Int32, Int64] dtype)
=> Tensor v'1 t	values
-> Tensor v'2 t	value_range
-> Tensor v'3 Int32	nbins
-> Tensor Build dtype	out

histogramFixedWidth' Source #

Arguments

:: forall v'1 v'2 v'3 t dtype. (OneOf '[Int32, Int64, Double, Float] t, OneOf '[Int32, Int64] dtype)
=> OpParams
-> Tensor v'1 t	values
-> Tensor v'2 t	value_range
-> Tensor v'3 Int32	nbins
-> Tensor Build dtype	out

histogramSummary Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 ByteString	tag
-> Tensor v'2 t	values
-> Tensor Build ByteString	summary

histogramSummary' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 ByteString	tag
-> Tensor v'2 t	values
-> Tensor Build ByteString	summary

hostConst Source #

Arguments

:: forall dtype. TensorType dtype
=> Tensor Build dtype	output

hostConst' Source #

Arguments

:: forall dtype. TensorType dtype
=> OpParams
-> Tensor Build dtype	output

iFFT Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

iFFT' Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> OpParams
-> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

iFFT2D Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

iFFT2D' Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> OpParams
-> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

iFFT3D Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

iFFT3D' Source #

Arguments

:: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex
=> OpParams
-> Tensor v'1 tcomplex	input
-> Tensor Build tcomplex	output

iRFFT Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> Tensor v'1 tcomplex	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build treal	output

iRFFT' Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> OpParams
-> Tensor v'1 tcomplex	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build treal	output

iRFFT2D Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> Tensor v'1 tcomplex	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build treal	output

iRFFT2D' Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> OpParams
-> Tensor v'1 tcomplex	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build treal	output

iRFFT3D Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> Tensor v'1 tcomplex	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build treal	output

iRFFT3D' Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> OpParams
-> Tensor v'1 tcomplex	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build treal	output

identity Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	output

identity' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

identityN Source #

Arguments

:: forall v'1 t. TensorTypes t
=> TensorList v'1 t	input
-> TensorList Build t	output

identityN' Source #

Arguments

:: forall v'1 t. TensorTypes t
=> OpParams
-> TensorList v'1 t	input
-> TensorList Build t	output

identityReader Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Ref ByteString)	reader_handle

identityReader' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Ref ByteString)	reader_handle

identityReaderV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	reader_handle

identityReaderV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	reader_handle

igamma Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 t	a
-> Tensor v'2 t	x
-> Tensor Build t	z

igamma' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	a
-> Tensor v'2 t	x
-> Tensor Build t	z

igammaGradA Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 t	a
-> Tensor v'2 t	x
-> Tensor Build t	z

igammaGradA' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	a
-> Tensor v'2 t	x
-> Tensor Build t	z

igammac Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 t	a
-> Tensor v'2 t	x
-> Tensor Build t	z

igammac' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	a
-> Tensor v'2 t	x
-> Tensor Build t	z

ignoreErrorsDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

ignoreErrorsDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

imag Source #

Arguments

:: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout)
=> Tensor v'1 t	input
-> Tensor Build tout	output

imag' Source #

Arguments

:: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout)
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build tout	output

imageProjectiveTransformV2 Source #

Arguments

:: forall v'1 v'2 v'3 dtype. OneOf '[Int32, Int64, Word16, Word8, Double, Float] dtype
=> ByteString	interpolation
-> Tensor v'1 dtype	images
-> Tensor v'2 Float	transforms
-> Tensor v'3 Int32	output_shape
-> Tensor Build dtype	transformed_images

imageProjectiveTransformV2' Source #

Arguments

:: forall v'1 v'2 v'3 dtype. OneOf '[Int32, Int64, Word16, Word8, Double, Float] dtype
=> OpParams
-> ByteString	interpolation
-> Tensor v'1 dtype	images
-> Tensor v'2 Float	transforms
-> Tensor v'3 Int32	output_shape
-> Tensor Build dtype	transformed_images

imageSummary Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Word8, Double, Float] t
=> Tensor v'1 ByteString	tag
-> Tensor v'2 t	tensor
-> Tensor Build ByteString	summary

imageSummary' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 ByteString	tag
-> Tensor v'2 t	tensor
-> Tensor Build ByteString	summary

immutableConst Source #

Arguments

:: forall dtype. TensorType dtype
=> ByteString	memory_region_name
-> Shape	shape
-> Tensor Build dtype	tensor

immutableConst' Source #

Arguments

:: forall dtype. TensorType dtype
=> OpParams
-> ByteString	memory_region_name
-> Shape	shape
-> Tensor Build dtype	tensor

importEvent Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	writer
-> Tensor v'2 ByteString	event
-> m' ControlNode

importEvent' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	writer
-> Tensor v'2 ByteString	event
-> m' ControlNode

inTopK Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64] t
=> Int64	k
-> Tensor v'1 Float	predictions
-> Tensor v'2 t	targets
-> Tensor Build Bool	precision

inTopK' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64] t
=> OpParams
-> Int64	k
-> Tensor v'1 Float	predictions
-> Tensor v'2 t	targets
-> Tensor Build Bool	precision

inTopKV2 Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64] t
=> Tensor v'1 Float	predictions
-> Tensor v'2 t	targets
-> Tensor v'3 t	k
-> Tensor Build Bool	precision

inTopKV2' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64] t
=> OpParams
-> Tensor v'1 Float	predictions
-> Tensor v'2 t	targets
-> Tensor v'3 t	k
-> Tensor Build Bool	precision

infeedDequeue Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> Shape	shape
-> m' (Tensor Value dtype)	output

infeedDequeue' Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Shape	shape
-> m' (Tensor Value dtype)	output

infeedDequeueTuple Source #

Arguments

:: forall dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> m' (TensorList Value dtypes)	outputs

infeedDequeueTuple' Source #

Arguments

:: forall dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> m' (TensorList Value dtypes)	outputs

infeedEnqueue Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor v'1 dtype	input
-> m' ControlNode

infeedEnqueue' Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor v'1 dtype	input
-> m' ControlNode

infeedEnqueuePrelinearizedBuffer Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 Variant	input
-> m' ControlNode

infeedEnqueuePrelinearizedBuffer' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 Variant	input
-> m' ControlNode

infeedEnqueueTuple Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> TensorList v'1 dtypes	inputs
-> m' ControlNode

infeedEnqueueTuple' Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> TensorList v'1 dtypes	inputs
-> m' ControlNode

initializeTable Source #

Arguments

:: forall v'2 v'3 tkey tval m'. (MonadBuild m', TensorType tkey, TensorType tval)
=> Tensor Ref ByteString	table_handle
-> Tensor v'2 tkey	keys
-> Tensor v'3 tval	values
-> m' ControlNode

initializeTable' Source #

Arguments

:: forall v'2 v'3 tkey tval m'. (MonadBuild m', TensorType tkey, TensorType tval)
=> OpParams
-> Tensor Ref ByteString	table_handle
-> Tensor v'2 tkey	keys
-> Tensor v'3 tval	values
-> m' ControlNode

initializeTableFromDataset Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	table_handle
-> Tensor v'2 Variant	dataset
-> m' ControlNode

initializeTableFromDataset' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	table_handle
-> Tensor v'2 Variant	dataset
-> m' ControlNode

initializeTableFromTextFile Source #

Arguments

:: forall v'2 m'. MonadBuild m'
=> Int64	key_index
-> Int64	value_index
-> Tensor Ref ByteString	table_handle
-> Tensor v'2 ByteString	filename
-> m' ControlNode

initializeTableFromTextFile' Source #

Arguments

:: forall v'2 m'. MonadBuild m'
=> OpParams
-> Int64	key_index
-> Int64	value_index
-> Tensor Ref ByteString	table_handle
-> Tensor v'2 ByteString	filename
-> m' ControlNode

initializeTableFromTextFileV2 Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Int64	key_index
-> Int64	value_index
-> Tensor v'1 ResourceHandle	table_handle
-> Tensor v'2 ByteString	filename
-> m' ControlNode

initializeTableFromTextFileV2' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Int64	key_index
-> Int64	value_index
-> Tensor v'1 ResourceHandle	table_handle
-> Tensor v'2 ByteString	filename
-> m' ControlNode

initializeTableV2 Source #

Arguments

:: forall v'1 v'2 v'3 tkey tval m'. (MonadBuild m', TensorType tkey, TensorType tval)
=> Tensor v'1 ResourceHandle	table_handle
-> Tensor v'2 tkey	keys
-> Tensor v'3 tval	values
-> m' ControlNode

initializeTableV2' Source #

Arguments

:: forall v'1 v'2 v'3 tkey tval m'. (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

inplaceAdd Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 t	x
-> Tensor v'2 Int32	i
-> Tensor v'3 t	v
-> Tensor Build t	y

inplaceAdd' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 Int32	i
-> Tensor v'3 t	v
-> Tensor Build t	y

inplaceSub Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 t	x
-> Tensor v'2 Int32	i
-> Tensor v'3 t	v
-> Tensor Build t	y

inplaceSub' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 Int32	i
-> Tensor v'3 t	v
-> Tensor Build t	y

inplaceUpdate Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 t	x
-> Tensor v'2 Int32	i
-> Tensor v'3 t	v
-> Tensor Build t	y

inplaceUpdate' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 Int32	i
-> Tensor v'3 t	v
-> Tensor Build t	y

inv Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

inv' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

invGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

invGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

invert Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> Tensor v'1 t	x
-> Tensor Build t	y

invert' Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

invertPermutation Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64] t
=> Tensor v'1 t	x
-> Tensor Build t	y

invertPermutation' Source #

Arguments

:: forall v'1 t. OneOf '[Int32, Int64] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

isBoostedTreesEnsembleInitialized Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	tree_ensemble_handle
-> m' (Tensor Value Bool)	is_initialized

isBoostedTreesEnsembleInitialized' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	tree_ensemble_handle
-> m' (Tensor Value Bool)	is_initialized

isBoostedTreesQuantileStreamResourceInitialized Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> m' (Tensor Value Bool)	is_initialized

isBoostedTreesQuantileStreamResourceInitialized' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	quantile_stream_resource_handle
-> m' (Tensor Value Bool)	is_initialized

isFinite Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build Bool	y

isFinite' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build Bool	y

isInf Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build Bool	y

isInf' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build Bool	y

isNan Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build Bool	y

isNan' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build Bool	y

isVariableInitialized Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor Ref dtype	ref
-> m' (Tensor Value Bool)	is_initialized

isVariableInitialized' Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor Ref dtype	ref
-> m' (Tensor Value Bool)	is_initialized

iterator Source #

Arguments

:: forall m'. MonadBuild m'
=> ByteString	container
-> [DataType]	output_types
-> ByteString	shared_name
-> m' (Tensor Value ResourceHandle)	handle

iterator' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> ByteString	container
-> [DataType]	output_types
-> ByteString	shared_name
-> m' (Tensor Value ResourceHandle)	handle

iteratorFromStringHandle Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ByteString	string_handle
-> m' (Tensor Value ResourceHandle)	resource_handle

iteratorFromStringHandle' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	string_handle
-> m' (Tensor Value ResourceHandle)	resource_handle

iteratorFromStringHandleV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ByteString	string_handle
-> m' (Tensor Value ResourceHandle)	resource_handle

iteratorFromStringHandleV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	string_handle
-> m' (Tensor Value ResourceHandle)	resource_handle

iteratorGetDevice Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Value ByteString)	device

iteratorGetDevice' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Value ByteString)	device

iteratorGetNext Source #

Arguments

:: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types)
=> Tensor v'1 ResourceHandle	iterator
-> m' (TensorList Value output_types)	components

iteratorGetNext' Source #

Arguments

:: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types)
=> OpParams
-> Tensor v'1 ResourceHandle	iterator
-> m' (TensorList Value output_types)	components

iteratorGetNextAsOptional Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 ResourceHandle	iterator
-> m' (Tensor Value Variant)	optional

iteratorGetNextAsOptional' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 ResourceHandle	iterator
-> m' (Tensor Value Variant)	optional

iteratorGetNextSync Source #

Arguments

:: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types)
=> Tensor v'1 ResourceHandle	iterator
-> m' (TensorList Value output_types)	components

iteratorGetNextSync' Source #

Arguments

:: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types)
=> OpParams
-> Tensor v'1 ResourceHandle	iterator
-> m' (TensorList Value output_types)	components

iteratorToStringHandle Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	resource_handle
-> m' (Tensor Value ByteString)	string_handle

iteratorToStringHandle' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	resource_handle
-> m' (Tensor Value ByteString)	string_handle

iteratorV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> ByteString	container
-> [DataType]	output_types
-> ByteString	shared_name
-> m' (Tensor Value ResourceHandle)	handle

iteratorV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> ByteString	container
-> [DataType]	output_types
-> ByteString	shared_name
-> m' (Tensor Value ResourceHandle)	handle

kMC2ChainInitialization Source #

Arguments

:: Tensor v'1 Float	distances
-> Tensor v'2 Int64	seed
-> Tensor Build Int64	index

kMC2ChainInitialization' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	distances
-> Tensor v'2 Int64	seed
-> Tensor Build Int64	index

kmeansPlusPlusInitialization Source #

Arguments

:: Tensor v'1 Float	points
-> Tensor v'2 Int64	num_to_sample
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	num_retries_per_sample
-> Tensor Build Float	samples

kmeansPlusPlusInitialization' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	points
-> Tensor v'2 Int64	num_to_sample
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	num_retries_per_sample
-> Tensor Build Float	samples

l2Loss Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	t
-> Tensor Build t	output

l2Loss' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	t
-> Tensor Build t	output

lMDBDataset Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 ByteString	filenames
-> m' (Tensor Value Variant)	handle

lMDBDataset' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 ByteString	filenames
-> m' (Tensor Value Variant)	handle

lMDBReader Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Ref ByteString)	reader_handle

lMDBReader' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Ref ByteString)	reader_handle

lRN Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

lRN' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

lRNGrad Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Float] t
=> Tensor v'1 t	input_grads
-> Tensor v'2 t	input_image
-> Tensor v'3 t	output_image
-> Tensor Build t	output

lRNGrad' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Float] t
=> OpParams
-> Tensor v'1 t	input_grads
-> Tensor v'2 t	input_image
-> Tensor v'3 t	output_image
-> Tensor Build t	output

lSTMBlockCell Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t. OneOf '[Word16, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	cs_prev
-> Tensor v'3 t	h_prev
-> Tensor v'4 t	w
-> Tensor v'5 t	wci
-> Tensor v'6 t	wcf
-> Tensor v'7 t	wco
-> Tensor v'8 t	b
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(i, cs, f, o, ci, co, h) i cs f o ci co h

lSTMBlockCell' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t. OneOf '[Word16, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	cs_prev
-> Tensor v'3 t	h_prev
-> Tensor v'4 t	w
-> Tensor v'5 t	wci
-> Tensor v'6 t	wcf
-> Tensor v'7 t	wco
-> Tensor v'8 t	b
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(i, cs, f, o, ci, co, h) i cs f o ci co h

lSTMBlockCellGrad Source #

Arguments

:: 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 v'16 t. OneOf '[Word16, Float] t
=> Bool	use_peephole
-> Tensor v'1 t	x
-> Tensor v'2 t	cs_prev
-> Tensor v'3 t	h_prev
-> Tensor v'4 t	w
-> Tensor v'5 t	wci
-> Tensor v'6 t	wcf
-> Tensor v'7 t	wco
-> Tensor v'8 t	b
-> Tensor v'9 t	i
-> Tensor v'10 t	cs
-> Tensor v'11 t	f
-> Tensor v'12 t	o
-> Tensor v'13 t	ci
-> Tensor v'14 t	co
-> Tensor v'15 t	cs_grad
-> Tensor v'16 t	h_grad
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(cs_prev_grad, dicfo, wci_grad, wcf_grad, wco_grad) cs_prev_grad dicfo wci_grad wcf_grad wco_grad

lSTMBlockCellGrad' Source #

Arguments

:: 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 v'16 t. OneOf '[Word16, Float] t
=> OpParams
-> Bool	use_peephole
-> Tensor v'1 t	x
-> Tensor v'2 t	cs_prev
-> Tensor v'3 t	h_prev
-> Tensor v'4 t	w
-> Tensor v'5 t	wci
-> Tensor v'6 t	wcf
-> Tensor v'7 t	wco
-> Tensor v'8 t	b
-> Tensor v'9 t	i
-> Tensor v'10 t	cs
-> Tensor v'11 t	f
-> Tensor v'12 t	o
-> Tensor v'13 t	ci
-> Tensor v'14 t	co
-> Tensor v'15 t	cs_grad
-> Tensor v'16 t	h_grad
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)	(cs_prev_grad, dicfo, wci_grad, wcf_grad, wco_grad) cs_prev_grad dicfo wci_grad wcf_grad wco_grad

latencyStatsDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	tag
-> Tensor Build Variant	handle

latencyStatsDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	tag
-> Tensor Build Variant	handle

leakyRelu Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	features
-> Tensor Build t	activations

leakyRelu' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	features
-> Tensor Build t	activations

leakyReluGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	gradients
-> Tensor v'2 t	features
-> Tensor Build t	backprops

leakyReluGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	gradients
-> Tensor v'2 t	features
-> Tensor Build t	backprops

learnedUnigramCandidateSampler Source #

Arguments

:: forall v'1 m'. 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) sampled_candidates true_expected_count sampled_expected_count

learnedUnigramCandidateSampler' Source #

Arguments

:: forall v'1 m'. 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) sampled_candidates true_expected_count sampled_expected_count

leftShift Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

leftShift' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

less Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

less' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

lessEqual Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

lessEqual' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

lgamma Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

lgamma' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

linSpace Source #

Arguments

:: forall v'1 v'2 v'3 t tidx. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx)
=> Tensor v'1 t	start
-> Tensor v'2 t	stop
-> Tensor v'3 tidx	num
-> Tensor Build t	output

linSpace' Source #

Arguments

:: forall v'1 v'2 v'3 t tidx. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx)
=> OpParams
-> Tensor v'1 t	start
-> Tensor v'2 t	stop
-> Tensor v'3 tidx	num
-> Tensor Build t	output

listDiff Source #

Arguments

:: forall v'1 v'2 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx)
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> (Tensor Build t, Tensor Build out_idx)	(out, idx) out idx

listDiff' Source #

Arguments

:: forall v'1 v'2 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx)
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> (Tensor Build t, Tensor Build out_idx)	(out, idx) out idx

loadAndRemapMatrix Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m'
=> Int64	num_cols
-> Int64	num_rows
-> Tensor v'1 ByteString	ckpt_path
-> Tensor v'2 ByteString	old_tensor_name
-> Tensor v'3 Int64	row_remapping
-> Tensor v'4 Int64	col_remapping
-> Tensor v'5 Float	initializing_values
-> m' (Tensor Value Float)	output_matrix

loadAndRemapMatrix' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m'
=> OpParams
-> Int64	num_cols
-> Int64	num_rows
-> Tensor v'1 ByteString	ckpt_path
-> Tensor v'2 ByteString	old_tensor_name
-> Tensor v'3 Int64	row_remapping
-> Tensor v'4 Int64	col_remapping
-> Tensor v'5 Float	initializing_values
-> m' (Tensor Value Float)	output_matrix

loadTPUEmbeddingADAMParameters Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	momenta
-> Tensor v'3 Float	velocities
-> m' ControlNode

loadTPUEmbeddingADAMParameters' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	momenta
-> Tensor v'3 Float	velocities
-> m' ControlNode

loadTPUEmbeddingADAMParametersGradAccumDebug Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	momenta
-> Tensor v'3 Float	velocities
-> Tensor v'4 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingADAMParametersGradAccumDebug' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	momenta
-> Tensor v'3 Float	velocities
-> Tensor v'4 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingAdadeltaParameters Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	updates
-> m' ControlNode

loadTPUEmbeddingAdadeltaParameters' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	updates
-> m' ControlNode

loadTPUEmbeddingAdadeltaParametersGradAccumDebug Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	updates
-> Tensor v'4 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingAdadeltaParametersGradAccumDebug' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	updates
-> Tensor v'4 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingAdagradParameters Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> m' ControlNode

loadTPUEmbeddingAdagradParameters' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> m' ControlNode

loadTPUEmbeddingAdagradParametersGradAccumDebug Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingAdagradParametersGradAccumDebug' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingCenteredRMSPropParameters Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	ms
-> Tensor v'3 Float	mom
-> Tensor v'4 Float	mg
-> m' ControlNode

loadTPUEmbeddingCenteredRMSPropParameters' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	ms
-> Tensor v'3 Float	mom
-> Tensor v'4 Float	mg
-> m' ControlNode

loadTPUEmbeddingFTRLParameters Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	linears
-> m' ControlNode

loadTPUEmbeddingFTRLParameters' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	linears
-> m' ControlNode

loadTPUEmbeddingFTRLParametersGradAccumDebug Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	linears
-> Tensor v'4 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingFTRLParametersGradAccumDebug' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	linears
-> Tensor v'4 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingMDLAdagradLightParameters Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	weights
-> Tensor v'4 Float	benefits
-> m' ControlNode

loadTPUEmbeddingMDLAdagradLightParameters' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	weights
-> Tensor v'4 Float	benefits
-> m' ControlNode

loadTPUEmbeddingMomentumParameters Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	momenta
-> m' ControlNode

loadTPUEmbeddingMomentumParameters' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	momenta
-> m' ControlNode

loadTPUEmbeddingMomentumParametersGradAccumDebug Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	momenta
-> Tensor v'3 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingMomentumParametersGradAccumDebug' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	momenta
-> Tensor v'3 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingProximalAdagradParameters Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> m' ControlNode

loadTPUEmbeddingProximalAdagradParameters' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> m' ControlNode

loadTPUEmbeddingProximalAdagradParametersGradAccumDebug Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingProximalAdagradParametersGradAccumDebug' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	accumulators
-> Tensor v'3 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingProximalYogiParameters Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	v
-> Tensor v'3 Float	m
-> m' ControlNode

loadTPUEmbeddingProximalYogiParameters' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	v
-> Tensor v'3 Float	m
-> m' ControlNode

loadTPUEmbeddingProximalYogiParametersGradAccumDebug Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	v
-> Tensor v'3 Float	m
-> Tensor v'4 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingProximalYogiParametersGradAccumDebug' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	v
-> Tensor v'3 Float	m
-> Tensor v'4 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingRMSPropParameters Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	ms
-> Tensor v'3 Float	mom
-> m' ControlNode

loadTPUEmbeddingRMSPropParameters' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	ms
-> Tensor v'3 Float	mom
-> m' ControlNode

loadTPUEmbeddingRMSPropParametersGradAccumDebug Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	ms
-> Tensor v'3 Float	mom
-> Tensor v'4 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingRMSPropParametersGradAccumDebug' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	ms
-> Tensor v'3 Float	mom
-> Tensor v'4 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingStochasticGradientDescentParameters Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> m' ControlNode

loadTPUEmbeddingStochasticGradientDescentParameters' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> m' ControlNode

loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	gradient_accumulators
-> m' ControlNode

loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> Tensor v'1 Float	parameters
-> Tensor v'2 Float	gradient_accumulators
-> m' ControlNode

log Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

log' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

log1p Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

log1p' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

logMatrixDeterminant Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t

=> Tensor v'1 t

input

-> (Tensor Build t, Tensor Build t)

(sign, log_abs_determinant)

sign
log_abs_determinant

logMatrixDeterminant' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> (Tensor Build t, Tensor Build t)	(sign, log_abs_determinant) sign log_abs_determinant

logSoftmax Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	logits
-> Tensor Build t	logsoftmax

logSoftmax' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	logits
-> Tensor Build t	logsoftmax

logUniformCandidateSampler Source #

Arguments

:: forall v'1 m'. 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) sampled_candidates true_expected_count sampled_expected_count

logUniformCandidateSampler' Source #

Arguments

:: forall v'1 m'. 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) sampled_candidates true_expected_count sampled_expected_count

logicalAnd Source #

Arguments

:: Tensor v'1 Bool	x
-> Tensor v'2 Bool	y
-> Tensor Build Bool	z

logicalAnd' Source #

Arguments

:: OpParams
-> Tensor v'1 Bool	x
-> Tensor v'2 Bool	y
-> Tensor Build Bool	z

logicalNot Source #

Arguments

:: Tensor v'1 Bool	x
-> Tensor Build Bool	y

logicalNot' Source #

Arguments

:: OpParams
-> Tensor v'1 Bool	x
-> Tensor Build Bool	y

logicalOr Source #

Arguments

:: Tensor v'1 Bool	x
-> Tensor v'2 Bool	y
-> Tensor Build Bool	z

logicalOr' Source #

Arguments

:: OpParams
-> Tensor v'1 Bool	x
-> Tensor v'2 Bool	y
-> Tensor Build Bool	z

lookupTableExport Source #

Arguments

:: forall tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues)

=> Tensor Ref ByteString

table_handle

-> m' (Tensor Value tkeys, Tensor Value tvalues)

(keys, values)

keys
values

lookupTableExport' Source #

Arguments

:: forall tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues)
=> OpParams
-> Tensor Ref ByteString	table_handle
-> m' (Tensor Value tkeys, Tensor Value tvalues)	(keys, values) keys values

lookupTableExportV2 Source #

Arguments

:: forall v'1 tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues)

=> Tensor v'1 ResourceHandle

table_handle

-> m' (Tensor Value tkeys, Tensor Value tvalues)

(keys, values)

keys
values

lookupTableExportV2' Source #

Arguments

:: forall v'1 tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues)
=> OpParams
-> Tensor v'1 ResourceHandle	table_handle
-> m' (Tensor Value tkeys, Tensor Value tvalues)	(keys, values) keys values

lookupTableFind Source #

Arguments

:: forall v'2 v'3 tin tout m'. (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

lookupTableFind' Source #

Arguments

:: forall v'2 v'3 tin tout m'. (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

lookupTableFindV2 Source #

Arguments

:: forall v'1 v'2 v'3 tin tout m'. (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

lookupTableFindV2' Source #

Arguments

:: forall v'1 v'2 v'3 tin tout m'. (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

lookupTableImport Source #

Arguments

:: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout)
=> Tensor Ref ByteString	table_handle
-> Tensor v'2 tin	keys
-> Tensor v'3 tout	values
-> m' ControlNode

lookupTableImport' Source #

Arguments

:: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout)
=> OpParams
-> Tensor Ref ByteString	table_handle
-> Tensor v'2 tin	keys
-> Tensor v'3 tout	values
-> m' ControlNode

lookupTableImportV2 Source #

Arguments

:: forall v'1 v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout)
=> Tensor v'1 ResourceHandle	table_handle
-> Tensor v'2 tin	keys
-> Tensor v'3 tout	values
-> m' ControlNode

lookupTableImportV2' Source #

Arguments

:: forall v'1 v'2 v'3 tin tout m'. (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

lookupTableInsert Source #

Arguments

:: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout)
=> Tensor Ref ByteString	table_handle
-> Tensor v'2 tin	keys
-> Tensor v'3 tout	values
-> m' ControlNode

lookupTableInsert' Source #

Arguments

:: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout)
=> OpParams
-> Tensor Ref ByteString	table_handle
-> Tensor v'2 tin	keys
-> Tensor v'3 tout	values
-> m' ControlNode

lookupTableInsertV2 Source #

Arguments

:: forall v'1 v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout)
=> Tensor v'1 ResourceHandle	table_handle
-> Tensor v'2 tin	keys
-> Tensor v'3 tout	values
-> m' ControlNode

lookupTableInsertV2' Source #

Arguments

:: forall v'1 v'2 v'3 tin tout m'. (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

lookupTableRemoveV2 Source #

Arguments

:: forall v'1 v'2 tin m'. (MonadBuild m', TensorType tin)
=> Tensor v'1 ResourceHandle	table_handle
-> Tensor v'2 tin	keys
-> m' ControlNode

lookupTableRemoveV2' Source #

Arguments

:: forall v'1 v'2 tin m'. (MonadBuild m', TensorType tin)
=> OpParams
-> Tensor v'1 ResourceHandle	table_handle
-> Tensor v'2 tin	keys
-> m' ControlNode

lookupTableSize Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	table_handle
-> m' (Tensor Value Int64)	size

lookupTableSize' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	table_handle
-> m' (Tensor Value Int64)	size

lookupTableSizeV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	table_handle
-> m' (Tensor Value Int64)	size

lookupTableSizeV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	table_handle
-> m' (Tensor Value Int64)	size

loopCond Source #

Arguments

:: Tensor v'1 Bool	input
-> Tensor Build Bool	output

loopCond' Source #

Arguments

:: OpParams
-> Tensor v'1 Bool	input
-> Tensor Build Bool	output

lowerBound Source #

Arguments

:: forall v'1 v'2 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type)
=> Tensor v'1 t	sorted_inputs
-> Tensor v'2 t	values
-> Tensor Build out_type	output

lowerBound' Source #

Arguments

:: forall v'1 v'2 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type)
=> OpParams
-> Tensor v'1 t	sorted_inputs
-> Tensor v'2 t	values
-> Tensor Build out_type	output

lu Source #

Arguments

:: forall v'1 t output_idx_type. (OneOf '[Complex Double, Complex Float, Word16, Double, Float] t, OneOf '[Int32, Int64] output_idx_type)

=> Tensor v'1 t

input

-> (Tensor Build t, Tensor Build output_idx_type)

(lu, p)

lu
p

lu' Source #

Arguments

:: forall v'1 t output_idx_type. (OneOf '[Complex Double, Complex Float, Word16, Double, Float] t, OneOf '[Int32, Int64] output_idx_type)
=> OpParams
-> Tensor v'1 t	input
-> (Tensor Build t, Tensor Build output_idx_type)	(lu, p) lu p

makeIterator Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 Variant	dataset
-> Tensor v'2 ResourceHandle	iterator
-> m' ControlNode

makeIterator' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 Variant	dataset
-> Tensor v'2 ResourceHandle	iterator
-> m' ControlNode

mapClear Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	dtypes
-> m' ControlNode

mapClear' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	dtypes
-> m' ControlNode

mapIncompleteSize Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	dtypes
-> m' (Tensor Value Int32)	size

mapIncompleteSize' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	dtypes
-> m' (Tensor Value Int32)	size

mapPeek Source #

Arguments

:: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> Tensor v'1 Int64	key
-> Tensor v'2 Int32	indices
-> m' (TensorList Value dtypes)	values

mapPeek' Source #

Arguments

:: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> Tensor v'1 Int64	key
-> Tensor v'2 Int32	indices
-> m' (TensorList Value dtypes)	values

mapSize Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	dtypes
-> m' (Tensor Value Int32)	size

mapSize' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	dtypes
-> m' (Tensor Value Int32)	size

mapStage Source #

Arguments

:: forall v'1 v'2 v'3 fake_dtypes m'. (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

mapStage' Source #

Arguments

:: forall v'1 v'2 v'3 fake_dtypes m'. (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

mapUnstage Source #

Arguments

:: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> Tensor v'1 Int64	key
-> Tensor v'2 Int32	indices
-> m' (TensorList Value dtypes)	values

mapUnstage' Source #

Arguments

:: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> Tensor v'1 Int64	key
-> Tensor v'2 Int32	indices
-> m' (TensorList Value dtypes)	values

mapUnstageNoKey Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)

=> Tensor v'1 Int32

indices

-> m' (Tensor Value Int64, TensorList Value dtypes)

(key, values)

key
values

mapUnstageNoKey' Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> Tensor v'1 Int32	indices
-> m' (Tensor Value Int64, TensorList Value dtypes)	(key, values) key values

matMul Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> Tensor v'1 t	a
-> Tensor v'2 t	b
-> Tensor Build t	product

matMul' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	a
-> Tensor v'2 t	b
-> Tensor Build t	product

matchingFiles Source #

Arguments

:: Tensor v'1 ByteString	pattern
-> Tensor Build ByteString	filenames

matchingFiles' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	pattern
-> Tensor Build ByteString	filenames

matchingFilesDataset Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ByteString	patterns
-> m' (Tensor Value Variant)	handle

matchingFilesDataset' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	patterns
-> m' (Tensor Value Variant)	handle

matrixBandPart Source #

Arguments

:: forall v'1 v'2 v'3 t tindex. (TensorType t, OneOf '[Int32, Int64] tindex)
=> Tensor v'1 t	input
-> Tensor v'2 tindex	num_lower
-> Tensor v'3 tindex	num_upper
-> Tensor Build t	band

matrixBandPart' Source #

Arguments

:: forall v'1 v'2 v'3 t tindex. (TensorType t, OneOf '[Int32, Int64] tindex)
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 tindex	num_lower
-> Tensor v'3 tindex	num_upper
-> Tensor Build t	band

matrixDeterminant Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

matrixDeterminant' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

matrixDiag Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	diagonal
-> Tensor Build t	output

matrixDiag' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	diagonal
-> Tensor Build t	output

matrixDiagPart Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	diagonal

matrixDiagPart' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	diagonal

matrixDiagPartV2 Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 t	input
-> Tensor v'2 Int32	k
-> Tensor v'3 t	padding_value
-> Tensor Build t	diagonal

matrixDiagPartV2' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 Int32	k
-> Tensor v'3 t	padding_value
-> Tensor Build t	diagonal

matrixDiagPartV3 Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 t	input
-> Tensor v'2 Int32	k
-> Tensor v'3 t	padding_value
-> Tensor Build t	diagonal

matrixDiagPartV3' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 Int32	k
-> Tensor v'3 t	padding_value
-> Tensor Build t	diagonal

matrixDiagV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t
=> Tensor v'1 t	diagonal
-> Tensor v'2 Int32	k
-> Tensor v'3 Int32	num_rows
-> Tensor v'4 Int32	num_cols
-> Tensor v'5 t	padding_value
-> Tensor Build t	output

matrixDiagV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t
=> OpParams
-> Tensor v'1 t	diagonal
-> Tensor v'2 Int32	k
-> Tensor v'3 Int32	num_rows
-> Tensor v'4 Int32	num_cols
-> Tensor v'5 t	padding_value
-> Tensor Build t	output

matrixDiagV3 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t
=> Tensor v'1 t	diagonal
-> Tensor v'2 Int32	k
-> Tensor v'3 Int32	num_rows
-> Tensor v'4 Int32	num_cols
-> Tensor v'5 t	padding_value
-> Tensor Build t	output

matrixDiagV3' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t
=> OpParams
-> Tensor v'1 t	diagonal
-> Tensor v'2 Int32	k
-> Tensor v'3 Int32	num_rows
-> Tensor v'4 Int32	num_cols
-> Tensor v'5 t	padding_value
-> Tensor Build t	output

matrixExponential Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

matrixExponential' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

matrixInverse Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

matrixInverse' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

matrixLogarithm Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

matrixLogarithm' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

matrixSetDiag Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Tensor v'1 t	input
-> Tensor v'2 t	diagonal
-> Tensor Build t	output

matrixSetDiag' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 t	diagonal
-> Tensor Build t	output

matrixSetDiagV2 Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 t	input
-> Tensor v'2 t	diagonal
-> Tensor v'3 Int32	k
-> Tensor Build t	output

matrixSetDiagV2' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 t	diagonal
-> Tensor v'3 Int32	k
-> Tensor Build t	output

matrixSetDiagV3 Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 t	input
-> Tensor v'2 t	diagonal
-> Tensor v'3 Int32	k
-> Tensor Build t	output

matrixSetDiagV3' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 t	diagonal
-> Tensor v'3 Int32	k
-> Tensor Build t	output

matrixSolve Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor Build t	output

matrixSolve' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor Build t	output

matrixSolveLs Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor v'3 Double	l2_regularizer
-> Tensor Build t	output

matrixSolveLs' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor v'3 Double	l2_regularizer
-> Tensor Build t	output

matrixSquareRoot Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

matrixSquareRoot' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

matrixTriangularSolve Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor Build t	output

matrixTriangularSolve' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	matrix
-> Tensor v'2 t	rhs
-> Tensor Build t	output

max Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

max' Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

maxIntraOpParallelismDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	max_intra_op_parallelism
-> Tensor Build Variant	handle

maxIntraOpParallelismDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	max_intra_op_parallelism
-> Tensor Build Variant	handle

maxPool Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor Build t	output

maxPool' Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor Build t	output

maxPool3D Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor Build t	output

maxPool3D' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor Build t	output

maxPool3DGrad Source #

Arguments

:: forall v'1 v'2 v'3 t tInput. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] tInput)
=> ByteString	padding
-> Tensor v'1 tInput	orig_input
-> Tensor v'2 tInput	orig_output
-> Tensor v'3 t	grad
-> Tensor Build t	output

maxPool3DGrad' Source #

Arguments

:: forall v'1 v'2 v'3 t tInput. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] tInput)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tInput	orig_input
-> Tensor v'2 tInput	orig_output
-> Tensor v'3 t	grad
-> Tensor Build t	output

maxPool3DGradGrad Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	grad
-> Tensor Build t	output

maxPool3DGradGrad' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	grad
-> Tensor Build t	output

maxPoolGrad Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	grad
-> Tensor Build t	output

maxPoolGrad' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	grad
-> Tensor Build t	output

maxPoolGradGrad Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	grad
-> Tensor Build t	output

maxPoolGradGrad' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	grad
-> Tensor Build t	output

maxPoolGradGradV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	grad
-> Tensor v'4 Int32	ksize
-> Tensor v'5 Int32	strides
-> Tensor Build t	output

maxPoolGradGradV2' Source #

Arguments

:: 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
-> ByteString	padding
-> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	grad
-> Tensor v'4 Int32	ksize
-> Tensor v'5 Int32	strides
-> Tensor Build t	output

maxPoolGradGradWithArgmax Source #

Arguments

:: 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)
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	grad
-> Tensor v'3 targmax	argmax
-> Tensor Build t	output

maxPoolGradGradWithArgmax' Source #

Arguments

:: 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
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	grad
-> Tensor v'3 targmax	argmax
-> Tensor Build t	output

maxPoolGradV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	grad
-> Tensor v'4 Int32	ksize
-> Tensor v'5 Int32	strides
-> Tensor Build t	output

maxPoolGradV2' Source #

Arguments

:: 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
-> ByteString	padding
-> Tensor v'1 t	orig_input
-> Tensor v'2 t	orig_output
-> Tensor v'3 t	grad
-> Tensor v'4 Int32	ksize
-> Tensor v'5 Int32	strides
-> Tensor Build t	output

maxPoolGradWithArgmax Source #

Arguments

:: 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)
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	grad
-> Tensor v'3 targmax	argmax
-> Tensor Build t	output

maxPoolGradWithArgmax' Source #

Arguments

:: 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
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	grad
-> Tensor v'3 targmax	argmax
-> Tensor Build t	output

maxPoolV2 Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	ksize
-> Tensor v'3 Int32	strides
-> Tensor Build t	output

maxPoolV2' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Int32	ksize
-> Tensor v'3 Int32	strides
-> Tensor Build t	output

maxPoolWithArgmax Source #

Arguments

:: forall v'1 targmax t. (OneOf '[Int32, Int64] targmax, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t)
=> ByteString	padding
-> Tensor v'1 t	input
-> (Tensor Build t, Tensor Build targmax)	(output, argmax) output argmax

maxPoolWithArgmax' Source #

Arguments

:: forall v'1 targmax t. (OneOf '[Int32, Int64] targmax, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t)
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> (Tensor Build t, Tensor Build targmax)	(output, argmax) output argmax

maximum Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Word16, Word8, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

maximum' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Word16, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

mean Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

mean' Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

merge Source #

Arguments

:: forall v'1 t. TensorType t

=> [Tensor v'1 t]

inputs

-> (Tensor Build t, Tensor Build Int32)

(output, value_index)

output
value_index

merge' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> [Tensor v'1 t]	inputs
-> (Tensor Build t, Tensor Build Int32)	(output, value_index) output value_index

mergeSummary Source #

Arguments

:: [Tensor v'1 ByteString]	inputs
-> Tensor Build ByteString	summary

mergeSummary' Source #

Arguments

:: OpParams
-> [Tensor v'1 ByteString]	inputs
-> Tensor Build ByteString	summary

mergeV2Checkpoints Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ByteString	checkpoint_prefixes
-> Tensor v'2 ByteString	destination_prefix
-> m' ControlNode

mergeV2Checkpoints' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	checkpoint_prefixes
-> Tensor v'2 ByteString	destination_prefix
-> m' ControlNode

mfcc Source #

Arguments

:: Tensor v'1 Float	spectrogram
-> Tensor v'2 Int32	sample_rate
-> Tensor Build Float	output

mfcc' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	spectrogram
-> Tensor v'2 Int32	sample_rate
-> Tensor Build Float	output

min Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

min' Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

minimum Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Word16, Word8, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

minimum' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Word16, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

mirrorPad Source #

Arguments

:: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings)
=> ByteString	mode
-> Tensor v'1 t	input
-> Tensor v'2 tpaddings	paddings
-> Tensor Build t	output

mirrorPad' Source #

Arguments

:: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings)
=> OpParams
-> ByteString	mode
-> Tensor v'1 t	input
-> Tensor v'2 tpaddings	paddings
-> Tensor Build t	output

mirrorPadGrad Source #

Arguments

:: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings)
=> ByteString	mode
-> Tensor v'1 t	input
-> Tensor v'2 tpaddings	paddings
-> Tensor Build t	output

mirrorPadGrad' Source #

Arguments

:: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings)
=> OpParams
-> ByteString	mode
-> Tensor v'1 t	input
-> Tensor v'2 tpaddings	paddings
-> Tensor Build t	output

mlirPassthroughOp Source #

Arguments

:: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs)
=> ByteString	mlir_module
-> TensorList v'1 tinputs	inputs
-> TensorList Build toutputs	outputs

mlirPassthroughOp' Source #

Arguments

:: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs)
=> OpParams
-> ByteString	mlir_module
-> TensorList v'1 tinputs	inputs
-> TensorList Build toutputs	outputs

mod Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

mod' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

modelDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

modelDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

mul Source #

Arguments

:: forall v'1 v'2 t. 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 Build t	z

mul' Source #

Arguments

:: 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	x
-> Tensor v'2 t	y
-> Tensor Build t	z

mulNoNan Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

mulNoNan' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

multiDeviceIterator Source #

Arguments

:: forall m'. MonadBuild m'
=> ByteString	container
-> [DataType]	output_types
-> ByteString	shared_name
-> m' (Tensor Value ResourceHandle)	handle

multiDeviceIterator' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> ByteString	container
-> [DataType]	output_types
-> ByteString	shared_name
-> m' (Tensor Value ResourceHandle)	handle

multiDeviceIteratorFromStringHandle Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ByteString	string_handle
-> m' (Tensor Value ResourceHandle)	multi_device_iterator

multiDeviceIteratorFromStringHandle' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	string_handle
-> m' (Tensor Value ResourceHandle)	multi_device_iterator

multiDeviceIteratorGetNextFromShard Source #

Arguments

:: forall v'1 v'2 v'3 output_types m'. (MonadBuild m', TensorTypes output_types)
=> Tensor v'1 ResourceHandle	multi_device_iterator
-> Tensor v'2 Int32	shard_num
-> Tensor v'3 Int64	incarnation_id
-> m' (TensorList Value output_types)	components

multiDeviceIteratorGetNextFromShard' Source #

Arguments

:: forall v'1 v'2 v'3 output_types m'. (MonadBuild m', TensorTypes output_types)
=> OpParams
-> Tensor v'1 ResourceHandle	multi_device_iterator
-> Tensor v'2 Int32	shard_num
-> Tensor v'3 Int64	incarnation_id
-> m' (TensorList Value output_types)	components

multiDeviceIteratorInit Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 Variant	dataset
-> Tensor v'2 ResourceHandle	multi_device_iterator
-> Tensor v'3 Int64	max_buffer_size
-> m' (Tensor Value Int64)	incarnation_id

multiDeviceIteratorInit' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 Variant	dataset
-> Tensor v'2 ResourceHandle	multi_device_iterator
-> Tensor v'3 Int64	max_buffer_size
-> m' (Tensor Value Int64)	incarnation_id

multiDeviceIteratorToStringHandle Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	multi_device_iterator
-> m' (Tensor Value ByteString)	string_handle

multiDeviceIteratorToStringHandle' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	multi_device_iterator
-> m' (Tensor Value ByteString)	string_handle

multinomial Source #

Arguments

:: 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	logits
-> Tensor v'2 Int32	num_samples
-> m' (Tensor Value output_dtype)	output

multinomial' Source #

Arguments

:: 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	logits
-> Tensor v'2 Int32	num_samples
-> m' (Tensor Value output_dtype)	output

mutableDenseHashTable Source #

Arguments

:: forall v'1 key_dtype m'. (MonadBuild m', TensorType key_dtype)
=> DataType	value_dtype
-> Tensor v'1 key_dtype	empty_key
-> m' (Tensor Ref ByteString)	table_handle

mutableDenseHashTable' Source #

Arguments

:: forall v'1 key_dtype m'. (MonadBuild m', TensorType key_dtype)
=> OpParams
-> DataType	value_dtype
-> Tensor v'1 key_dtype	empty_key
-> m' (Tensor Ref ByteString)	table_handle

mutableDenseHashTableV2 Source #

Arguments

:: forall v'1 v'2 key_dtype m'. (MonadBuild m', TensorType key_dtype)
=> DataType	value_dtype
-> Tensor v'1 key_dtype	empty_key
-> Tensor v'2 key_dtype	deleted_key
-> m' (Tensor Value ResourceHandle)	table_handle

mutableDenseHashTableV2' Source #

Arguments

:: forall v'1 v'2 key_dtype m'. (MonadBuild m', TensorType key_dtype)
=> OpParams
-> DataType	value_dtype
-> Tensor v'1 key_dtype	empty_key
-> Tensor v'2 key_dtype	deleted_key
-> m' (Tensor Value ResourceHandle)	table_handle

mutableHashTable Source #

Arguments

:: forall m'. MonadBuild m'
=> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Ref ByteString)	table_handle

mutableHashTable' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Ref ByteString)	table_handle

mutableHashTableOfTensors Source #

Arguments

:: forall m'. MonadBuild m'
=> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Ref ByteString)	table_handle

mutableHashTableOfTensors' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Ref ByteString)	table_handle

mutableHashTableOfTensorsV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Value ResourceHandle)	table_handle

mutableHashTableOfTensorsV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Value ResourceHandle)	table_handle

mutableHashTableV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Value ResourceHandle)	table_handle

mutableHashTableV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> DataType	key_dtype
-> DataType	value_dtype
-> m' (Tensor Value ResourceHandle)	table_handle

mutexLock Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	mutex
-> m' (Tensor Value Variant)	mutex_lock

mutexLock' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	mutex
-> m' (Tensor Value Variant)	mutex_lock

mutexV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	resource

mutexV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	resource

ncclAllReduce Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> Int64	num_devices
-> ByteString	reduction
-> ByteString	shared_name
-> Tensor v'1 t	input
-> m' (Tensor Value t)	data

ncclAllReduce' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> OpParams
-> Int64	num_devices
-> ByteString	reduction
-> ByteString	shared_name
-> Tensor v'1 t	input
-> m' (Tensor Value t)	data

ncclBroadcast Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> Shape	shape
-> Tensor v'1 t	input
-> m' (Tensor Value t)	output

ncclBroadcast' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> OpParams
-> Shape	shape
-> Tensor v'1 t	input
-> m' (Tensor Value t)	output

ncclReduce Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> ByteString	reduction
-> [Tensor v'1 t]	input
-> m' (Tensor Value t)	data

ncclReduce' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> OpParams
-> ByteString	reduction
-> [Tensor v'1 t]	input
-> m' (Tensor Value t)	data

ndtri Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

ndtri' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

nearestNeighbors Source #

Arguments

:: Tensor v'1 Float	points
-> Tensor v'2 Float	centers
-> Tensor v'3 Int64	k
-> (Tensor Build Int64, Tensor Build Float)	(nearest_center_indices, nearest_center_distances) nearest_center_indices nearest_center_distances

nearestNeighbors' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	points
-> Tensor v'2 Float	centers
-> Tensor v'3 Int64	k
-> (Tensor Build Int64, Tensor Build Float)	(nearest_center_indices, nearest_center_distances) nearest_center_indices nearest_center_distances

neg Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

neg' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

negTrain Source #

Arguments

:: forall v'3 v'4 v'5 m'. MonadBuild m'
=> Int64	num_negative_samples
-> Tensor Ref Float	w_in
-> Tensor Ref Float	w_out
-> Tensor v'3 Int32	examples
-> Tensor v'4 Int32	labels
-> Tensor v'5 Float	lr
-> m' ControlNode

negTrain' Source #

Arguments

:: forall v'3 v'4 v'5 m'. MonadBuild m'
=> OpParams
-> Int64	num_negative_samples
-> Tensor Ref Float	w_in
-> Tensor Ref Float	w_out
-> Tensor v'3 Int32	examples
-> Tensor v'4 Int32	labels
-> Tensor v'5 Float	lr
-> m' ControlNode

nextAfter Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 t	x1
-> Tensor v'2 t	x2
-> Tensor Build t	output

nextAfter' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	x1
-> Tensor v'2 t	x2
-> Tensor Build t	output

nextIteration Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	data
-> Tensor Build t	output

nextIteration' Source #

Arguments

:: forall v'1 t. 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 #

nonDeterministicInts Source #

Arguments

:: forall v'1 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> Tensor v'1 shape_dtype	shape
-> m' (Tensor Value dtype)	output

nonDeterministicInts' Source #

Arguments

:: forall v'1 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> OpParams
-> Tensor v'1 shape_dtype	shape
-> m' (Tensor Value dtype)	output

nonMaxSuppression Source #

Arguments

:: Tensor v'1 Float	boxes
-> Tensor v'2 Float	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor Build Int32	selected_indices

nonMaxSuppression' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	boxes
-> Tensor v'2 Float	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor Build Int32	selected_indices

nonMaxSuppressionV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold)
=> Tensor v'1 t	boxes
-> Tensor v'2 t	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor v'4 t_threshold	iou_threshold
-> Tensor Build Int32	selected_indices

nonMaxSuppressionV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold)
=> OpParams
-> Tensor v'1 t	boxes
-> Tensor v'2 t	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor v'4 t_threshold	iou_threshold
-> Tensor Build Int32	selected_indices

nonMaxSuppressionV3 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold)
=> Tensor v'1 t	boxes
-> Tensor v'2 t	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor v'4 t_threshold	iou_threshold
-> Tensor v'5 t_threshold	score_threshold
-> Tensor Build Int32	selected_indices

nonMaxSuppressionV3' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold)
=> OpParams
-> Tensor v'1 t	boxes
-> Tensor v'2 t	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor v'4 t_threshold	iou_threshold
-> Tensor v'5 t_threshold	score_threshold
-> Tensor Build Int32	selected_indices

nonMaxSuppressionV4 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold)
=> Tensor v'1 t	boxes
-> Tensor v'2 t	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor v'4 t_threshold	iou_threshold
-> Tensor v'5 t_threshold	score_threshold
-> (Tensor Build Int32, Tensor Build Int32)	(selected_indices, valid_outputs) selected_indices valid_outputs

nonMaxSuppressionV4' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold)
=> OpParams
-> Tensor v'1 t	boxes
-> Tensor v'2 t	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor v'4 t_threshold	iou_threshold
-> Tensor v'5 t_threshold	score_threshold
-> (Tensor Build Int32, Tensor Build Int32)	(selected_indices, valid_outputs) selected_indices valid_outputs

nonMaxSuppressionV5 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Word16, Float] t
=> Tensor v'1 t	boxes
-> Tensor v'2 t	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor v'4 t	iou_threshold
-> Tensor v'5 t	score_threshold
-> Tensor v'6 t	soft_nms_sigma
-> (Tensor Build Int32, Tensor Build t, Tensor Build Int32)	(selected_indices, selected_scores, valid_outputs) selected_indices selected_scores valid_outputs

nonMaxSuppressionV5' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Word16, Float] t
=> OpParams
-> Tensor v'1 t	boxes
-> Tensor v'2 t	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor v'4 t	iou_threshold
-> Tensor v'5 t	score_threshold
-> Tensor v'6 t	soft_nms_sigma
-> (Tensor Build Int32, Tensor Build t, Tensor Build Int32)	(selected_indices, selected_scores, valid_outputs) selected_indices selected_scores valid_outputs

nonMaxSuppressionWithOverlaps Source #

Arguments

:: Tensor v'1 Float	overlaps
-> Tensor v'2 Float	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor v'4 Float	overlap_threshold
-> Tensor v'5 Float	score_threshold
-> Tensor Build Int32	selected_indices

nonMaxSuppressionWithOverlaps' Source #

Arguments

:: OpParams
-> Tensor v'1 Float	overlaps
-> Tensor v'2 Float	scores
-> Tensor v'3 Int32	max_output_size
-> Tensor v'4 Float	overlap_threshold
-> Tensor v'5 Float	score_threshold
-> Tensor Build Int32	selected_indices

nonSerializableDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

nonSerializableDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

notEqual Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

notEqual' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build Bool	z

nthElement Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	input
-> Tensor v'2 Int32	n
-> Tensor Build t	values

nthElement' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 Int32	n
-> Tensor Build t	values

oneHot Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tI. (TensorType t, OneOf '[Int32, Int64, Word8] tI)
=> Tensor v'1 tI	indices
-> Tensor v'2 Int32	depth
-> Tensor v'3 t	on_value
-> Tensor v'4 t	off_value
-> Tensor Build t	output

oneHot' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tI. (TensorType t, OneOf '[Int32, Int64, Word8] tI)
=> OpParams
-> Tensor v'1 tI	indices
-> Tensor v'2 Int32	depth
-> Tensor v'3 t	on_value
-> Tensor v'4 t	off_value
-> Tensor Build t	output

onesLike Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

onesLike' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

optimizeDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	optimizations
-> Tensor Build Variant	handle

optimizeDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	optimizations
-> Tensor Build Variant	handle

optionalFromValue Source #

Arguments

:: forall v'1 toutput_types. TensorTypes toutput_types
=> TensorList v'1 toutput_types	components
-> Tensor Build Variant	optional

optionalFromValue' Source #

Arguments

:: forall v'1 toutput_types. TensorTypes toutput_types
=> OpParams
-> TensorList v'1 toutput_types	components
-> Tensor Build Variant	optional

optionalGetValue Source #

Arguments

:: forall v'1 output_types. TensorTypes output_types
=> Tensor v'1 Variant	optional
-> TensorList Build output_types	components

optionalGetValue' Source #

Arguments

:: forall v'1 output_types. TensorTypes output_types
=> OpParams
-> Tensor v'1 Variant	optional
-> TensorList Build output_types	components

optionalHasValue Source #

Arguments

:: Tensor v'1 Variant	optional
-> Tensor Build Bool	has_value

optionalHasValue' Source #

Arguments

:: OpParams
-> Tensor v'1 Variant	optional
-> Tensor Build Bool	has_value

optionalNone Source #

Arguments

:: Tensor Build Variant

optional

optionalNone' Source #

Arguments

:: OpParams
-> Tensor Build Variant	optional

orderedMapClear Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	dtypes
-> m' ControlNode

orderedMapClear' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	dtypes
-> m' ControlNode

orderedMapIncompleteSize Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	dtypes
-> m' (Tensor Value Int32)	size

orderedMapIncompleteSize' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	dtypes
-> m' (Tensor Value Int32)	size

orderedMapPeek Source #

Arguments

:: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> Tensor v'1 Int64	key
-> Tensor v'2 Int32	indices
-> m' (TensorList Value dtypes)	values

orderedMapPeek' Source #

Arguments

:: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> Tensor v'1 Int64	key
-> Tensor v'2 Int32	indices
-> m' (TensorList Value dtypes)	values

orderedMapSize Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	dtypes
-> m' (Tensor Value Int32)	size

orderedMapSize' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	dtypes
-> m' (Tensor Value Int32)	size

orderedMapStage Source #

Arguments

:: forall v'1 v'2 v'3 fake_dtypes m'. (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

orderedMapStage' Source #

Arguments

:: forall v'1 v'2 v'3 fake_dtypes m'. (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

orderedMapUnstage Source #

Arguments

:: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> Tensor v'1 Int64	key
-> Tensor v'2 Int32	indices
-> m' (TensorList Value dtypes)	values

orderedMapUnstage' Source #

Arguments

:: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> Tensor v'1 Int64	key
-> Tensor v'2 Int32	indices
-> m' (TensorList Value dtypes)	values

orderedMapUnstageNoKey Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)

=> Tensor v'1 Int32

indices

-> m' (Tensor Value Int64, TensorList Value dtypes)

(key, values)

key
values

orderedMapUnstageNoKey' Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> Tensor v'1 Int32	indices
-> m' (Tensor Value Int64, TensorList Value dtypes)	(key, values) key values

outfeedDequeue Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> Shape	shape
-> m' (Tensor Value dtype)	output

outfeedDequeue' Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Shape	shape
-> m' (Tensor Value dtype)	output

outfeedDequeueTuple Source #

Arguments

:: forall dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> m' (TensorList Value dtypes)	outputs

outfeedDequeueTuple' Source #

Arguments

:: forall dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> m' (TensorList Value dtypes)	outputs

outfeedEnqueue Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor v'1 dtype	input
-> m' ControlNode

outfeedEnqueue' Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor v'1 dtype	input
-> m' ControlNode

outfeedEnqueueTuple Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> TensorList v'1 dtypes	inputs
-> m' ControlNode

outfeedEnqueueTuple' Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> TensorList v'1 dtypes	inputs
-> m' ControlNode

pack Source #

Arguments

:: forall v'1 t. TensorType t
=> [Tensor v'1 t]	values
-> Tensor Build t	output

pack' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> [Tensor v'1 t]	values
-> Tensor Build t	output

pad Source #

Arguments

:: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings)
=> Tensor v'1 t	input
-> Tensor v'2 tpaddings	paddings
-> Tensor Build t	output

pad' Source #

Arguments

:: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings)
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 tpaddings	paddings
-> Tensor Build t	output

padV2 Source #

Arguments

:: forall v'1 v'2 v'3 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings)
=> Tensor v'1 t	input
-> Tensor v'2 tpaddings	paddings
-> Tensor v'3 t	constant_values
-> Tensor Build t	output

padV2' Source #

Arguments

:: forall v'1 v'2 v'3 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings)
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 tpaddings	paddings
-> Tensor v'3 t	constant_values
-> Tensor Build t	output

paddedBatchDataset Source #

Arguments

:: forall v'1 v'2 v'3 v'4 toutput_types. TensorTypes toutput_types
=> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> [Tensor v'3 Int64]	padded_shapes
-> TensorList v'4 toutput_types	padding_values
-> Tensor Build Variant	handle

paddedBatchDataset' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 toutput_types. TensorTypes toutput_types
=> OpParams
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> [Tensor v'3 Int64]	padded_shapes
-> TensorList v'4 toutput_types	padding_values
-> Tensor Build Variant	handle

paddedBatchDatasetV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 toutput_types. TensorTypes toutput_types
=> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> [Tensor v'3 Int64]	padded_shapes
-> TensorList v'4 toutput_types	padding_values
-> Tensor v'5 Bool	drop_remainder
-> Tensor Build Variant	handle

paddedBatchDatasetV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 toutput_types. TensorTypes toutput_types
=> OpParams
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	batch_size
-> [Tensor v'3 Int64]	padded_shapes
-> TensorList v'4 toutput_types	padding_values
-> Tensor v'5 Bool	drop_remainder
-> Tensor Build Variant	handle

paddingFIFOQueue Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	component_types
-> m' (Tensor Ref ByteString)	handle

paddingFIFOQueue' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	component_types
-> m' (Tensor Ref ByteString)	handle

paddingFIFOQueueV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	component_types
-> m' (Tensor Value ResourceHandle)	handle

paddingFIFOQueueV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	component_types
-> m' (Tensor Value ResourceHandle)	handle

parallelConcat Source #

Arguments

:: forall v'1 t. TensorType t
=> Shape	shape
-> [Tensor v'1 t]	values
-> Tensor Build t	output

parallelConcat' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Shape	shape
-> [Tensor v'1 t]	values
-> Tensor Build t	output

parallelDynamicStitch Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> [Tensor v'1 Int32]	indices
-> [Tensor v'2 t]	data
-> Tensor Build t	merged

parallelDynamicStitch' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> [Tensor v'1 Int32]	indices
-> [Tensor v'2 t]	data
-> Tensor Build t	merged

parameterizedTruncatedNormal Source #

Arguments

:: 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	shape
-> Tensor v'2 dtype	means
-> Tensor v'3 dtype	stdevs
-> Tensor v'4 dtype	minvals
-> Tensor v'5 dtype	maxvals
-> m' (Tensor Value dtype)	output

parameterizedTruncatedNormal' Source #

Arguments

:: 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	shape
-> Tensor v'2 dtype	means
-> Tensor v'3 dtype	stdevs
-> Tensor v'4 dtype	minvals
-> Tensor v'5 dtype	maxvals
-> m' (Tensor Value dtype)	output

parseExample Source #

Arguments

:: 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	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) sparse_indices sparse_values sparse_shapes dense_values

parseExample' Source #

Arguments

:: 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	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) sparse_indices sparse_values sparse_shapes dense_values

parseExampleDataset Source #

Arguments

:: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense
=> [DataType]	output_types
-> [DataType]	sparse_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_parallel_calls
-> TensorList v'3 tdense	dense_defaults
-> Tensor Build Variant	handle

parseExampleDataset' Source #

Arguments

:: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense
=> OpParams
-> [DataType]	output_types
-> [DataType]	sparse_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_parallel_calls
-> TensorList v'3 tdense	dense_defaults
-> Tensor Build Variant	handle

parseExampleDatasetV2 Source #

Arguments

:: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense
=> [DataType]	output_types
-> [DataType]	sparse_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_parallel_calls
-> TensorList v'3 tdense	dense_defaults
-> Tensor Build Variant	handle

parseExampleDatasetV2' Source #

Arguments

:: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense
=> OpParams
-> [DataType]	output_types
-> [DataType]	sparse_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_parallel_calls
-> TensorList v'3 tdense	dense_defaults
-> Tensor Build Variant	handle

parseExampleV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 tdense sparse_types ragged_value_types ragged_split_types. (OneOfs '[ByteString, Int64, Float] tdense, OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] ragged_value_types, OneOfs '[Int32, Int64] ragged_split_types)
=> Int64	num_sparse
-> Tensor v'1 ByteString	serialized
-> Tensor v'2 ByteString	names
-> Tensor v'3 ByteString	sparse_keys
-> Tensor v'4 ByteString	dense_keys
-> Tensor v'5 ByteString	ragged_keys
-> TensorList v'6 tdense	dense_defaults
-> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense, TensorList Build ragged_value_types, TensorList Build ragged_split_types)	(sparse_indices, sparse_values, sparse_shapes, dense_values, ragged_values, ragged_row_splits) sparse_indices sparse_values sparse_shapes dense_values ragged_values ragged_row_splits

parseExampleV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 tdense sparse_types ragged_value_types ragged_split_types. (OneOfs '[ByteString, Int64, Float] tdense, OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] ragged_value_types, OneOfs '[Int32, Int64] ragged_split_types)
=> OpParams
-> Int64	num_sparse
-> Tensor v'1 ByteString	serialized
-> Tensor v'2 ByteString	names
-> Tensor v'3 ByteString	sparse_keys
-> Tensor v'4 ByteString	dense_keys
-> Tensor v'5 ByteString	ragged_keys
-> TensorList v'6 tdense	dense_defaults
-> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense, TensorList Build ragged_value_types, TensorList Build ragged_split_types)	(sparse_indices, sparse_values, sparse_shapes, dense_values, ragged_values, ragged_row_splits) sparse_indices sparse_values sparse_shapes dense_values ragged_values ragged_row_splits

parseSingleExample Source #

Arguments

:: forall v'1 v'2 sparse_types tdense. (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) sparse_indices sparse_values sparse_shapes dense_values

parseSingleExample' Source #

Arguments

:: forall v'1 v'2 sparse_types tdense. (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) sparse_indices sparse_values sparse_shapes dense_values

parseTensor Source #

Arguments

:: forall v'1 out_type. TensorType out_type
=> Tensor v'1 ByteString	serialized
-> Tensor Build out_type	output

parseTensor' Source #

Arguments

:: forall v'1 out_type. TensorType out_type
=> OpParams
-> Tensor v'1 ByteString	serialized
-> Tensor Build out_type	output

placeholder Source #

Arguments

:: forall dtype. TensorType dtype
=> Tensor Build dtype	output

placeholder' Source #

Arguments

:: forall dtype. TensorType dtype
=> OpParams
-> Tensor Build dtype	output

placeholderV2 Source #

Arguments

:: forall dtype. TensorType dtype
=> Shape	shape
-> Tensor Build dtype	output

placeholderV2' Source #

Arguments

:: forall dtype. TensorType dtype
=> OpParams
-> Shape	shape
-> Tensor Build dtype	output

placeholderWithDefault Source #

Arguments

:: forall v'1 dtype. TensorType dtype
=> Shape	shape
-> Tensor v'1 dtype	input
-> Tensor Build dtype	output

placeholderWithDefault' Source #

Arguments

:: forall v'1 dtype. TensorType dtype
=> OpParams
-> Shape	shape
-> Tensor v'1 dtype	input
-> Tensor Build dtype	output

polygamma Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 t	a
-> Tensor v'2 t	x
-> Tensor Build t	z

polygamma' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	a
-> Tensor v'2 t	x
-> Tensor Build t	z

populationCount Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> Tensor v'1 t	x
-> Tensor Build Word8	y

populationCount' Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build Word8	y

pow Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

pow' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

prefetchDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor Build Variant	handle

prefetchDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor Build Variant	handle

prelinearize Source #

Arguments

:: forall v'1 dtype. TensorType dtype
=> Tensor v'1 dtype	input
-> Tensor Build Variant	output

prelinearize' Source #

Arguments

:: forall v'1 dtype. TensorType dtype
=> OpParams
-> Tensor v'1 dtype	input
-> Tensor Build Variant	output

prelinearizeTuple Source #

Arguments

:: forall v'1 dtypes. TensorTypes dtypes
=> TensorList v'1 dtypes	inputs
-> Tensor Build Variant	output

prelinearizeTuple' Source #

Arguments

:: forall v'1 dtypes. TensorTypes dtypes
=> OpParams
-> TensorList v'1 dtypes	inputs
-> Tensor Build Variant	output

preventGradient Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	output

preventGradient' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

print Source #

Arguments

:: forall v'1 v'2 t u m'. (MonadBuild m', TensorType t, TensorTypes u)
=> Tensor v'1 t	input
-> TensorList v'2 u	data
-> m' (Tensor Value t)	output

print' Source #

Arguments

:: forall v'1 v'2 t u m'. (MonadBuild m', TensorType t, TensorTypes u)
=> OpParams
-> Tensor v'1 t	input
-> TensorList v'2 u	data
-> m' (Tensor Value t)	output

printV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ByteString	input
-> m' ControlNode

printV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	input
-> m' ControlNode

priorityQueue Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Ref ByteString)	handle

priorityQueue' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Ref ByteString)	handle

priorityQueueV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	handle

priorityQueueV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	handle

privateThreadPoolDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_threads
-> Tensor Build Variant	handle

privateThreadPoolDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_threads
-> Tensor Build Variant	handle

prod Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

prod' Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

pyFunc Source #

Arguments

:: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout)
=> ByteString	token
-> TensorList v'1 tin	input
-> m' (TensorList Value tout)	output

pyFunc' Source #

Arguments

:: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout)
=> OpParams
-> ByteString	token
-> TensorList v'1 tin	input
-> m' (TensorList Value tout)	output

pyFuncStateless Source #

Arguments

:: forall v'1 tin tout. (TensorTypes tin, TensorTypes tout)
=> ByteString	token
-> TensorList v'1 tin	input
-> TensorList Build tout	output

pyFuncStateless' Source #

Arguments

:: forall v'1 tin tout. (TensorTypes tin, TensorTypes tout)
=> OpParams
-> ByteString	token
-> TensorList v'1 tin	input
-> TensorList Build tout	output

qr Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t

=> Tensor v'1 t

input

-> (Tensor Build t, Tensor Build t)

(q, r)

q
r

qr' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> (Tensor Build t, Tensor Build t)	(q, r) q r

quantizeAndDequantize Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

quantizeAndDequantize' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

quantizeAndDequantizeV2 Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	input
-> Tensor v'2 t	input_min
-> Tensor v'3 t	input_max
-> Tensor Build t	output

quantizeAndDequantizeV2' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 t	input_min
-> Tensor v'3 t	input_max
-> Tensor Build t	output

quantizeAndDequantizeV3 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	input
-> Tensor v'2 t	input_min
-> Tensor v'3 t	input_max
-> Tensor v'4 Int32	num_bits
-> Tensor Build t	output

quantizeAndDequantizeV3' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 t	input_min
-> Tensor v'3 t	input_max
-> Tensor v'4 Int32	num_bits
-> Tensor Build t	output

quantizeDownAndShrinkRange Source #

Arguments

:: 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	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) output output_min output_max

quantizeDownAndShrinkRange' Source #

Arguments

:: 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	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) output output_min output_max

quantizeV2 Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t
=> Tensor v'1 Float	input
-> Tensor v'2 Float	min_range
-> Tensor v'3 Float	max_range
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(output, output_min, output_max) output output_min output_max

quantizeV2' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t
=> OpParams
-> Tensor v'1 Float	input
-> Tensor v'2 Float	min_range
-> Tensor v'3 Float	max_range
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(output, output_min, output_max) output output_min output_max

quantizedAdd Source #

Arguments

:: 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	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) z min_z max_z

quantizedAdd' Source #

Arguments

:: 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	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) z min_z max_z

quantizedAvgPool Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Float	min_input
-> Tensor v'3 Float	max_input
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedAvgPool' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Float	min_input
-> Tensor v'3 Float	max_input
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedBatchNormWithGlobalNormalization Source #

Arguments

:: 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	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) result result_min result_max

quantizedBatchNormWithGlobalNormalization' Source #

Arguments

:: 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	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) result result_min result_max

quantizedBiasAdd Source #

Arguments

:: 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	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) output min_out max_out

quantizedBiasAdd' Source #

Arguments

:: 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	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) output min_out max_out

quantizedConcat Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> Tensor v'1 Int32	concat_dim
-> [Tensor v'2 t]	values
-> [Tensor v'3 Float]	input_mins
-> [Tensor v'4 Float]	input_maxes
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(output, output_min, output_max) output output_min output_max

quantizedConcat' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> OpParams
-> Tensor v'1 Int32	concat_dim
-> [Tensor v'2 t]	values
-> [Tensor v'3 Float]	input_mins
-> [Tensor v'4 Float]	input_maxes
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(output, output_min, output_max) output output_min output_max

quantizedConv2D Source #

Arguments

:: 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)
=> ByteString	padding
-> 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) output min_output max_output

quantizedConv2D' Source #

Arguments

:: 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
-> ByteString	padding
-> 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) output min_output max_output

quantizedConv2DAndRelu Source #

Arguments

:: 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)
=> ByteString	padding
-> 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) output min_output max_output

quantizedConv2DAndRelu' Source #

Arguments

:: 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
-> ByteString	padding
-> 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) output min_output max_output

quantizedConv2DAndReluAndRequantize Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> 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 v'7 Float	min_freezed_output
-> Tensor v'8 Float	max_freezed_output
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DAndReluAndRequantize' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> 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 v'7 Float	min_freezed_output
-> Tensor v'8 Float	max_freezed_output
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DAndRequantize Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> 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 v'7 Float	min_freezed_output
-> Tensor v'8 Float	max_freezed_output
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DAndRequantize' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> 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 v'7 Float	min_freezed_output
-> Tensor v'8 Float	max_freezed_output
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DPerChannel Source #

Arguments

:: 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)
=> ByteString	padding
-> 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) output min_output max_output

quantizedConv2DPerChannel' Source #

Arguments

:: 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
-> ByteString	padding
-> 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) output min_output max_output

quantizedConv2DWithBias Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBias' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasAndRelu Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasAndRelu' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasAndReluAndRequantize Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasAndReluAndRequantize' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasAndRequantize Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasAndRequantize' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasSignedSumAndReluAndRequantize Source #

Arguments

:: 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 tinput tfilter tbias tsummand out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] tsummand, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> Tensor v'10 tsummand	summand
-> Tensor v'11 Float	min_summand
-> Tensor v'12 Float	max_summand
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasSignedSumAndReluAndRequantize' Source #

Arguments

:: 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 tinput tfilter tbias tsummand out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] tsummand, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> Tensor v'10 tsummand	summand
-> Tensor v'11 Float	min_summand
-> Tensor v'12 Float	max_summand
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasSumAndRelu Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	summand
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasSumAndRelu' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	summand
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasSumAndReluAndRequantize Source #

Arguments

:: 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 tinput tfilter tbias tsummand out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] tsummand, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> Tensor v'10 tsummand	summand
-> Tensor v'11 Float	min_summand
-> Tensor v'12 Float	max_summand
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedConv2DWithBiasSumAndReluAndRequantize' Source #

Arguments

:: 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 tinput tfilter tbias tsummand out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] tsummand, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> Tensor v'10 tsummand	summand
-> Tensor v'11 Float	min_summand
-> Tensor v'12 Float	max_summand
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedDepthwiseConv2D Source #

Arguments

:: 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)
=> ByteString	padding
-> 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) output min_output max_output

quantizedDepthwiseConv2D' Source #

Arguments

:: 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
-> ByteString	padding
-> 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) output min_output max_output

quantizedDepthwiseConv2DWithBias Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedDepthwiseConv2DWithBias' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedDepthwiseConv2DWithBiasAndRelu Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedDepthwiseConv2DWithBiasAndRelu' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedDepthwiseConv2DWithBiasAndReluAndRequantize Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedDepthwiseConv2DWithBiasAndReluAndRequantize' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 tinput tfilter tbias out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> ByteString	padding
-> Tensor v'1 tinput	input
-> Tensor v'2 tfilter	filter
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_input
-> Tensor v'5 Float	max_input
-> Tensor v'6 Float	min_filter
-> Tensor v'7 Float	max_filter
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedInstanceNorm Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t
=> Tensor v'1 t	x
-> Tensor v'2 Float	x_min
-> Tensor v'3 Float	x_max
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(y, y_min, y_max) y y_min y_max

quantizedInstanceNorm' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 Float	x_min
-> Tensor v'3 Float	x_max
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(y, y_min, y_max) y y_min y_max

quantizedMatMul Source #

Arguments

:: 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	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) out min_out max_out

quantizedMatMul' Source #

Arguments

:: 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	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) out min_out max_out

quantizedMatMulWithBias Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput)
=> Tensor v'1 t1	a
-> Tensor v'2 t2	b
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_a
-> Tensor v'5 Float	max_a
-> Tensor v'6 Float	min_b
-> Tensor v'7 Float	max_b
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)	(out, min_out, max_out) out min_out max_out

quantizedMatMulWithBias' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput)
=> OpParams
-> Tensor v'1 t1	a
-> Tensor v'2 t2	b
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_a
-> Tensor v'5 Float	max_a
-> Tensor v'6 Float	min_b
-> Tensor v'7 Float	max_b
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)	(out, min_out, max_out) out min_out max_out

quantizedMatMulWithBiasAndDequantize Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Float] toutput)
=> Tensor v'1 t1	a
-> Tensor v'2 t2	b
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_a
-> Tensor v'5 Float	max_a
-> Tensor v'6 Float	min_b
-> Tensor v'7 Float	max_b
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> Tensor Build toutput	out

quantizedMatMulWithBiasAndDequantize' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Float] toutput)
=> OpParams
-> Tensor v'1 t1	a
-> Tensor v'2 t2	b
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_a
-> Tensor v'5 Float	max_a
-> Tensor v'6 Float	min_b
-> Tensor v'7 Float	max_b
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> Tensor Build toutput	out

quantizedMatMulWithBiasAndRelu Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 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	a
-> Tensor v'2 t2	b
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_a
-> Tensor v'5 Float	max_a
-> Tensor v'6 Float	min_b
-> Tensor v'7 Float	max_b
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)	(out, min_out, max_out) out min_out max_out

quantizedMatMulWithBiasAndRelu' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 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	a
-> Tensor v'2 t2	b
-> Tensor v'3 Float	bias
-> Tensor v'4 Float	min_a
-> Tensor v'5 Float	max_a
-> Tensor v'6 Float	min_b
-> Tensor v'7 Float	max_b
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)	(out, min_out, max_out) out min_out max_out

quantizedMatMulWithBiasAndReluAndRequantize Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput)
=> Tensor v'1 t1	a
-> Tensor v'2 t2	b
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_a
-> Tensor v'5 Float	max_a
-> Tensor v'6 Float	min_b
-> Tensor v'7 Float	max_b
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)	(out, min_out, max_out) out min_out max_out

quantizedMatMulWithBiasAndReluAndRequantize' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput)
=> OpParams
-> Tensor v'1 t1	a
-> Tensor v'2 t2	b
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_a
-> Tensor v'5 Float	max_a
-> Tensor v'6 Float	min_b
-> Tensor v'7 Float	max_b
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)	(out, min_out, max_out) out min_out max_out

quantizedMatMulWithBiasAndRequantize Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput)
=> Tensor v'1 t1	a
-> Tensor v'2 t2	b
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_a
-> Tensor v'5 Float	max_a
-> Tensor v'6 Float	min_b
-> Tensor v'7 Float	max_b
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)	(out, min_out, max_out) out min_out max_out

quantizedMatMulWithBiasAndRequantize' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t1 t2 tbias toutput. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput)
=> OpParams
-> Tensor v'1 t1	a
-> Tensor v'2 t2	b
-> Tensor v'3 tbias	bias
-> Tensor v'4 Float	min_a
-> Tensor v'5 Float	max_a
-> Tensor v'6 Float	min_b
-> Tensor v'7 Float	max_b
-> Tensor v'8 Float	min_freezed_output
-> Tensor v'9 Float	max_freezed_output
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)	(out, min_out, max_out) out min_out max_out

quantizedMaxPool Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Float	min_input
-> Tensor v'3 Float	max_input
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedMaxPool' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 Float	min_input
-> Tensor v'3 Float	max_input
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(output, min_output, max_output) output min_output max_output

quantizedMul Source #

Arguments

:: 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	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) z min_z max_z

quantizedMul' Source #

Arguments

:: 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	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) z min_z max_z

quantizedRelu Source #

Arguments

:: 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	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) activations min_activations max_activations

quantizedRelu' Source #

Arguments

:: 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	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) activations min_activations max_activations

quantizedRelu6 Source #

Arguments

:: 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	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) activations min_activations max_activations

quantizedRelu6' Source #

Arguments

:: 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	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) activations min_activations max_activations

quantizedReluX Source #

Arguments

:: 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	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) activations min_activations max_activations

quantizedReluX' Source #

Arguments

:: 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	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) activations min_activations max_activations

quantizedReshape Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tshape. (TensorType t, OneOf '[Int32, Int64] tshape)
=> Tensor v'1 t	tensor
-> Tensor v'2 tshape	shape
-> Tensor v'3 Float	input_min
-> Tensor v'4 Float	input_max
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(output, output_min, output_max) output output_min output_max

quantizedReshape' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tshape. (TensorType t, OneOf '[Int32, Int64] tshape)
=> OpParams
-> Tensor v'1 t	tensor
-> Tensor v'2 tshape	shape
-> Tensor v'3 Float	input_min
-> Tensor v'4 Float	input_max
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(output, output_min, output_max) output output_min output_max

quantizedResizeBilinear Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Word8, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Int32	size
-> Tensor v'3 Float	min
-> Tensor v'4 Float	max
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(resized_images, out_min, out_max) resized_images out_min out_max

quantizedResizeBilinear' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Word8, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor v'2 Int32	size
-> Tensor v'3 Float	min
-> Tensor v'4 Float	max
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)	(resized_images, out_min, out_max) resized_images out_min out_max

queueClose Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	handle
-> m' ControlNode

queueClose' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	handle
-> m' ControlNode

queueCloseV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> m' ControlNode

queueCloseV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> m' ControlNode

queueDequeue Source #

Arguments

:: forall component_types m'. (MonadBuild m', TensorTypes component_types)
=> Tensor Ref ByteString	handle
-> m' (TensorList Value component_types)	components

queueDequeue' Source #

Arguments

:: forall component_types m'. (MonadBuild m', TensorTypes component_types)
=> OpParams
-> Tensor Ref ByteString	handle
-> m' (TensorList Value component_types)	components

queueDequeueMany Source #

Arguments

:: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types)
=> Tensor Ref ByteString	handle
-> Tensor v'2 Int32	n
-> m' (TensorList Value component_types)	components

queueDequeueMany' Source #

Arguments

:: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types)
=> OpParams
-> Tensor Ref ByteString	handle
-> Tensor v'2 Int32	n
-> m' (TensorList Value component_types)	components

queueDequeueManyV2 Source #

Arguments

:: forall v'1 v'2 component_types m'. (MonadBuild m', TensorTypes component_types)
=> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Int32	n
-> m' (TensorList Value component_types)	components

queueDequeueManyV2' Source #

Arguments

:: forall v'1 v'2 component_types m'. (MonadBuild m', TensorTypes component_types)
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Int32	n
-> m' (TensorList Value component_types)	components

queueDequeueUpTo Source #

Arguments

:: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types)
=> Tensor Ref ByteString	handle
-> Tensor v'2 Int32	n
-> m' (TensorList Value component_types)	components

queueDequeueUpTo' Source #

Arguments

:: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types)
=> OpParams
-> Tensor Ref ByteString	handle
-> Tensor v'2 Int32	n
-> m' (TensorList Value component_types)	components

queueDequeueUpToV2 Source #

Arguments

:: forall v'1 v'2 component_types m'. (MonadBuild m', TensorTypes component_types)
=> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Int32	n
-> m' (TensorList Value component_types)	components

queueDequeueUpToV2' Source #

Arguments

:: forall v'1 v'2 component_types m'. (MonadBuild m', TensorTypes component_types)
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Int32	n
-> m' (TensorList Value component_types)	components

queueDequeueV2 Source #

Arguments

:: forall v'1 component_types m'. (MonadBuild m', TensorTypes component_types)
=> Tensor v'1 ResourceHandle	handle
-> m' (TensorList Value component_types)	components

queueDequeueV2' Source #

Arguments

:: forall v'1 component_types m'. (MonadBuild m', TensorTypes component_types)
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> m' (TensorList Value component_types)	components

queueEnqueue Source #

Arguments

:: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents)
=> Tensor Ref ByteString	handle
-> TensorList v'2 tcomponents	components
-> m' ControlNode

queueEnqueue' Source #

Arguments

:: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents)
=> OpParams
-> Tensor Ref ByteString	handle
-> TensorList v'2 tcomponents	components
-> m' ControlNode

queueEnqueueMany Source #

Arguments

:: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents)
=> Tensor Ref ByteString	handle
-> TensorList v'2 tcomponents	components
-> m' ControlNode

queueEnqueueMany' Source #

Arguments

:: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents)
=> OpParams
-> Tensor Ref ByteString	handle
-> TensorList v'2 tcomponents	components
-> m' ControlNode

queueEnqueueManyV2 Source #

Arguments

:: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents)
=> Tensor v'1 ResourceHandle	handle
-> TensorList v'2 tcomponents	components
-> m' ControlNode

queueEnqueueManyV2' Source #

Arguments

:: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents)
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> TensorList v'2 tcomponents	components
-> m' ControlNode

queueEnqueueV2 Source #

Arguments

:: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents)
=> Tensor v'1 ResourceHandle	handle
-> TensorList v'2 tcomponents	components
-> m' ControlNode

queueEnqueueV2' Source #

Arguments

:: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents)
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> TensorList v'2 tcomponents	components
-> m' ControlNode

queueIsClosed Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	handle
-> m' (Tensor Value Bool)	is_closed

queueIsClosed' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	handle
-> m' (Tensor Value Bool)	is_closed

queueIsClosedV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> m' (Tensor Value Bool)	is_closed

queueIsClosedV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> m' (Tensor Value Bool)	is_closed

queueSize Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	handle
-> m' (Tensor Value Int32)	size

queueSize' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	handle
-> m' (Tensor Value Int32)	size

queueSizeV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> m' (Tensor Value Int32)	size

queueSizeV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> m' (Tensor Value Int32)	size

rFFT Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> Tensor v'1 treal	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build tcomplex	output

rFFT' Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> OpParams
-> Tensor v'1 treal	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build tcomplex	output

rFFT2D Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> Tensor v'1 treal	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build tcomplex	output

rFFT2D' Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> OpParams
-> Tensor v'1 treal	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build tcomplex	output

rFFT3D Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> Tensor v'1 treal	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build tcomplex	output

rFFT3D' Source #

Arguments

:: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex)
=> OpParams
-> Tensor v'1 treal	input
-> Tensor v'2 Int32	fft_length
-> Tensor Build tcomplex	output

rGBToHSV Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	images
-> Tensor Build t	output

rGBToHSV' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor Build t	output

raggedBincount Source #

Arguments

:: forall v'1 v'2 v'3 v'4 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t)
=> Tensor v'1 Int64	splits
-> Tensor v'2 tidx	values
-> Tensor v'3 tidx	size
-> Tensor v'4 t	weights
-> Tensor Build t	output

raggedBincount' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t)
=> OpParams
-> Tensor v'1 Int64	splits
-> Tensor v'2 tidx	values
-> Tensor v'3 tidx	size
-> Tensor v'4 t	weights
-> Tensor Build t	output

raggedCountSparseOutput Source #

Arguments

:: forall v'1 v'2 v'3 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type)
=> Bool	binary_output
-> Tensor v'1 Int64	splits
-> Tensor v'2 t	values
-> Tensor v'3 output_type	weights
-> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)	(output_indices, output_values, output_dense_shape) output_indices output_values output_dense_shape

raggedCountSparseOutput' Source #

Arguments

:: forall v'1 v'2 v'3 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type)
=> OpParams
-> Bool	binary_output
-> Tensor v'1 Int64	splits
-> Tensor v'2 t	values
-> Tensor v'3 output_type	weights
-> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)	(output_indices, output_values, output_dense_shape) output_indices output_values output_dense_shape

raggedCross Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 ragged_values_types ragged_splits_types sparse_values_types dense_types out_values_type out_row_splits_type. (OneOfs '[ByteString, Int64] ragged_values_types, OneOfs '[Int32, Int64] ragged_splits_types, OneOfs '[ByteString, Int64] sparse_values_types, OneOfs '[ByteString, Int64] dense_types, OneOf '[ByteString, Int64] out_values_type, OneOf '[Int32, Int64] out_row_splits_type)
=> Int64	hash_key
-> Bool	hashed_output
-> ByteString	input_order
-> Int64	num_buckets
-> TensorList v'1 ragged_values_types	ragged_values
-> TensorList v'2 ragged_splits_types	ragged_row_splits
-> [Tensor v'3 Int64]	sparse_indices
-> TensorList v'4 sparse_values_types	sparse_values
-> [Tensor v'5 Int64]	sparse_shape
-> TensorList v'6 dense_types	dense_inputs
-> (Tensor Build out_values_type, Tensor Build out_row_splits_type)	(output_values, output_row_splits) output_values output_row_splits

raggedCross' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 ragged_values_types ragged_splits_types sparse_values_types dense_types out_values_type out_row_splits_type. (OneOfs '[ByteString, Int64] ragged_values_types, OneOfs '[Int32, Int64] ragged_splits_types, OneOfs '[ByteString, Int64] sparse_values_types, OneOfs '[ByteString, Int64] dense_types, OneOf '[ByteString, Int64] out_values_type, OneOf '[Int32, Int64] out_row_splits_type)
=> OpParams
-> Int64	hash_key
-> Bool	hashed_output
-> ByteString	input_order
-> Int64	num_buckets
-> TensorList v'1 ragged_values_types	ragged_values
-> TensorList v'2 ragged_splits_types	ragged_row_splits
-> [Tensor v'3 Int64]	sparse_indices
-> TensorList v'4 sparse_values_types	sparse_values
-> [Tensor v'5 Int64]	sparse_shape
-> TensorList v'6 dense_types	dense_inputs
-> (Tensor Build out_values_type, Tensor Build out_row_splits_type)	(output_values, output_row_splits) output_values output_row_splits

raggedGather Source #

Arguments

:: forall v'1 v'2 v'3 tvalues tindices tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tsplits)
=> Int64	OUTPUT_RAGGED_RANK
-> [Tensor v'1 tsplits]	params_nested_splits
-> Tensor v'2 tvalues	params_dense_values
-> Tensor v'3 tindices	indices
-> ([Tensor Build tsplits], Tensor Build tvalues)	(output_nested_splits, output_dense_values) output_nested_splits output_dense_values

raggedGather' Source #

Arguments

:: forall v'1 v'2 v'3 tvalues tindices tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tsplits)
=> OpParams
-> Int64	OUTPUT_RAGGED_RANK
-> [Tensor v'1 tsplits]	params_nested_splits
-> Tensor v'2 tvalues	params_dense_values
-> Tensor v'3 tindices	indices
-> ([Tensor Build tsplits], Tensor Build tvalues)	(output_nested_splits, output_dense_values) output_nested_splits output_dense_values

raggedRange Source #

Arguments

:: forall v'1 v'2 v'3 t tsplits. (OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tsplits)
=> Tensor v'1 t	starts
-> Tensor v'2 t	limits
-> Tensor v'3 t	deltas
-> (Tensor Build tsplits, Tensor Build t)	(rt_nested_splits, rt_dense_values) rt_nested_splits rt_dense_values

raggedRange' Source #

Arguments

:: forall v'1 v'2 v'3 t tsplits. (OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tsplits)
=> OpParams
-> Tensor v'1 t	starts
-> Tensor v'2 t	limits
-> Tensor v'3 t	deltas
-> (Tensor Build tsplits, Tensor Build t)	(rt_nested_splits, rt_dense_values) rt_nested_splits rt_dense_values

raggedTensorFromVariant Source #

Arguments

:: forall v'1 tvalues tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tsplits)
=> Int64	input_ragged_rank
-> Int64	output_ragged_rank
-> Tensor v'1 Variant	encoded_ragged
-> ([Tensor Build tsplits], Tensor Build tvalues)	(output_nested_splits, output_dense_values) output_nested_splits output_dense_values

raggedTensorFromVariant' Source #

Arguments

:: forall v'1 tvalues tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tsplits)
=> OpParams
-> Int64	input_ragged_rank
-> Int64	output_ragged_rank
-> Tensor v'1 Variant	encoded_ragged
-> ([Tensor Build tsplits], Tensor Build tvalues)	(output_nested_splits, output_dense_values) output_nested_splits output_dense_values

raggedTensorToSparse Source #

Arguments

:: forall v'1 v'2 t tsplits. (TensorType t, OneOf '[Int32, Int64] tsplits)
=> [Tensor v'1 tsplits]	rt_nested_splits
-> Tensor v'2 t	rt_dense_values
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)	(sparse_indices, sparse_values, sparse_dense_shape) sparse_indices sparse_values sparse_dense_shape

raggedTensorToSparse' Source #

Arguments

:: forall v'1 v'2 t tsplits. (TensorType t, OneOf '[Int32, Int64] tsplits)
=> OpParams
-> [Tensor v'1 tsplits]	rt_nested_splits
-> Tensor v'2 t	rt_dense_values
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)	(sparse_indices, sparse_values, sparse_dense_shape) sparse_indices sparse_values sparse_dense_shape

raggedTensorToTensor Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tindex tshape. (TensorType t, OneOf '[Int32, Int64] tindex, OneOf '[Int32, Int64] tshape)
=> Tensor v'1 tshape	shape
-> Tensor v'2 t	values
-> Tensor v'3 t	default_value
-> [Tensor v'4 tindex]	row_partition_tensors
-> Tensor Build t	result

raggedTensorToTensor' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tindex tshape. (TensorType t, OneOf '[Int32, Int64] tindex, OneOf '[Int32, Int64] tshape)
=> OpParams
-> Tensor v'1 tshape	shape
-> Tensor v'2 t	values
-> Tensor v'3 t	default_value
-> [Tensor v'4 tindex]	row_partition_tensors
-> Tensor Build t	result

raggedTensorToVariant Source #

Arguments

:: forall v'1 v'2 tvalues tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tsplits)
=> Bool	batched_input
-> [Tensor v'1 tsplits]	rt_nested_splits
-> Tensor v'2 tvalues	rt_dense_values
-> Tensor Build Variant	encoded_ragged

raggedTensorToVariant' Source #

Arguments

:: forall v'1 v'2 tvalues tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tsplits)
=> OpParams
-> Bool	batched_input
-> [Tensor v'1 tsplits]	rt_nested_splits
-> Tensor v'2 tvalues	rt_dense_values
-> Tensor Build Variant	encoded_ragged

randomCrop Source #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t)
=> Tensor v'1 t	image
-> Tensor v'2 Int64	size
-> m' (Tensor Value t)	output

randomCrop' Source #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t)
=> OpParams
-> Tensor v'1 t	image
-> Tensor v'2 Int64	size
-> m' (Tensor Value t)	output

randomDataset Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Int64	seed
-> Tensor v'2 Int64	seed2
-> m' (Tensor Value Variant)	handle

randomDataset' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Int64	seed
-> Tensor v'2 Int64	seed2
-> m' (Tensor Value Variant)	handle

randomGamma Source #

Arguments

:: forall v'1 v'2 s t m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Word16, Double, Float] t)
=> Tensor v'1 s	shape
-> Tensor v'2 t	alpha
-> m' (Tensor Value t)	output

randomGamma' Source #

Arguments

:: forall v'1 v'2 s t m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Word16, Double, Float] t)
=> OpParams
-> Tensor v'1 s	shape
-> Tensor v'2 t	alpha
-> m' (Tensor Value t)	output

randomGammaGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 t	alpha
-> Tensor v'2 t	sample
-> Tensor Build t	output

randomGammaGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	alpha
-> Tensor v'2 t	sample
-> Tensor Build t	output

randomPoisson Source #

Arguments

:: forall v'1 v'2 s dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Word16, Double, Float] dtype)
=> Tensor v'1 s	shape
-> Tensor v'2 dtype	rate
-> m' (Tensor Value dtype)	output

randomPoisson' Source #

Arguments

:: forall v'1 v'2 s dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Word16, Double, Float] dtype)
=> OpParams
-> Tensor v'1 s	shape
-> Tensor v'2 dtype	rate
-> m' (Tensor Value dtype)	output

randomPoissonV2 Source #

Arguments

:: 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	shape
-> Tensor v'2 r	rate
-> m' (Tensor Value dtype)	output

randomPoissonV2' Source #

Arguments

:: 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	shape
-> Tensor v'2 r	rate
-> m' (Tensor Value dtype)	output

randomShuffle Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> Tensor v'1 t	value
-> m' (Tensor Value t)	output

randomShuffle' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor v'1 t	value
-> m' (Tensor Value t)	output

randomShuffleQueue Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	component_types
-> m' (Tensor Ref ByteString)	handle

randomShuffleQueue' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	component_types
-> m' (Tensor Ref ByteString)	handle

randomShuffleQueueV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	component_types
-> m' (Tensor Value ResourceHandle)	handle

randomShuffleQueueV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	component_types
-> m' (Tensor Value ResourceHandle)	handle

randomStandardNormal Source #

Arguments

:: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t)
=> Tensor v'1 t	shape
-> m' (Tensor Value dtype)	output

randomStandardNormal' Source #

Arguments

:: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t)
=> OpParams
-> Tensor v'1 t	shape
-> m' (Tensor Value dtype)	output

randomUniform Source #

Arguments

:: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t)
=> Tensor v'1 t	shape
-> m' (Tensor Value dtype)	output

randomUniform' Source #

Arguments

:: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t)
=> OpParams
-> Tensor v'1 t	shape
-> m' (Tensor Value dtype)	output

randomUniformInt Source #

Arguments

:: forall v'1 v'2 v'3 tout t m'. (MonadBuild m', OneOf '[Int32, Int64] tout, OneOf '[Int32, Int64] t)
=> Tensor v'1 t	shape
-> Tensor v'2 tout	minval
-> Tensor v'3 tout	maxval
-> m' (Tensor Value tout)	output

randomUniformInt' Source #

Arguments

:: 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	shape
-> Tensor v'2 tout	minval
-> Tensor v'3 tout	maxval
-> m' (Tensor Value tout)	output

range Source #

Arguments

:: forall v'1 v'2 v'3 tidx. OneOf '[Int32, Int64, Word16, Double, Float] tidx
=> Tensor v'1 tidx	start
-> Tensor v'2 tidx	limit
-> Tensor v'3 tidx	delta
-> Tensor Build tidx	output

range' Source #

Arguments

:: forall v'1 v'2 v'3 tidx. OneOf '[Int32, Int64, Word16, Double, Float] tidx
=> OpParams
-> Tensor v'1 tidx	start
-> Tensor v'2 tidx	limit
-> Tensor v'3 tidx	delta
-> Tensor Build tidx	output

rangeDataset Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Int64	start
-> Tensor v'2 Int64	stop
-> Tensor v'3 Int64	step
-> m' (Tensor Value Variant)	handle

rangeDataset' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Int64	start
-> Tensor v'2 Int64	stop
-> Tensor v'3 Int64	step
-> m' (Tensor Value Variant)	handle

rank Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build Int32	output

rank' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build Int32	output

readFile Source #

Arguments

:: Tensor v'1 ByteString	filename
-> Tensor Build ByteString	contents

readFile' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	filename
-> Tensor Build ByteString	contents

readVariableOp Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Value dtype)	value

readVariableOp' Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Value dtype)	value

readerNumRecordsProduced Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	reader_handle
-> m' (Tensor Value Int64)	records_produced

readerNumRecordsProduced' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	reader_handle
-> m' (Tensor Value Int64)	records_produced

readerNumRecordsProducedV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	reader_handle
-> m' (Tensor Value Int64)	records_produced

readerNumRecordsProducedV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	reader_handle
-> m' (Tensor Value Int64)	records_produced

readerNumWorkUnitsCompleted Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	reader_handle
-> m' (Tensor Value Int64)	units_completed

readerNumWorkUnitsCompleted' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	reader_handle
-> m' (Tensor Value Int64)	units_completed

readerNumWorkUnitsCompletedV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	reader_handle
-> m' (Tensor Value Int64)	units_completed

readerNumWorkUnitsCompletedV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	reader_handle
-> m' (Tensor Value Int64)	units_completed

readerRead Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	reader_handle
-> Tensor Ref ByteString	queue_handle
-> m' (Tensor Value ByteString, Tensor Value ByteString)	(key, value) key value

readerRead' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	reader_handle
-> Tensor Ref ByteString	queue_handle
-> m' (Tensor Value ByteString, Tensor Value ByteString)	(key, value) key value

readerReadUpTo Source #

Arguments

:: forall v'3 m'. 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) keys values

readerReadUpTo' Source #

Arguments

:: forall v'3 m'. 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) keys values

readerReadUpToV2 Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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) keys values

readerReadUpToV2' Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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) keys values

readerReadV2 Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	reader_handle
-> Tensor v'2 ResourceHandle	queue_handle
-> m' (Tensor Value ByteString, Tensor Value ByteString)	(key, value) key value

readerReadV2' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	reader_handle
-> Tensor v'2 ResourceHandle	queue_handle
-> m' (Tensor Value ByteString, Tensor Value ByteString)	(key, value) key value

readerReset Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	reader_handle
-> m' ControlNode

readerReset' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	reader_handle
-> m' ControlNode

readerResetV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	reader_handle
-> m' ControlNode

readerResetV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	reader_handle
-> m' ControlNode

readerRestoreState Source #

Arguments

:: forall v'2 m'. MonadBuild m'
=> Tensor Ref ByteString	reader_handle
-> Tensor v'2 ByteString	state
-> m' ControlNode

readerRestoreState' Source #

Arguments

:: forall v'2 m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	reader_handle
-> Tensor v'2 ByteString	state
-> m' ControlNode

readerRestoreStateV2 Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	reader_handle
-> Tensor v'2 ByteString	state
-> m' ControlNode

readerRestoreStateV2' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	reader_handle
-> Tensor v'2 ByteString	state
-> m' ControlNode

readerSerializeState Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	reader_handle
-> m' (Tensor Value ByteString)	state

readerSerializeState' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	reader_handle
-> m' (Tensor Value ByteString)	state

readerSerializeStateV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	reader_handle
-> m' (Tensor Value ByteString)	state

readerSerializeStateV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	reader_handle
-> m' (Tensor Value ByteString)	state

real Source #

Arguments

:: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout)
=> Tensor v'1 t	input
-> Tensor Build tout	output

real' Source #

Arguments

:: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout)
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build tout	output

realDiv Source #

Arguments

:: forall v'1 v'2 t. 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 Build t	z

realDiv' Source #

Arguments

:: 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	x
-> Tensor v'2 t	y
-> Tensor Build t	z

rebatchDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_replicas
-> Tensor Build Variant	handle

rebatchDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_replicas
-> Tensor Build Variant	handle

reciprocal Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

reciprocal' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

reciprocalGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

reciprocalGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

recordInput Source #

Arguments

:: forall m'. MonadBuild m'
=> ByteString	file_pattern
-> m' (Tensor Value ByteString)	records

recordInput' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> ByteString	file_pattern
-> m' (Tensor Value ByteString)	records

recv Source #

Arguments

:: forall tensor_type m'. (MonadBuild m', TensorType tensor_type)
=> ByteString	recv_device
-> ByteString	send_device
-> Int64	send_device_incarnation
-> ByteString	tensor_name
-> m' (Tensor Value tensor_type)	tensor

recv' Source #

Arguments

:: forall tensor_type m'. (MonadBuild m', TensorType tensor_type)
=> OpParams
-> ByteString	recv_device
-> ByteString	send_device
-> Int64	send_device_incarnation
-> ByteString	tensor_name
-> m' (Tensor Value tensor_type)	tensor

recvTPUEmbeddingActivations Source #

Arguments

:: forall m'. MonadBuild m'
=> ByteString	config
-> Int64	num_outputs
-> m' [Tensor Value Float]	outputs

recvTPUEmbeddingActivations' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> ByteString	config
-> Int64	num_outputs
-> m' [Tensor Value Float]	outputs

reduceJoin Source #

Arguments

:: Tensor v'1 ByteString	inputs
-> Tensor v'2 Int32	reduction_indices
-> Tensor Build ByteString	output

reduceJoin' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	inputs
-> Tensor v'2 Int32	reduction_indices
-> Tensor Build ByteString	output

refEnter Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> ByteString	frame_name
-> Tensor Ref t	data
-> m' (Tensor Ref t)	output

refEnter' Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> OpParams
-> ByteString	frame_name
-> Tensor Ref t	data
-> m' (Tensor Ref t)	output

refExit Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> Tensor Ref t	data
-> m' (Tensor Ref t)	output

refExit' Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor Ref t	data
-> m' (Tensor Ref t)	output

refIdentity Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> Tensor Ref t	input
-> m' (Tensor Ref t)	output

refIdentity' Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor Ref t	input
-> m' (Tensor Ref t)	output

refMerge Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)

=> [Tensor Ref t]

inputs

-> m' (Tensor Ref t, Tensor Value Int32)

(output, value_index)

output
value_index

refMerge' Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> OpParams
-> [Tensor Ref t]	inputs
-> m' (Tensor Ref t, Tensor Value Int32)	(output, value_index) output value_index

refNextIteration Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> Tensor Ref t	data
-> m' (Tensor Ref t)	output

refNextIteration' Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor Ref t	data
-> m' (Tensor Ref t)	output

refSelect Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> Tensor v'1 Int32	index
-> [Tensor Ref t]	inputs
-> m' (Tensor Ref t)	output

refSelect' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor v'1 Int32	index
-> [Tensor Ref t]	inputs
-> m' (Tensor Ref t)	output

refSwitch Source #

Arguments

:: forall v'2 t m'. (MonadBuild m', TensorType t)
=> Tensor Ref t	data
-> Tensor v'2 Bool	pred
-> m' (Tensor Ref t, Tensor Ref t)	(output_false, output_true) output_false output_true

refSwitch' Source #

Arguments

:: forall v'2 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor Ref t	data
-> Tensor v'2 Bool	pred
-> m' (Tensor Ref t, Tensor Ref t)	(output_false, output_true) output_false output_true

regexFullMatch Source #

Arguments

:: Tensor v'1 ByteString	input
-> Tensor v'2 ByteString	pattern
-> Tensor Build Bool	output

regexFullMatch' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	input
-> Tensor v'2 ByteString	pattern
-> Tensor Build Bool	output

regexReplace Source #

Arguments

:: Tensor v'1 ByteString	input
-> Tensor v'2 ByteString	pattern
-> Tensor v'3 ByteString	rewrite
-> Tensor Build ByteString	output

regexReplace' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	input
-> Tensor v'2 ByteString	pattern
-> Tensor v'3 ByteString	rewrite
-> Tensor Build ByteString	output

registerDataset Source #

Arguments

:: Int64	external_state_policy
-> Tensor v'1 Variant	dataset
-> Tensor v'2 ByteString	address
-> Tensor v'3 ByteString	protocol
-> Tensor Build Int64	dataset_id

registerDataset' Source #

Arguments

:: OpParams
-> Int64	external_state_policy
-> Tensor v'1 Variant	dataset
-> Tensor v'2 ByteString	address
-> Tensor v'3 ByteString	protocol
-> Tensor Build Int64	dataset_id

relu Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	features
-> Tensor Build t	activations

relu' Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	features
-> Tensor Build t	activations

relu6 Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	features
-> Tensor Build t	activations

relu6' Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	features
-> Tensor Build t	activations

relu6Grad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	gradients
-> Tensor v'2 t	features
-> Tensor Build t	backprops

relu6Grad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	gradients
-> Tensor v'2 t	features
-> Tensor Build t	backprops

reluGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	gradients
-> Tensor v'2 t	features
-> Tensor Build t	backprops

reluGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	gradients
-> Tensor v'2 t	features
-> Tensor Build t	backprops

remoteFusedGraphExecute Source #

Arguments

:: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs)
=> ByteString	serialized_remote_fused_graph_execute_info
-> TensorList v'1 tinputs	inputs
-> TensorList Build toutputs	outputs

remoteFusedGraphExecute' Source #

Arguments

:: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs)
=> OpParams
-> ByteString	serialized_remote_fused_graph_execute_info
-> TensorList v'1 tinputs	inputs
-> TensorList Build toutputs	outputs

repeatDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	count
-> Tensor Build Variant	handle

repeatDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	count
-> Tensor Build Variant	handle

requantizationRange Source #

Arguments

:: forall v'1 v'2 v'3 tinput. OneOf '[Int16, Int32, Word16, Word8] tinput
=> Tensor v'1 tinput	input
-> Tensor v'2 Float	input_min
-> Tensor v'3 Float	input_max
-> (Tensor Build Float, Tensor Build Float)	(output_min, output_max) output_min output_max

requantizationRange' Source #

Arguments

:: forall v'1 v'2 v'3 tinput. OneOf '[Int16, Int32, Word16, Word8] tinput
=> OpParams
-> Tensor v'1 tinput	input
-> Tensor v'2 Float	input_min
-> Tensor v'3 Float	input_max
-> (Tensor Build Float, Tensor Build Float)	(output_min, output_max) output_min output_max

requantizationRangePerChannel Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t
=> Float	clip_value_max
-> Tensor v'1 t	input
-> Tensor v'2 Float	input_min
-> Tensor v'3 Float	input_max
-> (Tensor Build Float, Tensor Build Float)	(output_min, output_max) output_min output_max

requantizationRangePerChannel' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t
=> OpParams
-> Float	clip_value_max
-> Tensor v'1 t	input
-> Tensor v'2 Float	input_min
-> Tensor v'3 Float	input_max
-> (Tensor Build Float, Tensor Build Float)	(output_min, output_max) output_min output_max

requantize Source #

Arguments

:: 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	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) output output_min output_max

requantize' Source #

Arguments

:: 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	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) output output_min output_max

requantizePerChannel Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t out_type. (OneOf '[Int16, Int32, Word16, Word8] t, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> Tensor v'1 t	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) output output_min output_max

requantizePerChannel' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t out_type. (OneOf '[Int16, Int32, Word16, Word8] t, OneOf '[Int16, Int32, Word16, Word8] out_type)
=> OpParams
-> Tensor v'1 t	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) output output_min output_max

reshape Source #

Arguments

:: forall v'1 v'2 t tshape. (TensorType t, OneOf '[Int32, Int64] tshape)
=> Tensor v'1 t	tensor
-> Tensor v'2 tshape	shape
-> Tensor Build t	output

reshape' Source #

Arguments

:: forall v'1 v'2 t tshape. (TensorType t, OneOf '[Int32, Int64] tshape)
=> OpParams
-> Tensor v'1 t	tensor
-> Tensor v'2 tshape	shape
-> Tensor Build t	output

resizeArea Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Int32	size
-> Tensor Build Float	resized_images

resizeArea' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor v'2 Int32	size
-> Tensor Build Float	resized_images

resizeBicubic Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Int32	size
-> Tensor Build Float	resized_images

resizeBicubic' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor v'2 Int32	size
-> Tensor Build Float	resized_images

resizeBicubicGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 Float	grads
-> Tensor v'2 t	original_image
-> Tensor Build t	output

resizeBicubicGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 Float	grads
-> Tensor v'2 t	original_image
-> Tensor Build t	output

resizeBilinear Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Int32	size
-> Tensor Build Float	resized_images

resizeBilinear' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor v'2 Int32	size
-> Tensor Build Float	resized_images

resizeBilinearGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 Float	grads
-> Tensor v'2 t	original_image
-> Tensor Build t	output

resizeBilinearGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 Float	grads
-> Tensor v'2 t	original_image
-> Tensor Build t	output

resizeNearestNeighbor Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Int32	size
-> Tensor Build t	resized_images

resizeNearestNeighbor' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	images
-> Tensor v'2 Int32	size
-> Tensor Build t	resized_images

resizeNearestNeighborGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int8, Word16, Word8, Double, Float] t
=> Tensor v'1 t	grads
-> Tensor v'2 Int32	size
-> Tensor Build t	output

resizeNearestNeighborGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int8, Word16, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	grads
-> Tensor v'2 Int32	size
-> Tensor Build t	output

resourceAccumulatorApplyGradient Source #

Arguments

:: forall v'1 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 v'1 ResourceHandle	handle
-> Tensor v'2 Int64	local_step
-> Tensor v'3 dtype	gradient
-> m' ControlNode

resourceAccumulatorApplyGradient' Source #

Arguments

:: forall v'1 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 v'1 ResourceHandle	handle
-> Tensor v'2 Int64	local_step
-> Tensor v'3 dtype	gradient
-> m' ControlNode

resourceAccumulatorNumAccumulated Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> m' (Tensor Value Int32)	num_accumulated

resourceAccumulatorNumAccumulated' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> m' (Tensor Value Int32)	num_accumulated

resourceAccumulatorSetGlobalStep Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Int64	new_global_step
-> m' ControlNode

resourceAccumulatorSetGlobalStep' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Int64	new_global_step
-> m' ControlNode

resourceAccumulatorTakeGradient Source #

Arguments

:: forall v'1 v'2 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype)
=> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Int32	num_required
-> m' (Tensor Value dtype)	average

resourceAccumulatorTakeGradient' Source #

Arguments

:: forall v'1 v'2 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Int32	num_required
-> m' (Tensor Value dtype)	average

resourceApplyAdaMax Source #

Arguments

:: 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	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

resourceApplyAdaMax' Source #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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

resourceApplyAdagrad Source #

Arguments

:: 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	var
-> Tensor v'2 ResourceHandle	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	grad
-> m' ControlNode

resourceApplyAdagrad' Source #

Arguments

:: 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	var
-> Tensor v'2 ResourceHandle	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	grad
-> m' ControlNode

resourceApplyAdagradDA Source #

Arguments

:: 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	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 #

Arguments

:: 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	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

resourceApplyAdagradV2 Source #

Arguments

:: 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	var
-> Tensor v'2 ResourceHandle	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	epsilon
-> Tensor v'5 t	grad
-> m' ControlNode

resourceApplyAdagradV2' Source #

Arguments

:: 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	var
-> Tensor v'2 ResourceHandle	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	epsilon
-> Tensor v'5 t	grad
-> m' ControlNode

resourceApplyAdam Source #

Arguments

:: 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	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

resourceApplyAdam' Source #

Arguments

:: 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	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

resourceApplyAdamWithAmsgrad Source #

Arguments

:: 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 '[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 ResourceHandle	vhat
-> Tensor v'5 t	beta1_power
-> Tensor v'6 t	beta2_power
-> Tensor v'7 t	lr
-> Tensor v'8 t	beta1
-> Tensor v'9 t	beta2
-> Tensor v'10 t	epsilon
-> Tensor v'11 t	grad
-> m' ControlNode

resourceApplyAdamWithAmsgrad' Source #

Arguments

:: 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 '[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 ResourceHandle	vhat
-> Tensor v'5 t	beta1_power
-> Tensor v'6 t	beta2_power
-> Tensor v'7 t	lr
-> Tensor v'8 t	beta1
-> Tensor v'9 t	beta2
-> Tensor v'10 t	epsilon
-> Tensor v'11 t	grad
-> m' ControlNode

resourceApplyAddSign Source #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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

resourceApplyFtrl Source #

Arguments

:: 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	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

resourceApplyFtrl' Source #

Arguments

:: 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	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

resourceApplyFtrlV2 Source #

Arguments

:: 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	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

resourceApplyFtrlV2' Source #

Arguments

:: 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	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

resourceApplyGradientDescent Source #

Arguments

:: 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	var
-> Tensor v'2 t	alpha
-> Tensor v'3 t	delta
-> m' ControlNode

resourceApplyGradientDescent' Source #

Arguments

:: 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	var
-> Tensor v'2 t	alpha
-> Tensor v'3 t	delta
-> m' ControlNode

resourceApplyKerasMomentum Source #

Arguments

:: 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	var
-> Tensor v'2 ResourceHandle	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	grad
-> Tensor v'5 t	momentum
-> m' ControlNode

resourceApplyKerasMomentum' Source #

Arguments

:: 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	var
-> Tensor v'2 ResourceHandle	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	grad
-> Tensor v'5 t	momentum
-> m' ControlNode

resourceApplyMomentum Source #

Arguments

:: 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	var
-> Tensor v'2 ResourceHandle	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	grad
-> Tensor v'5 t	momentum
-> m' ControlNode

resourceApplyMomentum' Source #

Arguments

:: 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	var
-> Tensor v'2 ResourceHandle	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	grad
-> Tensor v'5 t	momentum
-> m' ControlNode

resourceApplyPowerSign Source #

Arguments

:: 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	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 #

Arguments

:: 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	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

resourceApplyProximalAdagrad Source #

Arguments

:: 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	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
-> m' ControlNode

resourceApplyProximalAdagrad' Source #

Arguments

:: 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	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
-> m' ControlNode

resourceApplyProximalGradientDescent Source #

Arguments

:: 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	var
-> Tensor v'2 t	alpha
-> Tensor v'3 t	l1
-> Tensor v'4 t	l2
-> Tensor v'5 t	delta
-> m' ControlNode

resourceApplyProximalGradientDescent' Source #

Arguments

:: 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	var
-> Tensor v'2 t	alpha
-> Tensor v'3 t	l1
-> Tensor v'4 t	l2
-> Tensor v'5 t	delta
-> m' ControlNode

resourceApplyRMSProp Source #

Arguments

:: 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	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 #

Arguments

:: 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	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

resourceConditionalAccumulator Source #

Arguments

:: forall m'. MonadBuild m'
=> DataType	dtype
-> Shape	shape
-> m' (Tensor Value ResourceHandle)	handle

resourceConditionalAccumulator' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> DataType	dtype
-> Shape	shape
-> m' (Tensor Value ResourceHandle)	handle

resourceCountUpTo Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64] t)
=> Int64	limit
-> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Value t)	output

resourceCountUpTo' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64] t)
=> OpParams
-> Int64	limit
-> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Value t)	output

resourceGather Source #

Arguments

:: forall v'1 v'2 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 tindices	indices
-> m' (Tensor Value dtype)	output

resourceGather' Source #

Arguments

:: forall v'1 v'2 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 tindices	indices
-> m' (Tensor Value dtype)	output

resourceGatherNd Source #

Arguments

:: forall v'1 v'2 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 tindices	indices
-> m' (Tensor Value dtype)	output

resourceGatherNd' Source #

Arguments

:: forall v'1 v'2 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 tindices	indices
-> m' (Tensor Value dtype)	output

resourceScatterAdd Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterAdd' Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterDiv Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterDiv' Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterMax Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterMax' Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterMin Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterMin' Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterMul Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterMul' Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterNdAdd Source #

Arguments

:: forall v'1 v'2 v'3 t tindices m'. (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

resourceScatterNdAdd' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices m'. (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

resourceScatterNdMax Source #

Arguments

:: forall v'1 v'2 v'3 t tindices m'. (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

resourceScatterNdMax' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices m'. (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

resourceScatterNdMin Source #

Arguments

:: forall v'1 v'2 v'3 t tindices m'. (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

resourceScatterNdMin' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices m'. (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

resourceScatterNdSub Source #

Arguments

:: forall v'1 v'2 v'3 t tindices m'. (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

resourceScatterNdSub' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices m'. (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

resourceScatterNdUpdate Source #

Arguments

:: forall v'1 v'2 v'3 t tindices m'. (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 #

Arguments

:: forall v'1 v'2 v'3 t tindices m'. (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

resourceScatterSub Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterSub' Source #

Arguments

:: 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	resource
-> Tensor v'2 tindices	indices
-> Tensor v'3 dtype	updates
-> m' ControlNode

resourceScatterUpdate Source #

Arguments

:: forall v'1 v'2 v'3 dtype tindices m'. (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 #

Arguments

:: forall v'1 v'2 v'3 dtype tindices m'. (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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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

resourceSparseApplyAdagradV2 Source #

Arguments

:: 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	var
-> Tensor v'2 ResourceHandle	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	epsilon
-> Tensor v'5 t	grad
-> Tensor v'6 tindices	indices
-> m' ControlNode

resourceSparseApplyAdagradV2' Source #

Arguments

:: 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	var
-> Tensor v'2 ResourceHandle	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	epsilon
-> Tensor v'5 t	grad
-> Tensor v'6 tindices	indices
-> m' ControlNode

resourceSparseApplyCenteredRMSProp Source #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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

resourceSparseApplyKerasMomentum Source #

Arguments

:: 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	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

resourceSparseApplyKerasMomentum' Source #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	ref
-> Tensor v'2 index	begin
-> Tensor v'3 index	end
-> Tensor v'4 index	strides
-> Tensor v'5 t	value
-> m' ControlNode

restore Source #

Arguments

:: forall v'1 v'2 dt m'. (MonadBuild m', TensorType dt)
=> Tensor v'1 ByteString	file_pattern
-> Tensor v'2 ByteString	tensor_name
-> m' (Tensor Value dt)	tensor

restore' Source #

Arguments

:: forall v'1 v'2 dt m'. (MonadBuild m', TensorType dt)
=> OpParams
-> Tensor v'1 ByteString	file_pattern
-> Tensor v'2 ByteString	tensor_name
-> m' (Tensor Value dt)	tensor

restoreSlice Source #

Arguments

:: forall v'1 v'2 v'3 dt m'. (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

restoreSlice' Source #

Arguments

:: forall v'1 v'2 v'3 dt m'. (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

restoreV2 Source #

Arguments

:: forall v'1 v'2 v'3 dtypes m'. (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

restoreV2' Source #

Arguments

:: forall v'1 v'2 v'3 dtypes m'. (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

retrieveTPUEmbeddingADAMParameters Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, momenta, velocities) parameters momenta velocities

retrieveTPUEmbeddingADAMParameters' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, momenta, velocities) parameters momenta velocities

retrieveTPUEmbeddingADAMParametersGradAccumDebug Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, momenta, velocities, gradient_accumulators) parameters momenta velocities gradient_accumulators

retrieveTPUEmbeddingADAMParametersGradAccumDebug' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, momenta, velocities, gradient_accumulators) parameters momenta velocities gradient_accumulators

retrieveTPUEmbeddingAdadeltaParameters Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, updates) parameters accumulators updates

retrieveTPUEmbeddingAdadeltaParameters' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, updates) parameters accumulators updates

retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, updates, gradient_accumulators) parameters accumulators updates gradient_accumulators

retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, updates, gradient_accumulators) parameters accumulators updates gradient_accumulators

retrieveTPUEmbeddingAdagradParameters Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float)	(parameters, accumulators) parameters accumulators

retrieveTPUEmbeddingAdagradParameters' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float)	(parameters, accumulators) parameters accumulators

retrieveTPUEmbeddingAdagradParametersGradAccumDebug Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, gradient_accumulators) parameters accumulators gradient_accumulators

retrieveTPUEmbeddingAdagradParametersGradAccumDebug' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, gradient_accumulators) parameters accumulators gradient_accumulators

retrieveTPUEmbeddingCenteredRMSPropParameters Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, ms, mom, mg) parameters ms mom mg

retrieveTPUEmbeddingCenteredRMSPropParameters' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, ms, mom, mg) parameters ms mom mg

retrieveTPUEmbeddingFTRLParameters Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, linears) parameters accumulators linears

retrieveTPUEmbeddingFTRLParameters' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, linears) parameters accumulators linears

retrieveTPUEmbeddingFTRLParametersGradAccumDebug Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, linears, gradient_accumulators) parameters accumulators linears gradient_accumulators

retrieveTPUEmbeddingFTRLParametersGradAccumDebug' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, linears, gradient_accumulators) parameters accumulators linears gradient_accumulators

retrieveTPUEmbeddingMDLAdagradLightParameters Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, weights, benefits) parameters accumulators weights benefits

retrieveTPUEmbeddingMDLAdagradLightParameters' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, weights, benefits) parameters accumulators weights benefits

retrieveTPUEmbeddingMomentumParameters Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float)	(parameters, momenta) parameters momenta

retrieveTPUEmbeddingMomentumParameters' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float)	(parameters, momenta) parameters momenta

retrieveTPUEmbeddingMomentumParametersGradAccumDebug Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, momenta, gradient_accumulators) parameters momenta gradient_accumulators

retrieveTPUEmbeddingMomentumParametersGradAccumDebug' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, momenta, gradient_accumulators) parameters momenta gradient_accumulators

retrieveTPUEmbeddingProximalAdagradParameters Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float)	(parameters, accumulators) parameters accumulators

retrieveTPUEmbeddingProximalAdagradParameters' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float)	(parameters, accumulators) parameters accumulators

retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, gradient_accumulators) parameters accumulators gradient_accumulators

retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, accumulators, gradient_accumulators) parameters accumulators gradient_accumulators

retrieveTPUEmbeddingProximalYogiParameters Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, v, m) parameters v m

retrieveTPUEmbeddingProximalYogiParameters' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, v, m) parameters v m

retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, v, m, gradient_accumulators) parameters v m gradient_accumulators

retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, v, m, gradient_accumulators) parameters v m gradient_accumulators

retrieveTPUEmbeddingRMSPropParameters Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, ms, mom) parameters ms mom

retrieveTPUEmbeddingRMSPropParameters' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, ms, mom) parameters ms mom

retrieveTPUEmbeddingRMSPropParametersGradAccumDebug Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, ms, mom, gradient_accumulators) parameters ms mom gradient_accumulators

retrieveTPUEmbeddingRMSPropParametersGradAccumDebug' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)	(parameters, ms, mom, gradient_accumulators) parameters ms mom gradient_accumulators

retrieveTPUEmbeddingStochasticGradientDescentParameters Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float)	parameters

retrieveTPUEmbeddingStochasticGradientDescentParameters' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float)	parameters

retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float)	(parameters, gradient_accumulators) parameters gradient_accumulators

retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_shards
-> Int64	shard_id
-> m' (Tensor Value Float, Tensor Value Float)	(parameters, gradient_accumulators) parameters gradient_accumulators

reverse Source #

Arguments

:: 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
-> Tensor v'2 Bool	dims
-> Tensor Build t	output

reverse' Source #

Arguments

:: 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
-> Tensor v'2 Bool	dims
-> Tensor Build t	output

reverseSequence Source #

Arguments

:: forall v'1 v'2 t tlen. (TensorType t, OneOf '[Int32, Int64] tlen)
=> Int64	seq_dim
-> Tensor v'1 t	input
-> Tensor v'2 tlen	seq_lengths
-> Tensor Build t	output

reverseSequence' Source #

Arguments

:: forall v'1 v'2 t tlen. (TensorType t, OneOf '[Int32, Int64] tlen)
=> OpParams
-> Int64	seq_dim
-> Tensor v'1 t	input
-> Tensor v'2 tlen	seq_lengths
-> Tensor Build t	output

reverseV2 Source #

Arguments

:: 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
-> Tensor v'2 tidx	axis
-> Tensor Build t	output

reverseV2' Source #

Arguments

:: 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
-> Tensor v'2 tidx	axis
-> Tensor Build t	output

rightShift Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

rightShift' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

rint Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

rint' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

rngSkip Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 Int64	delta
-> m' ControlNode

rngSkip' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 Int64	delta
-> m' ControlNode

roll Source #

Arguments

:: forall v'1 v'2 v'3 t tshift taxis. (TensorType t, OneOf '[Int32, Int64] tshift, OneOf '[Int32, Int64] taxis)
=> Tensor v'1 t	input
-> Tensor v'2 tshift	shift
-> Tensor v'3 taxis	axis
-> Tensor Build t	output

roll' Source #

Arguments

:: 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	input
-> Tensor v'2 tshift	shift
-> Tensor v'3 taxis	axis
-> Tensor Build t	output

round Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

round' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

rpc Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 ByteString	address
-> Tensor v'2 ByteString	method
-> Tensor v'3 ByteString	request
-> m' (Tensor Value ByteString)	response

rpc' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	address
-> Tensor v'2 ByteString	method
-> Tensor v'3 ByteString	request
-> m' (Tensor Value ByteString)	response

rsqrt Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

rsqrt' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

rsqrtGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

rsqrtGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

sampleDistortedBoundingBox Source #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t)
=> Tensor v'1 t	image_size
-> Tensor v'2 Float	bounding_boxes
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)	(begin, size, bboxes) begin size bboxes

sampleDistortedBoundingBox' Source #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t)
=> OpParams
-> Tensor v'1 t	image_size
-> Tensor v'2 Float	bounding_boxes
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)	(begin, size, bboxes) begin size bboxes

sampleDistortedBoundingBoxV2 Source #

Arguments

:: forall v'1 v'2 v'3 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t)
=> Tensor v'1 t	image_size
-> Tensor v'2 Float	bounding_boxes
-> Tensor v'3 Float	min_object_covered
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)	(begin, size, bboxes) begin size bboxes

sampleDistortedBoundingBoxV2' Source #

Arguments

:: forall v'1 v'2 v'3 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t)
=> OpParams
-> Tensor v'1 t	image_size
-> Tensor v'2 Float	bounding_boxes
-> Tensor v'3 Float	min_object_covered
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)	(begin, size, bboxes) begin size bboxes

samplingDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Float	rate
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	seed2
-> Tensor Build Variant	handle

samplingDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Float	rate
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	seed2
-> Tensor Build Variant	handle

save Source #

Arguments

:: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorTypes t)
=> Tensor v'1 ByteString	filename
-> Tensor v'2 ByteString	tensor_names
-> TensorList v'3 t	data
-> m' ControlNode

save' Source #

Arguments

:: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorTypes t)
=> OpParams
-> Tensor v'1 ByteString	filename
-> Tensor v'2 ByteString	tensor_names
-> TensorList v'3 t	data
-> m' ControlNode

saveSlices Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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

saveSlices' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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

saveV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 dtypes m'. (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

saveV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 dtypes m'. (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

scalarSummary Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 ByteString	tags
-> Tensor v'2 t	values
-> Tensor Build ByteString	summary

scalarSummary' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 ByteString	tags
-> Tensor v'2 t	values
-> Tensor Build ByteString	summary

scaleAndTranslate Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t
=> Tensor v'1 t	images
-> Tensor v'2 Int32	size
-> Tensor v'3 Float	scale
-> Tensor v'4 Float	translation
-> Tensor Build Float	resized_images

scaleAndTranslate' Source #

Arguments

:: 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	images
-> Tensor v'2 Int32	size
-> Tensor v'3 Float	scale
-> Tensor v'4 Float	translation
-> Tensor Build Float	resized_images

scaleAndTranslateGrad Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Float] t
=> Tensor v'1 t	grads
-> Tensor v'2 t	original_image
-> Tensor v'3 Float	scale
-> Tensor v'4 Float	translation
-> Tensor Build t	output

scaleAndTranslateGrad' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Float] t
=> OpParams
-> Tensor v'1 t	grads
-> Tensor v'2 t	original_image
-> Tensor v'3 Float	scale
-> Tensor v'4 Float	translation
-> Tensor Build t	output

scatterAdd Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterAdd' Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterDiv Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterDiv' Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterMax Source #

Arguments

:: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tindices)
=> Tensor Ref t	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterMax' Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterMin Source #

Arguments

:: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tindices)
=> Tensor Ref t	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterMin' Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterMul Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterMul' Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterNd Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 tindices	indices
-> Tensor v'2 t	updates
-> Tensor v'3 tindices	shape
-> Tensor Build t	output

scatterNd' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 tindices	indices
-> Tensor v'2 t	updates
-> Tensor v'3 tindices	shape
-> Tensor Build t	output

scatterNdAdd Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterNdAdd' Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterNdMax Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterNdMax' Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterNdMin Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterNdMin' Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterNdNonAliasingAdd Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 t	input
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

scatterNdNonAliasingAdd' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

scatterNdSub Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterNdSub' Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterNdUpdate Source #

Arguments

:: forall v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor Ref t	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterNdUpdate' Source #

Arguments

:: forall v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor Ref t	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterSub Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterSub' Source #

Arguments

:: 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	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterUpdate Source #

Arguments

:: forall v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor Ref t	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

scatterUpdate' Source #

Arguments

:: forall v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor Ref t	ref
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> m' (Tensor Ref t)	output_ref

sdcaFprint Source #

Arguments

:: Tensor v'1 ByteString	input
-> Tensor Build Int64	output

sdcaFprint' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	input
-> Tensor Build Int64	output

sdcaOptimizer Source #

Arguments

:: Float	l1
-> Float	l2
-> ByteString	loss_type
-> 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) out_example_state_data out_delta_sparse_weights out_delta_dense_weights

sdcaOptimizer' Source #

Arguments

:: OpParams
-> Float	l1
-> Float	l2
-> ByteString	loss_type
-> 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) out_example_state_data out_delta_sparse_weights out_delta_dense_weights

sdcaOptimizerV2 Source #

Arguments

:: Float	l1
-> Float	l2
-> ByteString	loss_type
-> 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) out_example_state_data out_delta_sparse_weights out_delta_dense_weights

sdcaOptimizerV2' Source #

Arguments

:: OpParams
-> Float	l1
-> Float	l2
-> ByteString	loss_type
-> 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) out_example_state_data out_delta_sparse_weights out_delta_dense_weights

sdcaShrinkL1 Source #

Arguments

:: forall m'. MonadBuild m'
=> Float	l1
-> Float	l2
-> [Tensor Ref Float]	weights
-> m' ControlNode

sdcaShrinkL1' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Float	l1
-> Float	l2
-> [Tensor Ref Float]	weights
-> m' ControlNode

segmentMax Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor Build t	output

segmentMax' Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor Build t	output

segmentMean Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor Build t	output

segmentMean' Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor Build t	output

segmentMin Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor Build t	output

segmentMin' Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor Build t	output

segmentProd Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor Build t	output

segmentProd' Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor Build t	output

segmentSum Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor Build t	output

segmentSum' Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor Build t	output

select Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 Bool	condition
-> Tensor v'2 t	t
-> Tensor v'3 t	e
-> Tensor Build t	output

select' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 Bool	condition
-> Tensor v'2 t	t
-> Tensor v'3 t	e
-> Tensor Build t	output

selectV2 Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 Bool	condition
-> Tensor v'2 t	t
-> Tensor v'3 t	e
-> Tensor Build t	output

selectV2' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 Bool	condition
-> Tensor v'2 t	t
-> Tensor v'3 t	e
-> Tensor Build t	output

selfAdjointEig Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	input
-> Tensor Build t	output

selfAdjointEig' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

selfAdjointEigV2 Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t

=> Tensor v'1 t

input

-> (Tensor Build t, Tensor Build t)

(e, v)

e
v

selfAdjointEigV2' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> (Tensor Build t, Tensor Build t)	(e, v) e v

selu Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	features
-> Tensor Build t	activations

selu' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	features
-> Tensor Build t	activations

seluGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	gradients
-> Tensor v'2 t	outputs
-> Tensor Build t	backprops

seluGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	gradients
-> Tensor v'2 t	outputs
-> Tensor Build t	backprops

send Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> ByteString	recv_device
-> ByteString	send_device
-> Int64	send_device_incarnation
-> ByteString	tensor_name
-> Tensor v'1 t	tensor
-> m' ControlNode

send' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> ByteString	recv_device
-> ByteString	send_device
-> Int64	send_device_incarnation
-> ByteString	tensor_name
-> Tensor v'1 t	tensor
-> m' ControlNode

sendTPUEmbeddingGradients Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> ByteString	config
-> [Tensor v'1 Float]	inputs
-> [Tensor v'2 Float]	learning_rates
-> m' ControlNode

sendTPUEmbeddingGradients' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> ByteString	config
-> [Tensor v'1 Float]	inputs
-> [Tensor v'2 Float]	learning_rates
-> m' ControlNode

serializeIterator Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	resource_handle
-> m' (Tensor Value Variant)	serialized

serializeIterator' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	resource_handle
-> m' (Tensor Value Variant)	serialized

serializeManySparse Source #

Arguments

:: forall v'1 v'2 v'3 t out_type. (TensorType t, OneOf '[ByteString, Variant] out_type)
=> Tensor v'1 Int64	sparse_indices
-> Tensor v'2 t	sparse_values
-> Tensor v'3 Int64	sparse_shape
-> Tensor Build out_type	serialized_sparse

serializeManySparse' Source #

Arguments

:: forall v'1 v'2 v'3 t out_type. (TensorType t, OneOf '[ByteString, Variant] out_type)
=> OpParams
-> Tensor v'1 Int64	sparse_indices
-> Tensor v'2 t	sparse_values
-> Tensor v'3 Int64	sparse_shape
-> Tensor Build out_type	serialized_sparse

serializeSparse Source #

Arguments

:: forall v'1 v'2 v'3 t out_type. (TensorType t, OneOf '[ByteString, Variant] out_type)
=> Tensor v'1 Int64	sparse_indices
-> Tensor v'2 t	sparse_values
-> Tensor v'3 Int64	sparse_shape
-> Tensor Build out_type	serialized_sparse

serializeSparse' Source #

Arguments

:: forall v'1 v'2 v'3 t out_type. (TensorType t, OneOf '[ByteString, Variant] out_type)
=> OpParams
-> Tensor v'1 Int64	sparse_indices
-> Tensor v'2 t	sparse_values
-> Tensor v'3 Int64	sparse_shape
-> Tensor Build out_type	serialized_sparse

serializeTensor Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	tensor
-> Tensor Build ByteString	serialized

serializeTensor' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	tensor
-> Tensor Build ByteString	serialized

setSize Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t
=> Tensor v'1 Int64	set_indices
-> Tensor v'2 t	set_values
-> Tensor v'3 Int64	set_shape
-> Tensor Build Int32	size

setSize' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t
=> OpParams
-> Tensor v'1 Int64	set_indices
-> Tensor v'2 t	set_values
-> Tensor v'3 Int64	set_shape
-> Tensor Build Int32	size

setStatsAggregatorDataset Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ResourceHandle	stats_aggregator
-> Tensor v'3 ByteString	tag
-> Tensor v'4 ByteString	counter_prefix
-> m' (Tensor Value Variant)	handle

setStatsAggregatorDataset' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ResourceHandle	stats_aggregator
-> Tensor v'3 ByteString	tag
-> Tensor v'4 ByteString	counter_prefix
-> m' (Tensor Value Variant)	handle

shape Source #

Arguments

:: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type)
=> Tensor v'1 t	input
-> Tensor Build out_type	output

shape' Source #

Arguments

:: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type)
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build out_type	output

shapeN Source #

Arguments

:: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type)
=> [Tensor v'1 t]	input
-> [Tensor Build out_type]	output

shapeN' Source #

Arguments

:: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type)
=> OpParams
-> [Tensor v'1 t]	input
-> [Tensor Build out_type]	output

shardDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_shards
-> Tensor v'3 Int64	index
-> Tensor Build Variant	handle

shardDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	num_shards
-> Tensor v'3 Int64	index
-> Tensor Build Variant	handle

shardedFilename Source #

Arguments

:: Tensor v'1 ByteString	basename
-> Tensor v'2 Int32	shard
-> Tensor v'3 Int32	num_shards
-> Tensor Build ByteString	filename

shardedFilename' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	basename
-> Tensor v'2 Int32	shard
-> Tensor v'3 Int32	num_shards
-> Tensor Build ByteString	filename

shardedFilespec Source #

Arguments

:: Tensor v'1 ByteString	basename
-> Tensor v'2 Int32	num_shards
-> Tensor Build ByteString	filename

shardedFilespec' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	basename
-> Tensor v'2 Int32	num_shards
-> Tensor Build ByteString	filename

shuffleAndRepeatDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	seed2
-> Tensor v'5 Int64	count
-> Tensor Build Variant	handle

shuffleAndRepeatDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	seed2
-> Tensor v'5 Int64	count
-> Tensor Build Variant	handle

shuffleAndRepeatDatasetV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	seed2
-> Tensor v'5 Int64	count
-> Tensor v'6 ResourceHandle	seed_generator
-> m' (Tensor Value Variant)	handle

shuffleAndRepeatDatasetV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	seed2
-> Tensor v'5 Int64	count
-> Tensor v'6 ResourceHandle	seed_generator
-> m' (Tensor Value Variant)	handle

shuffleDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	seed2
-> Tensor Build Variant	handle

shuffleDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	seed2
-> Tensor Build Variant	handle

shuffleDatasetV2 Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor v'3 ResourceHandle	seed_generator
-> m' (Tensor Value Variant)	handle

shuffleDatasetV2' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor v'3 ResourceHandle	seed_generator
-> m' (Tensor Value Variant)	handle

shuffleDatasetV3 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	seed2
-> Tensor v'5 ResourceHandle	seed_generator
-> m' (Tensor Value Variant)	handle

shuffleDatasetV3' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	buffer_size
-> Tensor v'3 Int64	seed
-> Tensor v'4 Int64	seed2
-> Tensor v'5 ResourceHandle	seed_generator
-> m' (Tensor Value Variant)	handle

shutdownDistributedTPU :: forall m'. MonadBuild m' => m' ControlNode Source #

shutdownDistributedTPU' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode Source #

sigmoid Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

sigmoid' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

sigmoidGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

sigmoidGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

sign Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

sign' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

sin Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

sin' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

sinh Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

sinh' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

size Source #

Arguments

:: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type)
=> Tensor v'1 t	input
-> Tensor Build out_type	output

size' Source #

Arguments

:: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type)
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build out_type	output

skipDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	count
-> Tensor Build Variant	handle

skipDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	count
-> Tensor Build Variant	handle

skipgram Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	batch_size
-> ByteString	filename
-> 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) vocab_word vocab_freq words_per_epoch current_epoch total_words_processed examples labels

skipgram' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	batch_size
-> ByteString	filename
-> 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) vocab_word vocab_freq words_per_epoch current_epoch total_words_processed examples labels

sleepDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	sleep_microseconds
-> Tensor Build Variant	handle

sleepDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	sleep_microseconds
-> Tensor Build Variant	handle

slice Source #

Arguments

:: forall v'1 v'2 v'3 t index. (TensorType t, OneOf '[Int32, Int64] index)
=> Tensor v'1 t	input
-> Tensor v'2 index	begin
-> Tensor v'3 index	size
-> Tensor Build t	output

slice' Source #

Arguments

:: forall v'1 v'2 v'3 t index. (TensorType t, OneOf '[Int32, Int64] index)
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 index	begin
-> Tensor v'3 index	size
-> Tensor Build t	output

slidingWindowDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	window_size
-> Tensor v'3 Int64	window_shift
-> Tensor v'4 Int64	window_stride
-> Tensor Build Variant	handle

slidingWindowDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	window_size
-> Tensor v'3 Int64	window_shift
-> Tensor v'4 Int64	window_stride
-> Tensor Build Variant	handle

snapshot Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	output

snapshot' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

snapshotDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	path
-> Tensor Build Variant	handle

snapshotDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ByteString	path
-> Tensor Build Variant	handle

sobolSample Source #

Arguments

:: forall v'1 v'2 v'3 dtype. OneOf '[Double, Float] dtype
=> Tensor v'1 Int32	dim
-> Tensor v'2 Int32	num_results
-> Tensor v'3 Int32	skip
-> Tensor Build dtype	samples

sobolSample' Source #

Arguments

:: forall v'1 v'2 v'3 dtype. OneOf '[Double, Float] dtype
=> OpParams
-> Tensor v'1 Int32	dim
-> Tensor v'2 Int32	num_results
-> Tensor v'3 Int32	skip
-> Tensor Build dtype	samples

softmax Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	logits
-> Tensor Build t	softmax

softmax' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	logits
-> Tensor Build t	softmax

softmaxCrossEntropyWithLogits Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	features
-> Tensor v'2 t	labels
-> (Tensor Build t, Tensor Build t)	(loss, backprop) loss backprop

softmaxCrossEntropyWithLogits' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	features
-> Tensor v'2 t	labels
-> (Tensor Build t, Tensor Build t)	(loss, backprop) loss backprop

softplus Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	features
-> Tensor Build t	activations

softplus' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	features
-> Tensor Build t	activations

softplusGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	gradients
-> Tensor v'2 t	features
-> Tensor Build t	backprops

softplusGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	gradients
-> Tensor v'2 t	features
-> Tensor Build t	backprops

softsign Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	features
-> Tensor Build t	activations

softsign' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	features
-> Tensor Build t	activations

softsignGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	gradients
-> Tensor v'2 t	features
-> Tensor Build t	backprops

softsignGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	gradients
-> Tensor v'2 t	features
-> Tensor Build t	backprops

spaceToBatch Source #

Arguments

:: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings)
=> Int64	block_size
-> Tensor v'1 t	input
-> Tensor v'2 tpaddings	paddings
-> Tensor Build t	output

spaceToBatch' Source #

Arguments

:: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings)
=> OpParams
-> Int64	block_size
-> Tensor v'1 t	input
-> Tensor v'2 tpaddings	paddings
-> Tensor Build t	output

spaceToBatchND Source #

Arguments

:: 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	input
-> Tensor v'2 tblock_shape	block_shape
-> Tensor v'3 tpaddings	paddings
-> Tensor Build t	output

spaceToBatchND' Source #

Arguments

:: 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	input
-> Tensor v'2 tblock_shape	block_shape
-> Tensor v'3 tpaddings	paddings
-> Tensor Build t	output

spaceToDepth Source #

Arguments

:: forall v'1 t. TensorType t
=> Int64	block_size
-> Tensor v'1 t	input
-> Tensor Build t	output

spaceToDepth' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Int64	block_size
-> Tensor v'1 t	input
-> Tensor Build t	output

sparseAccumulatorApplyGradient Source #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	handle
-> Tensor v'2 Int32	num_required
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)	(indices, values, shape) indices values shape

sparseAccumulatorTakeGradient' Source #

Arguments

:: 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	handle
-> Tensor v'2 Int32	num_required
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)	(indices, values, shape) indices values shape

sparseAdd Source #

Arguments

:: 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	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) sum_indices sum_values sum_shape

sparseAdd' Source #

Arguments

:: 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	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) sum_indices sum_values sum_shape

sparseAddGrad Source #

Arguments

:: 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	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) a_val_grad b_val_grad

sparseAddGrad' Source #

Arguments

:: 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	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) a_val_grad b_val_grad

sparseApplyAdadelta Source #

Arguments

:: 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	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

sparseApplyAdadelta' Source #

Arguments

:: 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	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

sparseApplyAdagrad Source #

Arguments

:: 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	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

sparseApplyAdagrad' Source #

Arguments

:: 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	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

sparseApplyAdagradDA Source #

Arguments

:: 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	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 #

Arguments

:: 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	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

sparseApplyAdagradV2 Source #

Arguments

:: 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	var
-> Tensor Ref t	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	epsilon
-> Tensor v'5 t	grad
-> Tensor v'6 tindices	indices
-> m' (Tensor Ref t)	out

sparseApplyAdagradV2' Source #

Arguments

:: 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	var
-> Tensor Ref t	accum
-> Tensor v'3 t	lr
-> Tensor v'4 t	epsilon
-> Tensor v'5 t	grad
-> Tensor v'6 tindices	indices
-> m' (Tensor Ref t)	out

sparseApplyCenteredRMSProp Source #

Arguments

:: 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	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 #

Arguments

:: 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	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

sparseApplyFtrl Source #

Arguments

:: 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	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

sparseApplyFtrl' Source #

Arguments

:: 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	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

sparseApplyFtrlV2 Source #

Arguments

:: 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	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

sparseApplyFtrlV2' Source #

Arguments

:: 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	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

sparseApplyMomentum Source #

Arguments

:: 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	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

sparseApplyMomentum' Source #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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 #

Arguments

:: 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	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

sparseApplyRMSProp Source #

Arguments

:: 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	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

sparseApplyRMSProp' Source #

Arguments

:: 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	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

sparseBincount Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t)
=> Tensor v'1 Int64	indices
-> Tensor v'2 tidx	values
-> Tensor v'3 Int64	dense_shape
-> Tensor v'4 tidx	size
-> Tensor v'5 t	weights
-> Tensor Build t	output

sparseBincount' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t)
=> OpParams
-> Tensor v'1 Int64	indices
-> Tensor v'2 tidx	values
-> Tensor v'3 Int64	dense_shape
-> Tensor v'4 tidx	size
-> Tensor v'5 t	weights
-> Tensor Build t	output

sparseConcat Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Int64	concat_dim
-> [Tensor v'1 Int64]	indices
-> [Tensor v'2 t]	values
-> [Tensor v'3 Int64]	shapes
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)	(output_indices, output_values, output_shape) output_indices output_values output_shape

sparseConcat' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Int64	concat_dim
-> [Tensor v'1 Int64]	indices
-> [Tensor v'2 t]	values
-> [Tensor v'3 Int64]	shapes
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)	(output_indices, output_values, output_shape) output_indices output_values output_shape

sparseConditionalAccumulator Source #

Arguments

:: forall m'. MonadBuild m'
=> DataType	dtype
-> Shape	shape
-> m' (Tensor Ref ByteString)	handle

sparseConditionalAccumulator' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> DataType	dtype
-> Shape	shape
-> m' (Tensor Ref ByteString)	handle

sparseCountSparseOutput Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type)
=> Bool	binary_output
-> Tensor v'1 Int64	indices
-> Tensor v'2 t	values
-> Tensor v'3 Int64	dense_shape
-> Tensor v'4 output_type	weights
-> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)	(output_indices, output_values, output_dense_shape) output_indices output_values output_dense_shape

sparseCountSparseOutput' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type)
=> OpParams
-> Bool	binary_output
-> Tensor v'1 Int64	indices
-> Tensor v'2 t	values
-> Tensor v'3 Int64	dense_shape
-> Tensor v'4 output_type	weights
-> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)	(output_indices, output_values, output_dense_shape) output_indices output_values output_dense_shape

sparseCross Source #

Arguments

:: 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	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) output_indices output_values output_shape

sparseCross' Source #

Arguments

:: 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	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) output_indices output_values output_shape

sparseCrossHashed Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 sparse_types dense_types. (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types)
=> [Tensor v'1 Int64]	indices
-> TensorList v'2 sparse_types	values
-> [Tensor v'3 Int64]	shapes
-> TensorList v'4 dense_types	dense_inputs
-> Tensor v'5 Int64	num_buckets
-> Tensor v'6 Bool	strong_hash
-> Tensor v'7 Int64	salt
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64)	(output_indices, output_values, output_shape) output_indices output_values output_shape

sparseCrossHashed' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 sparse_types dense_types. (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types)
=> OpParams
-> [Tensor v'1 Int64]	indices
-> TensorList v'2 sparse_types	values
-> [Tensor v'3 Int64]	shapes
-> TensorList v'4 dense_types	dense_inputs
-> Tensor v'5 Int64	num_buckets
-> Tensor v'6 Bool	strong_hash
-> Tensor v'7 Int64	salt
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64)	(output_indices, output_values, output_shape) output_indices output_values output_shape

sparseCrossV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 sparse_types dense_types. (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types)
=> [Tensor v'1 Int64]	indices
-> TensorList v'2 sparse_types	values
-> [Tensor v'3 Int64]	shapes
-> TensorList v'4 dense_types	dense_inputs
-> Tensor v'5 ByteString	sep
-> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64)	(output_indices, output_values, output_shape) output_indices output_values output_shape

sparseCrossV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 sparse_types dense_types. (OneOfs '[ByteString, Int64] sparse_types, OneOfs '[ByteString, Int64] dense_types)
=> OpParams
-> [Tensor v'1 Int64]	indices
-> TensorList v'2 sparse_types	values
-> [Tensor v'3 Int64]	shapes
-> TensorList v'4 dense_types	dense_inputs
-> Tensor v'5 ByteString	sep
-> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64)	(output_indices, output_values, output_shape) output_indices output_values output_shape

sparseDenseCwiseAdd Source #

Arguments

:: 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	sp_indices
-> Tensor v'2 t	sp_values
-> Tensor v'3 Int64	sp_shape
-> Tensor v'4 t	dense
-> Tensor Build t	output

sparseDenseCwiseAdd' Source #

Arguments

:: 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	sp_indices
-> Tensor v'2 t	sp_values
-> Tensor v'3 Int64	sp_shape
-> Tensor v'4 t	dense
-> Tensor Build t	output

sparseDenseCwiseDiv Source #

Arguments

:: 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	sp_indices
-> Tensor v'2 t	sp_values
-> Tensor v'3 Int64	sp_shape
-> Tensor v'4 t	dense
-> Tensor Build t	output

sparseDenseCwiseDiv' Source #

Arguments

:: 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	sp_indices
-> Tensor v'2 t	sp_values
-> Tensor v'3 Int64	sp_shape
-> Tensor v'4 t	dense
-> Tensor Build t	output

sparseDenseCwiseMul Source #

Arguments

:: 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	sp_indices
-> Tensor v'2 t	sp_values
-> Tensor v'3 Int64	sp_shape
-> Tensor v'4 t	dense
-> Tensor Build t	output

sparseDenseCwiseMul' Source #

Arguments

:: 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	sp_indices
-> Tensor v'2 t	sp_values
-> Tensor v'3 Int64	sp_shape
-> Tensor v'4 t	dense
-> Tensor Build t	output

sparseFillEmptyRows Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. 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) output_indices output_values empty_row_indicator reverse_index_map

sparseFillEmptyRows' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. 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) output_indices output_values empty_row_indicator reverse_index_map

sparseFillEmptyRowsGrad Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Tensor v'1 Int64	reverse_index_map
-> Tensor v'2 t	grad_values
-> (Tensor Build t, Tensor Build t)	(d_values, d_default_value) d_values d_default_value

sparseFillEmptyRowsGrad' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Tensor v'1 Int64	reverse_index_map
-> Tensor v'2 t	grad_values
-> (Tensor Build t, Tensor Build t)	(d_values, d_default_value) d_values d_default_value

sparseMatMul Source #

Arguments

:: forall v'1 v'2 ta tb. (OneOf '[Word16, Float] ta, OneOf '[Word16, Float] tb)
=> Tensor v'1 ta	a
-> Tensor v'2 tb	b
-> Tensor Build Float	product

sparseMatMul' Source #

Arguments

:: forall v'1 v'2 ta tb. (OneOf '[Word16, Float] ta, OneOf '[Word16, Float] tb)
=> OpParams
-> Tensor v'1 ta	a
-> Tensor v'2 tb	b
-> Tensor Build Float	product

sparseMatrixAdd Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> Tensor v'1 Variant	a
-> Tensor v'2 Variant	b
-> Tensor v'3 t	alpha
-> Tensor v'4 t	beta
-> Tensor Build Variant	c

sparseMatrixAdd' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> OpParams
-> Tensor v'1 Variant	a
-> Tensor v'2 Variant	b
-> Tensor v'3 t	alpha
-> Tensor v'4 t	beta
-> Tensor Build Variant	c

sparseMatrixMatMul Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Tensor v'1 Variant	a
-> Tensor v'2 t	b
-> Tensor Build t	output

sparseMatrixMatMul' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Tensor v'1 Variant	a
-> Tensor v'2 t	b
-> Tensor Build t	output

sparseMatrixMul Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Tensor v'1 Variant	a
-> Tensor v'2 t	b
-> Tensor Build Variant	output

sparseMatrixMul' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Tensor v'1 Variant	a
-> Tensor v'2 t	b
-> Tensor Build Variant	output

sparseMatrixNNZ Source #

Arguments

:: Tensor v'1 Variant	sparse_matrix
-> Tensor Build Int32	nnz

sparseMatrixNNZ' Source #

Arguments

:: OpParams
-> Tensor v'1 Variant	sparse_matrix
-> Tensor Build Int32	nnz

sparseMatrixOrderingAMD Source #

Arguments

:: Tensor v'1 Variant	input
-> Tensor Build Int32	output

sparseMatrixOrderingAMD' Source #

Arguments

:: OpParams
-> Tensor v'1 Variant	input
-> Tensor Build Int32	output

sparseMatrixSoftmax Source #

Arguments

:: DataType	type
-> Tensor v'1 Variant	logits
-> Tensor Build Variant	softmax

sparseMatrixSoftmax' Source #

Arguments

:: OpParams
-> DataType	type
-> Tensor v'1 Variant	logits
-> Tensor Build Variant	softmax

sparseMatrixSoftmaxGrad Source #

Arguments

:: DataType	type
-> Tensor v'1 Variant	softmax
-> Tensor v'2 Variant	grad_softmax
-> Tensor Build Variant	gradient

sparseMatrixSoftmaxGrad' Source #

Arguments

:: OpParams
-> DataType	type
-> Tensor v'1 Variant	softmax
-> Tensor v'2 Variant	grad_softmax
-> Tensor Build Variant	gradient

sparseMatrixSparseCholesky Source #

Arguments

:: DataType	type
-> Tensor v'1 Variant	input
-> Tensor v'2 Int32	permutation
-> Tensor Build Variant	output

sparseMatrixSparseCholesky' Source #

Arguments

:: OpParams
-> DataType	type
-> Tensor v'1 Variant	input
-> Tensor v'2 Int32	permutation
-> Tensor Build Variant	output

sparseMatrixSparseMatMul Source #

Arguments

:: DataType	type
-> Tensor v'1 Variant	a
-> Tensor v'2 Variant	b
-> Tensor Build Variant	c

sparseMatrixSparseMatMul' Source #

Arguments

:: OpParams
-> DataType	type
-> Tensor v'1 Variant	a
-> Tensor v'2 Variant	b
-> Tensor Build Variant	c

sparseMatrixTranspose Source #

Arguments

:: DataType	type
-> Tensor v'1 Variant	input
-> Tensor Build Variant	output

sparseMatrixTranspose' Source #

Arguments

:: OpParams
-> DataType	type
-> Tensor v'1 Variant	input
-> Tensor Build Variant	output

sparseMatrixZeros Source #

Arguments

:: DataType	type
-> Tensor v'1 Int64	dense_shape
-> Tensor Build Variant	sparse_matrix

sparseMatrixZeros' Source #

Arguments

:: OpParams
-> DataType	type
-> Tensor v'1 Int64	dense_shape
-> Tensor Build Variant	sparse_matrix

sparseReduceMax Source #

Arguments

:: 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	input_indices
-> Tensor v'2 t	input_values
-> Tensor v'3 Int64	input_shape
-> Tensor v'4 Int32	reduction_axes
-> Tensor Build t	output

sparseReduceMax' Source #

Arguments

:: 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	input_indices
-> Tensor v'2 t	input_values
-> Tensor v'3 Int64	input_shape
-> Tensor v'4 Int32	reduction_axes
-> Tensor Build t	output

sparseReduceMaxSparse Source #

Arguments

:: 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	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) output_indices output_values output_shape

sparseReduceMaxSparse' Source #

Arguments

:: 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	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) output_indices output_values output_shape

sparseReduceSum Source #

Arguments

:: 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	input_indices
-> Tensor v'2 t	input_values
-> Tensor v'3 Int64	input_shape
-> Tensor v'4 Int32	reduction_axes
-> Tensor Build t	output

sparseReduceSum' Source #

Arguments

:: 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	input_indices
-> Tensor v'2 t	input_values
-> Tensor v'3 Int64	input_shape
-> Tensor v'4 Int32	reduction_axes
-> Tensor Build t	output

sparseReduceSumSparse Source #

Arguments

:: 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	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) output_indices output_values output_shape

sparseReduceSumSparse' Source #

Arguments

:: 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	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) output_indices output_values output_shape

sparseReorder Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 Int64	input_indices
-> Tensor v'2 t	input_values
-> Tensor v'3 Int64	input_shape
-> (Tensor Build Int64, Tensor Build t)	(output_indices, output_values) output_indices output_values

sparseReorder' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 Int64	input_indices
-> Tensor v'2 t	input_values
-> Tensor v'3 Int64	input_shape
-> (Tensor Build Int64, Tensor Build t)	(output_indices, output_values) output_indices output_values

sparseReshape Source #

Arguments

:: Tensor v'1 Int64	input_indices
-> Tensor v'2 Int64	input_shape
-> Tensor v'3 Int64	new_shape
-> (Tensor Build Int64, Tensor Build Int64)	(output_indices, output_shape) output_indices output_shape

sparseReshape' Source #

Arguments

:: OpParams
-> Tensor v'1 Int64	input_indices
-> Tensor v'2 Int64	input_shape
-> Tensor v'3 Int64	new_shape
-> (Tensor Build Int64, Tensor Build Int64)	(output_indices, output_shape) output_indices output_shape

sparseSegmentMean Source #

Arguments

:: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids)
=> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor Build t	output

sparseSegmentMean' Source #

Arguments

:: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids)
=> OpParams
-> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor Build t	output

sparseSegmentMeanGrad Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tidx tsegmentids. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids)
=> Tensor v'1 t	grad
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor v'4 Int32	output_dim0
-> Tensor Build t	output

sparseSegmentMeanGrad' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tidx tsegmentids. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids)
=> OpParams
-> Tensor v'1 t	grad
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor v'4 Int32	output_dim0
-> Tensor Build t	output

sparseSegmentMeanWithNumSegments Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids)
=> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor v'4 tnumsegments	num_segments
-> Tensor Build t	output

sparseSegmentMeanWithNumSegments' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids)
=> OpParams
-> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor v'4 tnumsegments	num_segments
-> Tensor Build t	output

sparseSegmentSqrtN Source #

Arguments

:: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids)
=> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor Build t	output

sparseSegmentSqrtN' Source #

Arguments

:: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids)
=> OpParams
-> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor Build t	output

sparseSegmentSqrtNGrad Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tidx tsegmentids. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids)
=> Tensor v'1 t	grad
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor v'4 Int32	output_dim0
-> Tensor Build t	output

sparseSegmentSqrtNGrad' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tidx tsegmentids. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids)
=> OpParams
-> Tensor v'1 t	grad
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor v'4 Int32	output_dim0
-> Tensor Build t	output

sparseSegmentSqrtNWithNumSegments Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids)
=> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor v'4 tnumsegments	num_segments
-> Tensor Build t	output

sparseSegmentSqrtNWithNumSegments' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids)
=> OpParams
-> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor v'4 tnumsegments	num_segments
-> Tensor Build t	output

sparseSegmentSum Source #

Arguments

:: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids)
=> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor Build t	output

sparseSegmentSum' Source #

Arguments

:: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids)
=> OpParams
-> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor Build t	output

sparseSegmentSumWithNumSegments Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids)
=> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor v'4 tnumsegments	num_segments
-> Tensor Build t	output

sparseSegmentSumWithNumSegments' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tidx tnumsegments tsegmentids. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments, OneOf '[Int32, Int64] tsegmentids)
=> OpParams
-> Tensor v'1 t	data
-> Tensor v'2 tidx	indices
-> Tensor v'3 tsegmentids	segment_ids
-> Tensor v'4 tnumsegments	num_segments
-> Tensor Build t	output

sparseSlice Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t
=> Tensor v'1 Int64	indices
-> Tensor v'2 t	values
-> Tensor v'3 Int64	shape
-> Tensor v'4 Int64	start
-> Tensor v'5 Int64	size
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)	(output_indices, output_values, output_shape) output_indices output_values output_shape

sparseSlice' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t
=> OpParams
-> Tensor v'1 Int64	indices
-> Tensor v'2 t	values
-> Tensor v'3 Int64	shape
-> Tensor v'4 Int64	start
-> Tensor v'5 Int64	size
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)	(output_indices, output_values, output_shape) output_indices output_values output_shape

sparseSliceGrad Source #

Arguments

:: 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	backprop_val_grad
-> Tensor v'2 Int64	input_indices
-> Tensor v'3 Int64	input_start
-> Tensor v'4 Int64	output_indices
-> Tensor Build t	val_grad

sparseSliceGrad' Source #

Arguments

:: 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	backprop_val_grad
-> Tensor v'2 Int64	input_indices
-> Tensor v'3 Int64	input_start
-> Tensor v'4 Int64	output_indices
-> Tensor Build t	val_grad

sparseSoftmax Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t
=> Tensor v'1 Int64	sp_indices
-> Tensor v'2 t	sp_values
-> Tensor v'3 Int64	sp_shape
-> Tensor Build t	output

sparseSoftmax' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 Int64	sp_indices
-> Tensor v'2 t	sp_values
-> Tensor v'3 Int64	sp_shape
-> Tensor Build t	output

sparseSoftmaxCrossEntropyWithLogits Source #

Arguments

:: forall v'1 v'2 t tlabels. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tlabels)
=> Tensor v'1 t	features
-> Tensor v'2 tlabels	labels
-> (Tensor Build t, Tensor Build t)	(loss, backprop) loss backprop

sparseSoftmaxCrossEntropyWithLogits' Source #

Arguments

:: forall v'1 v'2 t tlabels. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tlabels)
=> OpParams
-> Tensor v'1 t	features
-> Tensor v'2 tlabels	labels
-> (Tensor Build t, Tensor Build t)	(loss, backprop) loss backprop

sparseSparseMaximum Source #

Arguments

:: 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	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) output_indices output_values

sparseSparseMaximum' Source #

Arguments

:: 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	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) output_indices output_values

sparseSparseMinimum Source #

Arguments

:: 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	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) output_indices output_values

sparseSparseMinimum' Source #

Arguments

:: 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	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) output_indices output_values

sparseSplit Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> Int64	num_split
-> Tensor v'1 Int64	split_dim
-> Tensor v'2 Int64	indices
-> Tensor v'3 t	values
-> Tensor v'4 Int64	shape
-> ([Tensor Build Int64], [Tensor Build t], [Tensor Build Int64])	(output_indices, output_values, output_shape) output_indices output_values output_shape

sparseSplit' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> OpParams
-> Int64	num_split
-> Tensor v'1 Int64	split_dim
-> Tensor v'2 Int64	indices
-> Tensor v'3 t	values
-> Tensor v'4 Int64	shape
-> ([Tensor Build Int64], [Tensor Build t], [Tensor Build Int64])	(output_indices, output_values, output_shape) output_indices output_values output_shape

sparseTensorDenseAdd Source #

Arguments

:: 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	a_indices
-> Tensor v'2 t	a_values
-> Tensor v'3 tindices	a_shape
-> Tensor v'4 t	b
-> Tensor Build t	output

sparseTensorDenseAdd' Source #

Arguments

:: 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	a_indices
-> Tensor v'2 t	a_values
-> Tensor v'3 tindices	a_shape
-> Tensor v'4 t	b
-> Tensor Build t	output

sparseTensorDenseMatMul Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 tindices	a_indices
-> Tensor v'2 t	a_values
-> Tensor v'3 Int64	a_shape
-> Tensor v'4 t	b
-> Tensor Build t	product

sparseTensorDenseMatMul' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 tindices	a_indices
-> Tensor v'2 t	a_values
-> Tensor v'3 Int64	a_shape
-> Tensor v'4 t	b
-> Tensor Build t	product

sparseTensorSliceDataset Source #

Arguments

:: forall v'1 v'2 v'3 tvalues m'. (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

sparseTensorSliceDataset' Source #

Arguments

:: forall v'1 v'2 v'3 tvalues m'. (MonadBuild m', TensorType tvalues)
=> OpParams
-> Tensor v'1 Int64	indices
-> Tensor v'2 tvalues	values
-> Tensor v'3 Int64	dense_shape
-> m' (Tensor Value Variant)	handle

sparseTensorToCSRSparseMatrix Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> Tensor v'1 Int64	indices
-> Tensor v'2 t	values
-> Tensor v'3 Int64	dense_shape
-> Tensor Build Variant	sparse_matrix

sparseTensorToCSRSparseMatrix' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> OpParams
-> Tensor v'1 Int64	indices
-> Tensor v'2 t	values
-> Tensor v'3 Int64	dense_shape
-> Tensor Build Variant	sparse_matrix

sparseToDense Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 tindices	sparse_indices
-> Tensor v'2 tindices	output_shape
-> Tensor v'3 t	sparse_values
-> Tensor v'4 t	default_value
-> Tensor Build t	dense

sparseToDense' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 tindices	sparse_indices
-> Tensor v'2 tindices	output_shape
-> Tensor v'3 t	sparse_values
-> Tensor v'4 t	default_value
-> Tensor Build t	dense

sparseToSparseSetOperation Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t
=> ByteString	set_operation
-> 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) result_indices result_values result_shape

sparseToSparseSetOperation' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t
=> OpParams
-> ByteString	set_operation
-> 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) result_indices result_values result_shape

spence Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

spence' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

split Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Int64	num_split
-> Tensor v'1 Int32	split_dim
-> Tensor v'2 t	value
-> [Tensor Build t]	output

split' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Int64	num_split
-> Tensor v'1 Int32	split_dim
-> Tensor v'2 t	value
-> [Tensor Build t]	output

splitV Source #

Arguments

:: forall v'1 v'2 v'3 t tlen. (TensorType t, OneOf '[Int32, Int64] tlen)
=> Int64	num_split
-> Tensor v'1 t	value
-> Tensor v'2 tlen	size_splits
-> Tensor v'3 Int32	split_dim
-> [Tensor Build t]	output

splitV' Source #

Arguments

:: forall v'1 v'2 v'3 t tlen. (TensorType t, OneOf '[Int32, Int64] tlen)
=> OpParams
-> Int64	num_split
-> Tensor v'1 t	value
-> Tensor v'2 tlen	size_splits
-> Tensor v'3 Int32	split_dim
-> [Tensor Build t]	output

sqlDataset Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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

sqlDataset' Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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

sqrt Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

sqrt' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

sqrtGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

sqrtGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

square Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

square' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

squaredDifference Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

squaredDifference' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

squeeze Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	output

squeeze' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

stack Source #

Arguments

:: forall m'. MonadBuild m'
=> DataType	elem_type
-> m' (Tensor Ref ByteString)	handle

stack' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> DataType	elem_type
-> m' (Tensor Ref ByteString)	handle

stackClose Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	handle
-> m' ControlNode

stackClose' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	handle
-> m' ControlNode

stackCloseV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> m' ControlNode

stackCloseV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> m' ControlNode

stackPop Source #

Arguments

:: forall elem_type m'. (MonadBuild m', TensorType elem_type)
=> Tensor Ref ByteString	handle
-> m' (Tensor Value elem_type)	elem

stackPop' Source #

Arguments

:: forall elem_type m'. (MonadBuild m', TensorType elem_type)
=> OpParams
-> Tensor Ref ByteString	handle
-> m' (Tensor Value elem_type)	elem

stackPopV2 Source #

Arguments

:: forall v'1 elem_type m'. (MonadBuild m', TensorType elem_type)
=> Tensor v'1 ResourceHandle	handle
-> m' (Tensor Value elem_type)	elem

stackPopV2' Source #

Arguments

:: forall v'1 elem_type m'. (MonadBuild m', TensorType elem_type)
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> m' (Tensor Value elem_type)	elem

stackPush Source #

Arguments

:: forall v'2 t m'. (MonadBuild m', TensorType t)
=> Tensor Ref ByteString	handle
-> Tensor v'2 t	elem
-> m' (Tensor Value t)	output

stackPush' Source #

Arguments

:: forall v'2 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor Ref ByteString	handle
-> Tensor v'2 t	elem
-> m' (Tensor Value t)	output

stackPushV2 Source #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', TensorType t)
=> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 t	elem
-> m' (Tensor Value t)	output

stackPushV2' Source #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 t	elem
-> m' (Tensor Value t)	output

stackV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> DataType	elem_type
-> Tensor v'1 Int32	max_size
-> m' (Tensor Value ResourceHandle)	handle

stackV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> DataType	elem_type
-> Tensor v'1 Int32	max_size
-> m' (Tensor Value ResourceHandle)	handle

stage Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> TensorList v'1 dtypes	values
-> m' ControlNode

stage' Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> TensorList v'1 dtypes	values
-> m' ControlNode

stageClear Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	dtypes
-> m' ControlNode

stageClear' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	dtypes
-> m' ControlNode

stagePeek Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> Tensor v'1 Int32	index
-> m' (TensorList Value dtypes)	values

stagePeek' Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> Tensor v'1 Int32	index
-> m' (TensorList Value dtypes)	values

stageSize Source #

Arguments

:: forall m'. MonadBuild m'
=> [DataType]	dtypes
-> m' (Tensor Value Int32)	size

stageSize' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> [DataType]	dtypes
-> m' (Tensor Value Int32)	size

statefulRandomBinomial Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 s t dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64, Word16, Double, Float] dtype)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 s	shape
-> Tensor v'4 t	counts
-> Tensor v'5 t	probs
-> m' (Tensor Value dtype)	output

statefulRandomBinomial' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 s t dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64, Word16, Double, Float] dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 s	shape
-> Tensor v'4 t	counts
-> Tensor v'5 t	probs
-> m' (Tensor Value dtype)	output

statefulStandardNormal Source #

Arguments

:: forall v'1 v'2 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 shape_dtype	shape
-> m' (Tensor Value dtype)	output

statefulStandardNormal' Source #

Arguments

:: forall v'1 v'2 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 shape_dtype	shape
-> m' (Tensor Value dtype)	output

statefulStandardNormalV2 Source #

Arguments

:: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 shape_dtype	shape
-> m' (Tensor Value dtype)	output

statefulStandardNormalV2' Source #

Arguments

:: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 shape_dtype	shape
-> m' (Tensor Value dtype)	output

statefulTruncatedNormal Source #

Arguments

:: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 shape_dtype	shape
-> m' (Tensor Value dtype)	output

statefulTruncatedNormal' Source #

Arguments

:: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 shape_dtype	shape
-> m' (Tensor Value dtype)	output

statefulUniform Source #

Arguments

:: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 shape_dtype	shape
-> m' (Tensor Value dtype)	output

statefulUniform' Source #

Arguments

:: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 shape_dtype	shape
-> m' (Tensor Value dtype)	output

statefulUniformFullInt Source #

Arguments

:: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 shape_dtype	shape
-> m' (Tensor Value dtype)	output

statefulUniformFullInt' Source #

Arguments

:: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 shape_dtype	shape
-> m' (Tensor Value dtype)	output

statefulUniformInt Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 shape_dtype	shape
-> Tensor v'4 dtype	minval
-> Tensor v'5 dtype	maxval
-> m' (Tensor Value dtype)	output

statefulUniformInt' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype)
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> Tensor v'2 Int64	algorithm
-> Tensor v'3 shape_dtype	shape
-> Tensor v'4 dtype	minval
-> Tensor v'5 dtype	maxval
-> m' (Tensor Value dtype)	output

statelessMultinomial Source #

Arguments

:: 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	logits
-> Tensor v'2 Int32	num_samples
-> Tensor v'3 tseed	seed
-> Tensor Build output_dtype	output

statelessMultinomial' Source #

Arguments

:: 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	logits
-> Tensor v'2 Int32	num_samples
-> Tensor v'3 tseed	seed
-> Tensor Build output_dtype	output

statelessParameterizedTruncatedNormal Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 s tseed dtype. (OneOf '[Int32, Int64] s, OneOf '[Int32, Int64] tseed, OneOf '[Word16, Double, Float] dtype)
=> Tensor v'1 s	shape
-> Tensor v'2 tseed	seed
-> Tensor v'3 dtype	means
-> Tensor v'4 dtype	stddevs
-> Tensor v'5 dtype	minvals
-> Tensor v'6 dtype	maxvals
-> Tensor Build dtype	output

statelessParameterizedTruncatedNormal' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 s tseed dtype. (OneOf '[Int32, Int64] s, OneOf '[Int32, Int64] tseed, OneOf '[Word16, Double, Float] dtype)
=> OpParams
-> Tensor v'1 s	shape
-> Tensor v'2 tseed	seed
-> Tensor v'3 dtype	means
-> Tensor v'4 dtype	stddevs
-> Tensor v'5 dtype	minvals
-> Tensor v'6 dtype	maxvals
-> Tensor Build dtype	output

statelessRandomBinomial Source #

Arguments

:: forall v'1 v'2 v'3 v'4 s tseed t dtype. (OneOf '[Int32, Int64] s, OneOf '[Int32, Int64] tseed, OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64, Word16, Double, Float] dtype)
=> Tensor v'1 s	shape
-> Tensor v'2 tseed	seed
-> Tensor v'3 t	counts
-> Tensor v'4 t	probs
-> Tensor Build dtype	output

statelessRandomBinomial' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 s tseed t dtype. (OneOf '[Int32, Int64] s, OneOf '[Int32, Int64] tseed, OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64, Word16, Double, Float] dtype)
=> OpParams
-> Tensor v'1 s	shape
-> Tensor v'2 tseed	seed
-> Tensor v'3 t	counts
-> Tensor v'4 t	probs
-> Tensor Build dtype	output

statelessRandomGammaV2 Source #

Arguments

:: forall v'1 v'2 v'3 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed)
=> Tensor v'1 t	shape
-> Tensor v'2 tseed	seed
-> Tensor v'3 dtype	alpha
-> Tensor Build dtype	output

statelessRandomGammaV2' Source #

Arguments

:: forall v'1 v'2 v'3 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed)
=> OpParams
-> Tensor v'1 t	shape
-> Tensor v'2 tseed	seed
-> Tensor v'3 dtype	alpha
-> Tensor Build dtype	output

statelessRandomNormal Source #

Arguments

:: 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	shape
-> Tensor v'2 tseed	seed
-> Tensor Build dtype	output

statelessRandomNormal' Source #

Arguments

:: 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	shape
-> Tensor v'2 tseed	seed
-> Tensor Build dtype	output

statelessRandomPoisson Source #

Arguments

:: forall v'1 v'2 v'3 rtype dtype t tseed. (OneOf '[Int32, Int64, Word16, Double, Float] rtype, OneOf '[Int32, Int64, Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed)
=> Tensor v'1 t	shape
-> Tensor v'2 tseed	seed
-> Tensor v'3 rtype	lam
-> Tensor Build dtype	output

statelessRandomPoisson' Source #

Arguments

:: forall v'1 v'2 v'3 rtype dtype t tseed. (OneOf '[Int32, Int64, Word16, Double, Float] rtype, OneOf '[Int32, Int64, Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed)
=> OpParams
-> Tensor v'1 t	shape
-> Tensor v'2 tseed	seed
-> Tensor v'3 rtype	lam
-> Tensor Build dtype	output

statelessRandomUniform Source #

Arguments

:: 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	shape
-> Tensor v'2 tseed	seed
-> Tensor Build dtype	output

statelessRandomUniform' Source #

Arguments

:: 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	shape
-> Tensor v'2 tseed	seed
-> Tensor Build dtype	output

statelessRandomUniformFullInt Source #

Arguments

:: forall v'1 v'2 dtype t tseed. (OneOf '[Int32, Int64, Word32, Word64] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Word32, Word64] tseed)
=> Tensor v'1 t	shape
-> Tensor v'2 tseed	seed
-> Tensor Build dtype	output

statelessRandomUniformFullInt' Source #

Arguments

:: forall v'1 v'2 dtype t tseed. (OneOf '[Int32, Int64, Word32, Word64] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Word32, Word64] tseed)
=> OpParams
-> Tensor v'1 t	shape
-> Tensor v'2 tseed	seed
-> Tensor Build dtype	output

statelessRandomUniformInt Source #

Arguments

:: forall v'1 v'2 v'3 v'4 dtype t tseed. (OneOf '[Int32, Int64] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed)
=> Tensor v'1 t	shape
-> Tensor v'2 tseed	seed
-> Tensor v'3 dtype	minval
-> Tensor v'4 dtype	maxval
-> Tensor Build dtype	output

statelessRandomUniformInt' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 dtype t tseed. (OneOf '[Int32, Int64] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed)
=> OpParams
-> Tensor v'1 t	shape
-> Tensor v'2 tseed	seed
-> Tensor v'3 dtype	minval
-> Tensor v'4 dtype	maxval
-> Tensor Build dtype	output

statelessTruncatedNormal Source #

Arguments

:: 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	shape
-> Tensor v'2 tseed	seed
-> Tensor Build dtype	output

statelessTruncatedNormal' Source #

Arguments

:: 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	shape
-> Tensor v'2 tseed	seed
-> Tensor Build dtype	output

staticRegexFullMatch Source #

Arguments

:: ByteString	pattern
-> Tensor v'1 ByteString	input
-> Tensor Build Bool	output

staticRegexFullMatch' Source #

Arguments

:: OpParams
-> ByteString	pattern
-> Tensor v'1 ByteString	input
-> Tensor Build Bool	output

staticRegexReplace Source #

Arguments

:: ByteString	pattern
-> ByteString	rewrite
-> Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

staticRegexReplace' Source #

Arguments

:: OpParams
-> ByteString	pattern
-> ByteString	rewrite
-> Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

statsAggregatorHandle Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	handle

statsAggregatorHandle' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	handle

statsAggregatorHandleV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	handle

statsAggregatorHandleV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	handle

statsAggregatorSetSummaryWriter Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	stats_aggregator
-> Tensor v'2 ResourceHandle	summary
-> m' ControlNode

statsAggregatorSetSummaryWriter' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	stats_aggregator
-> Tensor v'2 ResourceHandle	summary
-> m' ControlNode

statsAggregatorSummary Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	iterator
-> m' (Tensor Value ByteString)	summary

statsAggregatorSummary' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	iterator
-> m' (Tensor Value ByteString)	summary

stopGradient Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	output

stopGradient' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

stridedSlice Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t index. (TensorType t, OneOf '[Int32, Int64] index)
=> Tensor v'1 t	input
-> Tensor v'2 index	begin
-> Tensor v'3 index	end
-> Tensor v'4 index	strides
-> Tensor Build t	output

stridedSlice' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t index. (TensorType t, OneOf '[Int32, Int64] index)
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 index	begin
-> Tensor v'3 index	end
-> Tensor v'4 index	strides
-> Tensor Build t	output

stridedSliceAssign Source #

Arguments

:: forall v'2 v'3 v'4 v'5 t index m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] index)
=> Tensor Ref t	ref
-> Tensor v'2 index	begin
-> Tensor v'3 index	end
-> Tensor v'4 index	strides
-> Tensor v'5 t	value
-> m' (Tensor Ref t)	output_ref

stridedSliceAssign' Source #

Arguments

:: forall v'2 v'3 v'4 v'5 t index m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] index)
=> OpParams
-> Tensor Ref t	ref
-> Tensor v'2 index	begin
-> Tensor v'3 index	end
-> Tensor v'4 index	strides
-> Tensor v'5 t	value
-> m' (Tensor Ref t)	output_ref

stridedSliceGrad Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t index. (TensorType t, OneOf '[Int32, Int64] index)
=> Tensor v'1 index	shape
-> Tensor v'2 index	begin
-> Tensor v'3 index	end
-> Tensor v'4 index	strides
-> Tensor v'5 t	dy
-> Tensor Build t	output

stridedSliceGrad' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t index. (TensorType t, OneOf '[Int32, Int64] index)
=> OpParams
-> Tensor v'1 index	shape
-> Tensor v'2 index	begin
-> Tensor v'3 index	end
-> Tensor v'4 index	strides
-> Tensor v'5 t	dy
-> Tensor Build t	output

stringFormat Source #

Arguments

:: forall v'1 t. TensorTypes t
=> TensorList v'1 t	inputs
-> Tensor Build ByteString	output

stringFormat' Source #

Arguments

:: forall v'1 t. TensorTypes t
=> OpParams
-> TensorList v'1 t	inputs
-> Tensor Build ByteString	output

stringJoin Source #

Arguments

:: [Tensor v'1 ByteString]	inputs
-> Tensor Build ByteString	output

stringJoin' Source #

Arguments

:: OpParams
-> [Tensor v'1 ByteString]	inputs
-> Tensor Build ByteString	output

stringLength Source #

Arguments

:: Tensor v'1 ByteString	input
-> Tensor Build Int32	output

stringLength' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	input
-> Tensor Build Int32	output

stringLower Source #

Arguments

:: Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

stringLower' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

stringNGrams Source #

Arguments

:: forall v'1 v'2 tsplits. OneOf '[Int32, Int64] tsplits
=> ByteString	left_pad
-> Int64	pad_width
-> Bool	preserve_short_sequences
-> ByteString	right_pad
-> ByteString	separator
-> Tensor v'1 ByteString	data
-> Tensor v'2 tsplits	data_splits
-> (Tensor Build ByteString, Tensor Build tsplits)	(ngrams, ngrams_splits) ngrams ngrams_splits

stringNGrams' Source #

Arguments

:: forall v'1 v'2 tsplits. OneOf '[Int32, Int64] tsplits
=> OpParams
-> ByteString	left_pad
-> Int64	pad_width
-> Bool	preserve_short_sequences
-> ByteString	right_pad
-> ByteString	separator
-> Tensor v'1 ByteString	data
-> Tensor v'2 tsplits	data_splits
-> (Tensor Build ByteString, Tensor Build tsplits)	(ngrams, ngrams_splits) ngrams ngrams_splits

stringSplit Source #

Arguments

:: Tensor v'1 ByteString

input

-> Tensor v'2 ByteString

delimiter

-> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64)

(indices, values, shape)

indices
values
shape

stringSplit' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	input
-> Tensor v'2 ByteString	delimiter
-> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64)	(indices, values, shape) indices values shape

stringSplitV2 Source #

Arguments

:: Tensor v'1 ByteString

input

-> Tensor v'2 ByteString

sep

-> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64)

(indices, values, shape)

indices
values
shape

stringSplitV2' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	input
-> Tensor v'2 ByteString	sep
-> (Tensor Build Int64, Tensor Build ByteString, Tensor Build Int64)	(indices, values, shape) indices values shape

stringStrip Source #

Arguments

:: Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

stringStrip' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

stringToHashBucket Source #

Arguments

:: Int64	num_buckets
-> Tensor v'1 ByteString	string_tensor
-> Tensor Build Int64	output

stringToHashBucket' Source #

Arguments

:: OpParams
-> Int64	num_buckets
-> Tensor v'1 ByteString	string_tensor
-> Tensor Build Int64	output

stringToHashBucketFast Source #

Arguments

:: Int64	num_buckets
-> Tensor v'1 ByteString	input
-> Tensor Build Int64	output

stringToHashBucketFast' Source #

Arguments

:: OpParams
-> Int64	num_buckets
-> Tensor v'1 ByteString	input
-> Tensor Build Int64	output

stringToHashBucketStrong Source #

Arguments

:: Int64	num_buckets
-> Tensor v'1 ByteString	input
-> Tensor Build Int64	output

stringToHashBucketStrong' Source #

Arguments

:: OpParams
-> Int64	num_buckets
-> Tensor v'1 ByteString	input
-> Tensor Build Int64	output

stringToNumber Source #

Arguments

:: forall v'1 out_type. OneOf '[Int32, Int64, Double, Float] out_type
=> Tensor v'1 ByteString	string_tensor
-> Tensor Build out_type	output

stringToNumber' Source #

Arguments

:: forall v'1 out_type. OneOf '[Int32, Int64, Double, Float] out_type
=> OpParams
-> Tensor v'1 ByteString	string_tensor
-> Tensor Build out_type	output

stringUpper Source #

Arguments

:: Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

stringUpper' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

sub Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word8, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

sub' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

substr Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64] t
=> Tensor v'1 ByteString	input
-> Tensor v'2 t	pos
-> Tensor v'3 t	len
-> Tensor Build ByteString	output

substr' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64] t
=> OpParams
-> Tensor v'1 ByteString	input
-> Tensor v'2 t	pos
-> Tensor v'3 t	len
-> Tensor Build ByteString	output

sum Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

sum' Source #

Arguments

:: 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	input
-> Tensor v'2 tidx	reduction_indices
-> Tensor Build t	output

summaryWriter Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	writer

summaryWriter' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	writer

svd Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t

=> Tensor v'1 t

input

-> (Tensor Build t, Tensor Build t, Tensor Build t)

(s, u, v)

s
u
v

svd' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> (Tensor Build t, Tensor Build t, Tensor Build t)	(s, u, v) s u v

switch Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Tensor v'1 t	data
-> Tensor v'2 Bool	pred
-> (Tensor Build t, Tensor Build t)	(output_false, output_true) output_false output_true

switch' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Tensor v'1 t	data
-> Tensor v'2 Bool	pred
-> (Tensor Build t, Tensor Build t)	(output_false, output_true) output_false output_true

tFRecordDataset Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 ByteString	filenames
-> Tensor v'2 ByteString	compression_type
-> Tensor v'3 Int64	buffer_size
-> m' (Tensor Value Variant)	handle

tFRecordDataset' Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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

tFRecordReader Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Ref ByteString)	reader_handle

tFRecordReader' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Ref ByteString)	reader_handle

tFRecordReaderV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	reader_handle

tFRecordReaderV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	reader_handle

tPUCompilationResult Source #

Arguments

:: Tensor Build ByteString

output

tPUCompilationResult' Source #

Arguments

:: OpParams
-> Tensor Build ByteString	output

tPUEmbeddingActivations Source #

Arguments

:: Int64	lookup_id
-> Int64	table_id
-> Tensor v'1 Float	embedding_variable
-> Tensor v'2 Float	sliced_activations
-> Tensor Build Float	output

tPUEmbeddingActivations' Source #

Arguments

:: OpParams
-> Int64	lookup_id
-> Int64	table_id
-> Tensor v'1 Float	embedding_variable
-> Tensor v'2 Float	sliced_activations
-> Tensor Build Float	output

tPUOrdinalSelector Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value Int32)	device_ordinals

tPUOrdinalSelector' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value Int32)	device_ordinals

tPUReplicateMetadata Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	num_replicas
-> m' ControlNode

tPUReplicateMetadata' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	num_replicas
-> m' ControlNode

tPUReplicatedInput Source #

Arguments

:: forall v'1 t. TensorType t
=> [Tensor v'1 t]	inputs
-> Tensor Build t	output

tPUReplicatedInput' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> [Tensor v'1 t]	inputs
-> Tensor Build t	output

tPUReplicatedOutput Source #

Arguments

:: forall v'1 t. TensorType t
=> Int64	num_replicas
-> Tensor v'1 t	input
-> [Tensor Build t]	outputs

tPUReplicatedOutput' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Int64	num_replicas
-> Tensor v'1 t	input
-> [Tensor Build t]	outputs

takeDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	count
-> Tensor Build Variant	handle

takeDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	count
-> Tensor Build Variant	handle

takeManySparseFromTensorsMap Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)

=> Tensor v'1 Int64

sparse_handles

-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)

(sparse_indices, sparse_values, sparse_shape)

sparse_indices
sparse_values
sparse_shape

takeManySparseFromTensorsMap' Source #

Arguments

:: forall v'1 dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor v'1 Int64	sparse_handles
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)	(sparse_indices, sparse_values, sparse_shape) sparse_indices sparse_values sparse_shape

tan Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

tan' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

tanh Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

tanh' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

tanhGrad Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

tanhGrad' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	y
-> Tensor v'2 t	dy
-> Tensor Build t	z

temporaryVariable Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> Shape	shape
-> m' (Tensor Ref dtype)	ref

temporaryVariable' Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Shape	shape
-> m' (Tensor Ref dtype)	ref

tensorArray Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> DataType	dtype
-> Tensor v'1 Int32	size
-> m' (Tensor Ref ByteString)	handle

tensorArray' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> DataType	dtype
-> Tensor v'1 Int32	size
-> m' (Tensor Ref ByteString)	handle

tensorArrayClose Source #

Arguments

:: forall m'. MonadBuild m'
=> Tensor Ref ByteString	handle
-> m' ControlNode

tensorArrayClose' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	handle
-> m' ControlNode

tensorArrayCloseV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ByteString	handle
-> m' ControlNode

tensorArrayCloseV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	handle
-> m' ControlNode

tensorArrayCloseV3 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> m' ControlNode

tensorArrayCloseV3' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> m' ControlNode

tensorArrayConcat Source #

Arguments

:: forall v'2 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor Ref ByteString	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value dtype, Tensor Value Int64)	(value, lengths) value lengths

tensorArrayConcat' Source #

Arguments

:: forall v'2 dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor Ref ByteString	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value dtype, Tensor Value Int64)	(value, lengths) value lengths

tensorArrayConcatV2 Source #

Arguments

:: forall v'1 v'2 dtype. TensorType dtype
=> Tensor v'1 ByteString	handle
-> Tensor v'2 Float	flow_in
-> (Tensor Build dtype, Tensor Build Int64)	(value, lengths) value lengths

tensorArrayConcatV2' Source #

Arguments

:: forall v'1 v'2 dtype. TensorType dtype
=> OpParams
-> Tensor v'1 ByteString	handle
-> Tensor v'2 Float	flow_in
-> (Tensor Build dtype, Tensor Build Int64)	(value, lengths) value lengths

tensorArrayConcatV3 Source #

Arguments

:: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value dtype, Tensor Value Int64)	(value, lengths) value lengths

tensorArrayConcatV3' Source #

Arguments

:: forall v'1 v'2 dtype m'. (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) value lengths

tensorArrayGather Source #

Arguments

:: forall v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor Ref ByteString	handle
-> Tensor v'2 Int32	indices
-> Tensor v'3 Float	flow_in
-> m' (Tensor Value dtype)	value

tensorArrayGather' Source #

Arguments

:: forall v'2 v'3 dtype m'. (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

tensorArrayGatherV2 Source #

Arguments

:: forall v'1 v'2 v'3 dtype. TensorType dtype
=> Tensor v'1 ByteString	handle
-> Tensor v'2 Int32	indices
-> Tensor v'3 Float	flow_in
-> Tensor Build dtype	value

tensorArrayGatherV2' Source #

Arguments

:: forall v'1 v'2 v'3 dtype. TensorType dtype
=> OpParams
-> Tensor v'1 ByteString	handle
-> Tensor v'2 Int32	indices
-> Tensor v'3 Float	flow_in
-> Tensor Build dtype	value

tensorArrayGatherV3 Source #

Arguments

:: forall v'1 v'2 v'3 dtype m'. (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

tensorArrayGatherV3' Source #

Arguments

:: forall v'1 v'2 v'3 dtype m'. (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

tensorArrayGrad Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> ByteString	source
-> Tensor v'1 ByteString	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Ref ByteString)	grad_handle

tensorArrayGrad' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> ByteString	source
-> Tensor v'1 ByteString	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Ref ByteString)	grad_handle

tensorArrayGradV2 Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> ByteString	source
-> Tensor v'1 ByteString	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value ByteString)	grad_handle

tensorArrayGradV2' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> ByteString	source
-> Tensor v'1 ByteString	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value ByteString)	grad_handle

tensorArrayGradV3 Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> ByteString	source
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value ResourceHandle, Tensor Value Float)	(grad_handle, flow_out) grad_handle flow_out

tensorArrayGradV3' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> ByteString	source
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value ResourceHandle, Tensor Value Float)	(grad_handle, flow_out) grad_handle flow_out

tensorArrayGradWithShape Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> ByteString	source
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Float	flow_in
-> Tensor v'3 Int32	shape_to_prepend
-> m' (Tensor Value ResourceHandle, Tensor Value Float)	(grad_handle, flow_out) grad_handle flow_out

tensorArrayGradWithShape' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> ByteString	source
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Float	flow_in
-> Tensor v'3 Int32	shape_to_prepend
-> m' (Tensor Value ResourceHandle, Tensor Value Float)	(grad_handle, flow_out) grad_handle flow_out

tensorArrayPack Source #

Arguments

:: forall v'2 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor Ref ByteString	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value dtype)	value

tensorArrayPack' Source #

Arguments

:: forall v'2 dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Tensor Ref ByteString	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value dtype)	value

tensorArrayRead Source #

Arguments

:: forall v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype)
=> Tensor Ref ByteString	handle
-> Tensor v'2 Int32	index
-> Tensor v'3 Float	flow_in
-> m' (Tensor Value dtype)	value

tensorArrayRead' Source #

Arguments

:: forall v'2 v'3 dtype m'. (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

tensorArrayReadV2 Source #

Arguments

:: forall v'1 v'2 v'3 dtype. TensorType dtype
=> Tensor v'1 ByteString	handle
-> Tensor v'2 Int32	index
-> Tensor v'3 Float	flow_in
-> Tensor Build dtype	value

tensorArrayReadV2' Source #

Arguments

:: forall v'1 v'2 v'3 dtype. TensorType dtype
=> OpParams
-> Tensor v'1 ByteString	handle
-> Tensor v'2 Int32	index
-> Tensor v'3 Float	flow_in
-> Tensor Build dtype	value

tensorArrayReadV3 Source #

Arguments

:: forall v'1 v'2 v'3 dtype m'. (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

tensorArrayReadV3' Source #

Arguments

:: forall v'1 v'2 v'3 dtype m'. (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

tensorArrayScatter Source #

Arguments

:: forall v'2 v'3 v'4 t m'. (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

tensorArrayScatter' Source #

Arguments

:: forall v'2 v'3 v'4 t m'. (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

tensorArrayScatterV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> Tensor v'1 ByteString	handle
-> Tensor v'2 Int32	indices
-> Tensor v'3 t	value
-> Tensor v'4 Float	flow_in
-> Tensor Build Float	flow_out

tensorArrayScatterV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> OpParams
-> Tensor v'1 ByteString	handle
-> Tensor v'2 Int32	indices
-> Tensor v'3 t	value
-> Tensor v'4 Float	flow_in
-> Tensor Build Float	flow_out

tensorArrayScatterV3 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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 #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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

tensorArraySize Source #

Arguments

:: forall v'2 m'. MonadBuild m'
=> Tensor Ref ByteString	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value Int32)	size

tensorArraySize' Source #

Arguments

:: forall v'2 m'. MonadBuild m'
=> OpParams
-> Tensor Ref ByteString	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value Int32)	size

tensorArraySizeV2 Source #

Arguments

:: Tensor v'1 ByteString	handle
-> Tensor v'2 Float	flow_in
-> Tensor Build Int32	size

tensorArraySizeV2' Source #

Arguments

:: OpParams
-> Tensor v'1 ByteString	handle
-> Tensor v'2 Float	flow_in
-> Tensor Build Int32	size

tensorArraySizeV3 Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value Int32)	size

tensorArraySizeV3' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	handle
-> Tensor v'2 Float	flow_in
-> m' (Tensor Value Int32)	size

tensorArraySplit Source #

Arguments

:: forall v'2 v'3 v'4 t m'. (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

tensorArraySplit' Source #

Arguments

:: forall v'2 v'3 v'4 t m'. (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

tensorArraySplitV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> Tensor v'1 ByteString	handle
-> Tensor v'2 t	value
-> Tensor v'3 Int64	lengths
-> Tensor v'4 Float	flow_in
-> Tensor Build Float	flow_out

tensorArraySplitV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> OpParams
-> Tensor v'1 ByteString	handle
-> Tensor v'2 t	value
-> Tensor v'3 Int64	lengths
-> Tensor v'4 Float	flow_in
-> Tensor Build Float	flow_out

tensorArraySplitV3 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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

tensorArraySplitV3' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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

tensorArrayUnpack Source #

Arguments

:: forall v'2 v'3 t m'. (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

tensorArrayUnpack' Source #

Arguments

:: forall v'2 v'3 t m'. (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

tensorArrayV2 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> DataType	dtype
-> Tensor v'1 Int32	size
-> m' (Tensor Value ByteString)	handle

tensorArrayV2' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> DataType	dtype
-> Tensor v'1 Int32	size
-> m' (Tensor Value ByteString)	handle

tensorArrayV3 Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> DataType	dtype
-> Tensor v'1 Int32	size
-> m' (Tensor Value ResourceHandle, Tensor Value Float)	(handle, flow) handle flow

tensorArrayV3' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> DataType	dtype
-> Tensor v'1 Int32	size
-> m' (Tensor Value ResourceHandle, Tensor Value Float)	(handle, flow) handle flow

tensorArrayWrite Source #

Arguments

:: forall v'2 v'3 v'4 t m'. (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

tensorArrayWrite' Source #

Arguments

:: forall v'2 v'3 v'4 t m'. (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

tensorArrayWriteV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> Tensor v'1 ByteString	handle
-> Tensor v'2 Int32	index
-> Tensor v'3 t	value
-> Tensor v'4 Float	flow_in
-> Tensor Build Float	flow_out

tensorArrayWriteV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> OpParams
-> Tensor v'1 ByteString	handle
-> Tensor v'2 Int32	index
-> Tensor v'3 t	value
-> Tensor v'4 Float	flow_in
-> Tensor Build Float	flow_out

tensorArrayWriteV3 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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

tensorArrayWriteV3' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t m'. (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

tensorDataset Source #

Arguments

:: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types)
=> TensorList v'1 toutput_types	components
-> m' (Tensor Value Variant)	handle

tensorDataset' Source #

Arguments

:: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types)
=> OpParams
-> TensorList v'1 toutput_types	components
-> m' (Tensor Value Variant)	handle

tensorForestCreateTreeVariable Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	tree_handle
-> Tensor v'2 ByteString	tree_config
-> m' ControlNode

tensorForestCreateTreeVariable' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	tree_handle
-> Tensor v'2 ByteString	tree_config
-> m' ControlNode

tensorForestTreeDeserialize Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	tree_handle
-> Tensor v'2 ByteString	tree_config
-> m' ControlNode

tensorForestTreeDeserialize' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	tree_handle
-> Tensor v'2 ByteString	tree_config
-> m' ControlNode

tensorForestTreeIsInitializedOp Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	tree_handle
-> m' (Tensor Value Bool)	is_initialized

tensorForestTreeIsInitializedOp' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	tree_handle
-> m' (Tensor Value Bool)	is_initialized

tensorForestTreePredict Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Int64	logits_dimension
-> Tensor v'1 ResourceHandle	tree_handle
-> Tensor v'2 Float	dense_features
-> m' (Tensor Value Float)	logits

tensorForestTreePredict' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Int64	logits_dimension
-> Tensor v'1 ResourceHandle	tree_handle
-> Tensor v'2 Float	dense_features
-> m' (Tensor Value Float)	logits

tensorForestTreeResourceHandleOp Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	resource

tensorForestTreeResourceHandleOp' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	resource

tensorForestTreeSerialize Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	tree_handle
-> m' (Tensor Value ByteString)	tree_config

tensorForestTreeSerialize' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	tree_handle
-> m' (Tensor Value ByteString)	tree_config

tensorForestTreeSize Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	tree_handle
-> m' (Tensor Value Int32)	tree_size

tensorForestTreeSize' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	tree_handle
-> m' (Tensor Value Int32)	tree_size

tensorListConcat Source #

Arguments

:: forall v'1 element_dtype. TensorType element_dtype

=> Tensor v'1 Variant

input_handle

-> (Tensor Build element_dtype, Tensor Build Int64)

(tensor, lengths)

tensor
lengths

tensorListConcat' Source #

Arguments

:: forall v'1 element_dtype. TensorType element_dtype
=> OpParams
-> Tensor v'1 Variant	input_handle
-> (Tensor Build element_dtype, Tensor Build Int64)	(tensor, lengths) tensor lengths

tensorListConcatLists Source #

Arguments

:: DataType	element_dtype
-> Tensor v'1 Variant	input_a
-> Tensor v'2 Variant	input_b
-> Tensor Build Variant	output

tensorListConcatLists' Source #

Arguments

:: OpParams
-> DataType	element_dtype
-> Tensor v'1 Variant	input_a
-> Tensor v'2 Variant	input_b
-> Tensor Build Variant	output

tensorListConcatV2 Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type)
=> Tensor v'1 Variant	input_handle
-> Tensor v'2 shape_type	element_shape
-> Tensor v'3 Int64	leading_dims
-> (Tensor Build element_dtype, Tensor Build Int64)	(tensor, lengths) tensor lengths

tensorListConcatV2' Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type)
=> OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor v'2 shape_type	element_shape
-> Tensor v'3 Int64	leading_dims
-> (Tensor Build element_dtype, Tensor Build Int64)	(tensor, lengths) tensor lengths

tensorListElementShape Source #

Arguments

:: forall v'1 shape_type. OneOf '[Int32, Int64] shape_type
=> Tensor v'1 Variant	input_handle
-> Tensor Build shape_type	element_shape

tensorListElementShape' Source #

Arguments

:: forall v'1 shape_type. OneOf '[Int32, Int64] shape_type
=> OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor Build shape_type	element_shape

tensorListFromTensor Source #

Arguments

:: forall v'1 v'2 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type)
=> Tensor v'1 element_dtype	tensor
-> Tensor v'2 shape_type	element_shape
-> Tensor Build Variant	output_handle

tensorListFromTensor' Source #

Arguments

:: forall v'1 v'2 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type)
=> OpParams
-> Tensor v'1 element_dtype	tensor
-> Tensor v'2 shape_type	element_shape
-> Tensor Build Variant	output_handle

tensorListGather Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype
=> Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	indices
-> Tensor v'3 Int32	element_shape
-> Tensor Build element_dtype	values

tensorListGather' Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype
=> OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	indices
-> Tensor v'3 Int32	element_shape
-> Tensor Build element_dtype	values

tensorListGetItem Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype
=> Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	index
-> Tensor v'3 Int32	element_shape
-> Tensor Build element_dtype	item

tensorListGetItem' Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype
=> OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	index
-> Tensor v'3 Int32	element_shape
-> Tensor Build element_dtype	item

tensorListLength Source #

Arguments

:: Tensor v'1 Variant	input_handle
-> Tensor Build Int32	length

tensorListLength' Source #

Arguments

:: OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor Build Int32	length

tensorListPopBack Source #

Arguments

:: forall v'1 v'2 element_dtype. TensorType element_dtype
=> Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	element_shape
-> (Tensor Build Variant, Tensor Build element_dtype)	(output_handle, tensor) output_handle tensor

tensorListPopBack' Source #

Arguments

:: forall v'1 v'2 element_dtype. TensorType element_dtype
=> OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	element_shape
-> (Tensor Build Variant, Tensor Build element_dtype)	(output_handle, tensor) output_handle tensor

tensorListPushBack Source #

Arguments

:: forall v'1 v'2 element_dtype. TensorType element_dtype
=> Tensor v'1 Variant	input_handle
-> Tensor v'2 element_dtype	tensor
-> Tensor Build Variant	output_handle

tensorListPushBack' Source #

Arguments

:: forall v'1 v'2 element_dtype. TensorType element_dtype
=> OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor v'2 element_dtype	tensor
-> Tensor Build Variant	output_handle

tensorListPushBackBatch Source #

Arguments

:: forall v'1 v'2 element_dtype. TensorType element_dtype
=> Tensor v'1 Variant	input_handles
-> Tensor v'2 element_dtype	tensor
-> Tensor Build Variant	output_handles

tensorListPushBackBatch' Source #

Arguments

:: forall v'1 v'2 element_dtype. TensorType element_dtype
=> OpParams
-> Tensor v'1 Variant	input_handles
-> Tensor v'2 element_dtype	tensor
-> Tensor Build Variant	output_handles

tensorListReserve Source #

Arguments

:: forall v'1 v'2 shape_type. OneOf '[Int32, Int64] shape_type
=> DataType	element_dtype
-> Tensor v'1 shape_type	element_shape
-> Tensor v'2 Int32	num_elements
-> Tensor Build Variant	handle

tensorListReserve' Source #

Arguments

:: forall v'1 v'2 shape_type. OneOf '[Int32, Int64] shape_type
=> OpParams
-> DataType	element_dtype
-> Tensor v'1 shape_type	element_shape
-> Tensor v'2 Int32	num_elements
-> Tensor Build Variant	handle

tensorListResize Source #

Arguments

:: Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	size
-> Tensor Build Variant	output_handle

tensorListResize' Source #

Arguments

:: OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	size
-> Tensor Build Variant	output_handle

tensorListScatter Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type)
=> Tensor v'1 element_dtype	tensor
-> Tensor v'2 Int32	indices
-> Tensor v'3 shape_type	element_shape
-> Tensor Build Variant	output_handle

tensorListScatter' Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type)
=> OpParams
-> Tensor v'1 element_dtype	tensor
-> Tensor v'2 Int32	indices
-> Tensor v'3 shape_type	element_shape
-> Tensor Build Variant	output_handle

tensorListScatterIntoExistingList Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype
=> Tensor v'1 Variant	input_handle
-> Tensor v'2 element_dtype	tensor
-> Tensor v'3 Int32	indices
-> Tensor Build Variant	output_handle

tensorListScatterIntoExistingList' Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype
=> OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor v'2 element_dtype	tensor
-> Tensor v'3 Int32	indices
-> Tensor Build Variant	output_handle

tensorListScatterV2 Source #

Arguments

:: forall v'1 v'2 v'3 v'4 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type)
=> Tensor v'1 element_dtype	tensor
-> Tensor v'2 Int32	indices
-> Tensor v'3 shape_type	element_shape
-> Tensor v'4 Int32	num_elements
-> Tensor Build Variant	output_handle

tensorListScatterV2' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type)
=> OpParams
-> Tensor v'1 element_dtype	tensor
-> Tensor v'2 Int32	indices
-> Tensor v'3 shape_type	element_shape
-> Tensor v'4 Int32	num_elements
-> Tensor Build Variant	output_handle

tensorListSetItem Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype
=> Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	index
-> Tensor v'3 element_dtype	item
-> Tensor Build Variant	output_handle

tensorListSetItem' Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype
=> OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	index
-> Tensor v'3 element_dtype	item
-> Tensor Build Variant	output_handle

tensorListSplit Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type)
=> Tensor v'1 element_dtype	tensor
-> Tensor v'2 shape_type	element_shape
-> Tensor v'3 Int64	lengths
-> Tensor Build Variant	output_handle

tensorListSplit' Source #

Arguments

:: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type)
=> OpParams
-> Tensor v'1 element_dtype	tensor
-> Tensor v'2 shape_type	element_shape
-> Tensor v'3 Int64	lengths
-> Tensor Build Variant	output_handle

tensorListStack Source #

Arguments

:: forall v'1 v'2 element_dtype. TensorType element_dtype
=> Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	element_shape
-> Tensor Build element_dtype	tensor

tensorListStack' Source #

Arguments

:: forall v'1 v'2 element_dtype. TensorType element_dtype
=> OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor v'2 Int32	element_shape
-> Tensor Build element_dtype	tensor

tensorScatterAdd Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 t	tensor
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

tensorScatterAdd' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 t	tensor
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

tensorScatterMax Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 t	tensor
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

tensorScatterMax' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 t	tensor
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

tensorScatterMin Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 t	tensor
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

tensorScatterMin' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 t	tensor
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

tensorScatterSub Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 t	tensor
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

tensorScatterSub' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 t	tensor
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

tensorScatterUpdate Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 t	tensor
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

tensorScatterUpdate' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 t	tensor
-> Tensor v'2 tindices	indices
-> Tensor v'3 t	updates
-> Tensor Build t	output

tensorSliceDataset Source #

Arguments

:: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types)
=> TensorList v'1 toutput_types	components
-> m' (Tensor Value Variant)	handle

tensorSliceDataset' Source #

Arguments

:: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types)
=> OpParams
-> TensorList v'1 toutput_types	components
-> m' (Tensor Value Variant)	handle

tensorStridedSliceUpdate Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t index. (TensorType t, OneOf '[Int32, Int64] index)
=> Tensor v'1 t	input
-> Tensor v'2 index	begin
-> Tensor v'3 index	end
-> Tensor v'4 index	strides
-> Tensor v'5 t	value
-> Tensor Build t	output

tensorStridedSliceUpdate' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t index. (TensorType t, OneOf '[Int32, Int64] index)
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 index	begin
-> Tensor v'3 index	end
-> Tensor v'4 index	strides
-> Tensor v'5 t	value
-> Tensor Build t	output

tensorSummary Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	tensor
-> Tensor Build ByteString	summary

tensorSummary' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	tensor
-> Tensor Build ByteString	summary

tensorSummaryV2 Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Tensor v'1 ByteString	tag
-> Tensor v'2 t	tensor
-> Tensor v'3 ByteString	serialized_summary_metadata
-> Tensor Build ByteString	summary

tensorSummaryV2' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Tensor v'1 ByteString	tag
-> Tensor v'2 t	tensor
-> Tensor v'3 ByteString	serialized_summary_metadata
-> Tensor Build ByteString	summary

textLineDataset Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 ByteString	filenames
-> Tensor v'2 ByteString	compression_type
-> Tensor v'3 Int64	buffer_size
-> m' (Tensor Value Variant)	handle

textLineDataset' Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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

textLineReader Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Ref ByteString)	reader_handle

textLineReader' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Ref ByteString)	reader_handle

textLineReaderV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	reader_handle

textLineReaderV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	reader_handle

threadPoolDataset Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ResourceHandle	thread_pool
-> m' (Tensor Value Variant)	handle

threadPoolDataset' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 ResourceHandle	thread_pool
-> m' (Tensor Value Variant)	handle

threadPoolHandle Source #

Arguments

:: forall m'. MonadBuild m'
=> ByteString	display_name
-> Int64	num_threads
-> m' (Tensor Value ResourceHandle)	handle

threadPoolHandle' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> ByteString	display_name
-> Int64	num_threads
-> m' (Tensor Value ResourceHandle)	handle

threadUnsafeUnigramCandidateSampler Source #

Arguments

:: forall v'1 m'. 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) sampled_candidates true_expected_count sampled_expected_count

threadUnsafeUnigramCandidateSampler' Source #

Arguments

:: forall v'1 m'. 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) sampled_candidates true_expected_count sampled_expected_count

tile Source #

Arguments

:: forall v'1 v'2 t tmultiples. (TensorType t, OneOf '[Int32, Int64] tmultiples)
=> Tensor v'1 t	input
-> Tensor v'2 tmultiples	multiples
-> Tensor Build t	output

tile' Source #

Arguments

:: forall v'1 v'2 t tmultiples. (TensorType t, OneOf '[Int32, Int64] tmultiples)
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 tmultiples	multiples
-> Tensor Build t	output

tileGrad Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Tensor v'1 t	input
-> Tensor v'2 Int32	multiples
-> Tensor Build t	output

tileGrad' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 Int32	multiples
-> Tensor Build t	output

timestamp Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value Double)	ts

timestamp' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value Double)	ts

toBool Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build Bool	output

toBool' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build Bool	output

topK Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Int64	k
-> Tensor v'1 t	input
-> (Tensor Build t, Tensor Build Int32)	(values, indices) values indices

topK' Source #

Arguments

:: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Int64	k
-> Tensor v'1 t	input
-> (Tensor Build t, Tensor Build Int32)	(values, indices) values indices

topKV2 Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Tensor v'1 t	input
-> Tensor v'2 Int32	k
-> (Tensor Build t, Tensor Build Int32)	(values, indices) values indices

topKV2' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Tensor v'1 t	input
-> Tensor v'2 Int32	k
-> (Tensor Build t, Tensor Build Int32)	(values, indices) values indices

transpose Source #

Arguments

:: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm)
=> Tensor v'1 t	x
-> Tensor v'2 tperm	perm
-> Tensor Build t	y

transpose' Source #

Arguments

:: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm)
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 tperm	perm
-> Tensor Build t	y

tridiagonalMatMul Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> Tensor v'1 t	superdiag
-> Tensor v'2 t	maindiag
-> Tensor v'3 t	subdiag
-> Tensor v'4 t	rhs
-> Tensor Build t	output

tridiagonalMatMul' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> OpParams
-> Tensor v'1 t	superdiag
-> Tensor v'2 t	maindiag
-> Tensor v'3 t	subdiag
-> Tensor v'4 t	rhs
-> Tensor Build t	output

tridiagonalSolve Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> Tensor v'1 t	diagonals
-> Tensor v'2 t	rhs
-> Tensor Build t	output

tridiagonalSolve' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t
=> OpParams
-> Tensor v'1 t	diagonals
-> Tensor v'2 t	rhs
-> Tensor Build t	output

truncateDiv Source #

Arguments

:: forall v'1 v'2 t. 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 Build t	z

truncateDiv' Source #

Arguments

:: 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	x
-> Tensor v'2 t	y
-> Tensor Build t	z

truncateMod Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

truncateMod' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

truncatedNormal Source #

Arguments

:: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t)
=> Tensor v'1 t	shape
-> m' (Tensor Value dtype)	output

truncatedNormal' Source #

Arguments

:: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t)
=> OpParams
-> Tensor v'1 t	shape
-> m' (Tensor Value dtype)	output

tryRpc Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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) response status_code status_message

tryRpc' Source #

Arguments

:: forall v'1 v'2 v'3 m'. 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) response status_code status_message

unbatch Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> Int64	timeout_micros
-> Tensor v'1 t	batched_tensor
-> Tensor v'2 Int64	batch_index
-> Tensor v'3 Int64	id
-> Tensor Build t	unbatched_tensor

unbatch' Source #

Arguments

:: forall v'1 v'2 v'3 t. TensorType t
=> OpParams
-> Int64	timeout_micros
-> Tensor v'1 t	batched_tensor
-> Tensor v'2 Int64	batch_index
-> Tensor v'3 Int64	id
-> Tensor Build t	unbatched_tensor

unbatchDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

unbatchDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

unbatchGrad Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> Tensor v'1 t	original_input
-> Tensor v'2 Int64	batch_index
-> Tensor v'3 t	grad
-> Tensor v'4 Int64	id
-> Tensor Build t	batched_grad

unbatchGrad' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. TensorType t
=> OpParams
-> Tensor v'1 t	original_input
-> Tensor v'2 Int64	batch_index
-> Tensor v'3 t	grad
-> Tensor v'4 Int64	id
-> Tensor Build t	batched_grad

uncompressElement Source #

Arguments

:: forall v'1 output_types. TensorTypes output_types
=> Tensor v'1 Variant	compressed
-> TensorList Build output_types	components

uncompressElement' Source #

Arguments

:: forall v'1 output_types. TensorTypes output_types
=> OpParams
-> Tensor v'1 Variant	compressed
-> TensorList Build output_types	components

unicodeDecode Source #

Arguments

:: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits
=> ByteString	input_encoding
-> Tensor v'1 ByteString	input
-> (Tensor Build tsplits, Tensor Build Int32)	(row_splits, char_values) row_splits char_values

unicodeDecode' Source #

Arguments

:: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits
=> OpParams
-> ByteString	input_encoding
-> Tensor v'1 ByteString	input
-> (Tensor Build tsplits, Tensor Build Int32)	(row_splits, char_values) row_splits char_values

unicodeDecodeWithOffsets Source #

Arguments

:: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits
=> ByteString	input_encoding
-> Tensor v'1 ByteString	input
-> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)	(row_splits, char_values, char_to_byte_starts) row_splits char_values char_to_byte_starts

unicodeDecodeWithOffsets' Source #

Arguments

:: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits
=> OpParams
-> ByteString	input_encoding
-> Tensor v'1 ByteString	input
-> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)	(row_splits, char_values, char_to_byte_starts) row_splits char_values char_to_byte_starts

unicodeEncode Source #

Arguments

:: forall v'1 v'2 tsplits. OneOf '[Int32, Int64] tsplits
=> ByteString	output_encoding
-> Tensor v'1 Int32	input_values
-> Tensor v'2 tsplits	input_splits
-> Tensor Build ByteString	output

unicodeEncode' Source #

Arguments

:: forall v'1 v'2 tsplits. OneOf '[Int32, Int64] tsplits
=> OpParams
-> ByteString	output_encoding
-> Tensor v'1 Int32	input_values
-> Tensor v'2 tsplits	input_splits
-> Tensor Build ByteString	output

unicodeScript Source #

Arguments

:: Tensor v'1 Int32	input
-> Tensor Build Int32	output

unicodeScript' Source #

Arguments

:: OpParams
-> Tensor v'1 Int32	input
-> Tensor Build Int32	output

unicodeTranscode Source #

Arguments

:: ByteString	input_encoding
-> ByteString	output_encoding
-> Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

unicodeTranscode' Source #

Arguments

:: OpParams
-> ByteString	input_encoding
-> ByteString	output_encoding
-> Tensor v'1 ByteString	input
-> Tensor Build ByteString	output

uniformCandidateSampler Source #

Arguments

:: forall v'1 m'. 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) sampled_candidates true_expected_count sampled_expected_count

uniformCandidateSampler' Source #

Arguments

:: forall v'1 m'. 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) sampled_candidates true_expected_count sampled_expected_count

unique Source #

Arguments

:: forall v'1 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx)

=> Tensor v'1 t

x

-> (Tensor Build t, Tensor Build out_idx)

(y, idx)

y
idx

unique' Source #

Arguments

:: forall v'1 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx)
=> OpParams
-> Tensor v'1 t	x
-> (Tensor Build t, Tensor Build out_idx)	(y, idx) y idx

uniqueDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

uniqueDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor Build Variant	handle

uniqueV2 Source #

Arguments

:: forall v'1 v'2 t taxis out_idx. (TensorType t, OneOf '[Int32, Int64] taxis, OneOf '[Int32, Int64] out_idx)
=> Tensor v'1 t	x
-> Tensor v'2 taxis	axis
-> (Tensor Build t, Tensor Build out_idx)	(y, idx) y idx

uniqueV2' Source #

Arguments

:: 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	x
-> Tensor v'2 taxis	axis
-> (Tensor Build t, Tensor Build out_idx)	(y, idx) y idx

uniqueWithCounts Source #

Arguments

:: forall v'1 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx)

=> Tensor v'1 t

x

-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)

(y, idx, count)

y
idx
count

uniqueWithCounts' Source #

Arguments

:: forall v'1 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx)
=> OpParams
-> Tensor v'1 t	x
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)	(y, idx, count) y idx count

uniqueWithCountsV2 Source #

Arguments

:: forall v'1 v'2 t taxis out_idx. (TensorType t, OneOf '[Int32, Int64] taxis, OneOf '[Int32, Int64] out_idx)
=> Tensor v'1 t	x
-> Tensor v'2 taxis	axis
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)	(y, idx, count) y idx count

uniqueWithCountsV2' Source #

Arguments

:: 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	x
-> Tensor v'2 taxis	axis
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)	(y, idx, count) y idx count

unpack Source #

Arguments

:: forall v'1 t. TensorType t
=> Int64	num
-> Tensor v'1 t	value
-> [Tensor Build t]	output

unpack' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Int64	num
-> Tensor v'1 t	value
-> [Tensor Build t]	output

unravelIndex Source #

Arguments

:: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx
=> Tensor v'1 tidx	indices
-> Tensor v'2 tidx	dims
-> Tensor Build tidx	output

unravelIndex' Source #

Arguments

:: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx
=> OpParams
-> Tensor v'1 tidx	indices
-> Tensor v'2 tidx	dims
-> Tensor Build tidx	output

unsortedSegmentJoin Source #

Arguments

:: forall v'1 v'2 v'3 tindices tnumsegments. (OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments)
=> Tensor v'1 ByteString	inputs
-> Tensor v'2 tindices	segment_ids
-> Tensor v'3 tnumsegments	num_segments
-> Tensor Build ByteString	output

unsortedSegmentJoin' Source #

Arguments

:: forall v'1 v'2 v'3 tindices tnumsegments. (OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments)
=> OpParams
-> Tensor v'1 ByteString	inputs
-> Tensor v'2 tindices	segment_ids
-> Tensor v'3 tnumsegments	num_segments
-> Tensor Build ByteString	output

unsortedSegmentMax Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor v'3 tnumsegments	num_segments
-> Tensor Build t	output

unsortedSegmentMax' Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor v'3 tnumsegments	num_segments
-> Tensor Build t	output

unsortedSegmentMin Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor v'3 tnumsegments	num_segments
-> Tensor Build t	output

unsortedSegmentMin' Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor v'3 tnumsegments	num_segments
-> Tensor Build t	output

unsortedSegmentProd Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor v'3 tnumsegments	num_segments
-> Tensor Build t	output

unsortedSegmentProd' Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor v'3 tnumsegments	num_segments
-> Tensor Build t	output

unsortedSegmentSum Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor v'3 tnumsegments	num_segments
-> Tensor Build t	output

unsortedSegmentSum' Source #

Arguments

:: 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	data
-> Tensor v'2 tindices	segment_ids
-> Tensor v'3 tnumsegments	num_segments
-> Tensor Build t	output

unstage Source #

Arguments

:: forall dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> m' (TensorList Value dtypes)	values

unstage' Source #

Arguments

:: forall dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> m' (TensorList Value dtypes)	values

unwrapDatasetVariant Source #

Arguments

:: Tensor v'1 Variant	input_handle
-> Tensor Build Variant	output_handle

unwrapDatasetVariant' Source #

Arguments

:: OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor Build Variant	output_handle

upperBound Source #

Arguments

:: forall v'1 v'2 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type)
=> Tensor v'1 t	sorted_inputs
-> Tensor v'2 t	values
-> Tensor Build out_type	output

upperBound' Source #

Arguments

:: forall v'1 v'2 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type)
=> OpParams
-> Tensor v'1 t	sorted_inputs
-> Tensor v'2 t	values
-> Tensor Build out_type	output

varHandleOp Source #

Arguments

:: forall m'. MonadBuild m'
=> DataType	dtype
-> Shape	shape
-> m' (Tensor Value ResourceHandle)	resource

varHandleOp' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> DataType	dtype
-> Shape	shape
-> m' (Tensor Value ResourceHandle)	resource

varIsInitializedOp Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Value Bool)	is_initialized

varIsInitializedOp' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	resource
-> m' (Tensor Value Bool)	is_initialized

variable Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> Shape	shape
-> m' (Tensor Ref dtype)	ref

variable' Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Shape	shape
-> m' (Tensor Ref dtype)	ref

variableShape Source #

Arguments

:: forall v'1 out_type m'. (MonadBuild m', OneOf '[Int32, Int64] out_type)
=> Tensor v'1 ResourceHandle	input
-> m' (Tensor Value out_type)	output

variableShape' Source #

Arguments

:: forall v'1 out_type m'. (MonadBuild m', OneOf '[Int32, Int64] out_type)
=> OpParams
-> Tensor v'1 ResourceHandle	input
-> m' (Tensor Value out_type)	output

variableV2 Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> Shape	shape
-> m' (Tensor Ref dtype)	ref

variableV2' Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Shape	shape
-> m' (Tensor Ref dtype)	ref

where' Source #

Arguments

:: 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	input
-> Tensor Build Int64	index

where'' Source #

Arguments

:: 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	input
-> Tensor Build Int64	index

wholeFileReader Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Ref ByteString)	reader_handle

wholeFileReader' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Ref ByteString)	reader_handle

wholeFileReaderV2 Source #

Arguments

:: forall m'. MonadBuild m'
=> m' (Tensor Value ResourceHandle)	reader_handle

wholeFileReaderV2' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value ResourceHandle)	reader_handle

windowDataset Source #

Arguments

:: [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	size
-> Tensor v'3 Int64	shift
-> Tensor v'4 Int64	stride
-> Tensor v'5 Bool	drop_remainder
-> Tensor Build Variant	handle

windowDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> Tensor v'1 Variant	input_dataset
-> Tensor v'2 Int64	size
-> Tensor v'3 Int64	shift
-> Tensor v'4 Int64	stride
-> Tensor v'5 Bool	drop_remainder
-> Tensor Build Variant	handle

workerHeartbeat Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Tensor v'1 ByteString	request
-> m' (Tensor Value ByteString)	response

workerHeartbeat' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	request
-> m' (Tensor Value ByteString)	response

wrapDatasetVariant Source #

Arguments

:: Tensor v'1 Variant	input_handle
-> Tensor Build Variant	output_handle

wrapDatasetVariant' Source #

Arguments

:: OpParams
-> Tensor v'1 Variant	input_handle
-> Tensor Build Variant	output_handle

writeAudioSummary Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. 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

writeAudioSummary' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 m'. 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

writeFile Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> Tensor v'1 ByteString	filename
-> Tensor v'2 ByteString	contents
-> m' ControlNode

writeFile' Source #

Arguments

:: forall v'1 v'2 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ByteString	filename
-> Tensor v'2 ByteString	contents
-> m' ControlNode

writeGraphSummary Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	writer
-> Tensor v'2 Int64	step
-> Tensor v'3 ByteString	tensor
-> m' ControlNode

writeGraphSummary' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	writer
-> Tensor v'2 Int64	step
-> Tensor v'3 ByteString	tensor
-> m' ControlNode

writeHistogramSummary Source #

Arguments

:: 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	writer
-> Tensor v'2 Int64	step
-> Tensor v'3 ByteString	tag
-> Tensor v'4 t	values
-> m' ControlNode

writeHistogramSummary' Source #

Arguments

:: 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	writer
-> Tensor v'2 Int64	step
-> Tensor v'3 ByteString	tag
-> Tensor v'4 t	values
-> m' ControlNode

writeImageSummary Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t m'. (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

writeImageSummary' Source #

Arguments

:: 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	writer
-> Tensor v'2 Int64	step
-> Tensor v'3 ByteString	tag
-> Tensor v'4 t	tensor
-> Tensor v'5 Word8	bad_color
-> m' ControlNode

writeRawProtoSummary Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> Tensor v'1 ResourceHandle	writer
-> Tensor v'2 Int64	step
-> Tensor v'3 ByteString	tensor
-> m' ControlNode

writeRawProtoSummary' Source #

Arguments

:: forall v'1 v'2 v'3 m'. MonadBuild m'
=> OpParams
-> Tensor v'1 ResourceHandle	writer
-> Tensor v'2 Int64	step
-> Tensor v'3 ByteString	tensor
-> m' ControlNode

writeScalarSummary Source #

Arguments

:: 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	writer
-> Tensor v'2 Int64	step
-> Tensor v'3 ByteString	tag
-> Tensor v'4 t	value
-> m' ControlNode

writeScalarSummary' Source #

Arguments

:: 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	writer
-> Tensor v'2 Int64	step
-> Tensor v'3 ByteString	tag
-> Tensor v'4 t	value
-> m' ControlNode

writeSummary Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t m'. (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

writeSummary' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t m'. (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

xdivy Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

xdivy' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

xlaBroadcastHelper Source #

Arguments

:: 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	lhs: the LHS input tensor
-> Tensor v'2 t	rhs: the RHS input tensor
-> Tensor v'3 tindices	broadcast_dims: an XLA-style broadcast dimension specification
-> (Tensor Build t, Tensor Build t)	(lhs_output, rhs_output) lhs_output: the broadcasted LHS tensor rhs_output: the broadcasted RHS tensor

Helper operator for performing XLA-style broadcasts

Broadcasts lhs and rhs to the same rank, by adding size 1 dimensions to whichever of lhs and rhs has the lower rank, using XLA's broadcasting rules for binary operators.

xlaBroadcastHelper' Source #

Arguments

:: 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	lhs: the LHS input tensor
-> Tensor v'2 t	rhs: the RHS input tensor
-> Tensor v'3 tindices	broadcast_dims: an XLA-style broadcast dimension specification
-> (Tensor Build t, Tensor Build t)	(lhs_output, rhs_output) lhs_output: the broadcasted LHS tensor rhs_output: the broadcasted RHS tensor

xlaClusterOutput Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	outputs

Operator that connects the output of an XLA computation to other consumer graph nodes.

xlaClusterOutput' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	outputs

xlaConv Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices)
=> ByteString	dimension_numbers: a serialized xla::ConvolutionDimensionNumbers proto.
-> ByteString	precision_config: a serialized xla::PrecisionConfig proto.
-> Tensor v'1 t	lhs: the input tensor
-> Tensor v'2 t	rhs: the kernel tensor
-> Tensor v'3 tindices	window_strides: the inter-window strides
-> Tensor v'4 tindices	padding: the padding to apply at the start and end of each input dimensions
-> Tensor v'5 tindices	lhs_dilation: dilation to apply between input elements
-> Tensor v'6 tindices	rhs_dilation: dilation to apply between kernel elements
-> Tensor v'7 tindices	feature_group_count: number of feature groups for grouped convolution.
-> Tensor Build t	output

Wraps the XLA ConvGeneralDilated operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#conv_convolution .

xlaConv' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> ByteString	dimension_numbers: a serialized xla::ConvolutionDimensionNumbers proto.
-> ByteString	precision_config: a serialized xla::PrecisionConfig proto.
-> Tensor v'1 t	lhs: the input tensor
-> Tensor v'2 t	rhs: the kernel tensor
-> Tensor v'3 tindices	window_strides: the inter-window strides
-> Tensor v'4 tindices	padding: the padding to apply at the start and end of each input dimensions
-> Tensor v'5 tindices	lhs_dilation: dilation to apply between input elements
-> Tensor v'6 tindices	rhs_dilation: dilation to apply between kernel elements
-> Tensor v'7 tindices	feature_group_count: number of feature groups for grouped convolution.
-> Tensor Build t	output

xlaDequantize Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> Float	max_range: The maximum scalar value possibly produced for the input.
-> Float	min_range: The minimum scalar value possibly produced for the input.
-> ByteString	mode: String to determine the dequantize mode in {MIN_COMBINED, MIN_FIRST, SCALED}.
-> Bool	transpose_output: Boolean to determine if output is transposed. transpose_output is faster when input is large and rank of input is higher than 1.
-> Tensor v'1 Word32	input: Input tensors whose types is uint32, shape is [d0, ..., dn].
-> m' (Tensor Value Word16)	output: Output tensors whose types is bloat16. If transpose_output is true, output shape is [dn * 4, dn-1, ..., d1, d0]. If transpose_output is false, output shape is [d0,..., dn * 4].

Takes the packed uint32 input and unpacks the input to uint8 to do

Dequantization on device.

xlaDequantize' Source #

Arguments

:: forall v'1 m'. MonadBuild m'
=> OpParams
-> Float	max_range: The maximum scalar value possibly produced for the input.
-> Float	min_range: The minimum scalar value possibly produced for the input.
-> ByteString	mode: String to determine the dequantize mode in {MIN_COMBINED, MIN_FIRST, SCALED}.
-> Bool	transpose_output: Boolean to determine if output is transposed. transpose_output is faster when input is large and rank of input is higher than 1.
-> Tensor v'1 Word32	input: Input tensors whose types is uint32, shape is [d0, ..., dn].
-> m' (Tensor Value Word16)	output: Output tensors whose types is bloat16. If transpose_output is true, output shape is [dn * 4, dn-1, ..., d1, d0]. If transpose_output is false, output shape is [d0,..., dn * 4].

xlaDot Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> ByteString	dimension_numbers: a serialized xla::DotDimensionNumbers proto.
-> ByteString	precision_config: a serialized xla::PrecisionConfig proto.
-> Tensor v'1 t	lhs: the LHS tensor
-> Tensor v'2 t	rhs: the RHS tensor
-> Tensor Build t	output

Wraps the XLA DotGeneral operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#dotgeneral .

xlaDot' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> ByteString	dimension_numbers: a serialized xla::DotDimensionNumbers proto.
-> ByteString	precision_config: a serialized xla::PrecisionConfig proto.
-> Tensor v'1 t	lhs: the LHS tensor
-> Tensor v'2 t	rhs: the RHS tensor
-> Tensor Build t	output

xlaDynamicSlice Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 t	input: A `Tensor` of type T.
-> Tensor v'2 tindices	start_indices: List of N integers containing the slice size for each dimension. Each value must be strictly greater than zero, and start + size must be less than or equal to the size of the dimension to avoid implementation defined behavior.
-> Tensor v'3 tindices	size_indices
-> Tensor Build t	output

Wraps the XLA DynamicSlice operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#dynamicslice .

DynamicSlice extracts a sub-array from the input array at dynamic start_indices. The size of the slice in each dimension is passed in size_indices, which specify the end point of exclusive slice intervals in each dimension -- [start, start + size). The shape of start_indices must have rank 1, with dimension size equal to the rank of operand.

xlaDynamicSlice' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 t	input: A `Tensor` of type T.
-> Tensor v'2 tindices	start_indices: List of N integers containing the slice size for each dimension. Each value must be strictly greater than zero, and start + size must be less than or equal to the size of the dimension to avoid implementation defined behavior.
-> Tensor v'3 tindices	size_indices
-> Tensor Build t	output

xlaDynamicUpdateSlice Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 t	input: A `Tensor` of type T.
-> Tensor v'2 t	update: A `Tensor` of type T. Same rank as `input`.
-> Tensor v'3 tindices	indices: A vector of indices into `input`. Must have length equal to the rank of `input`.
-> Tensor Build t	output: A `Tensor` of type T.

Wraps the XLA DynamicUpdateSlice operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#dynamicupdateslice .

XlaDynamicUpdateSlice generates a result which is the value of the input operand, with a slice update overwritten at indices. The shape of update determines the shape of the sub-array of the result which is updated. The shape of indices must be rank == 1, with dimension size equal to the rank of input.

Handling of out-of-bounds slice indices is implementation-defined.

xlaDynamicUpdateSlice' Source #

Arguments

:: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 t	input: A `Tensor` of type T.
-> Tensor v'2 t	update: A `Tensor` of type T. Same rank as `input`.
-> Tensor v'3 tindices	indices: A vector of indices into `input`. Must have length equal to the rank of `input`.
-> Tensor Build t	output: A `Tensor` of type T.

xlaEinsum Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Float, Word16, Float] t
=> ByteString	equation
-> Tensor v'1 t	a
-> Tensor v'2 t	b
-> Tensor Build t	product

An op which supports basic einsum op with 2 inputs and 1 output.

This op has better TPU performance since it doesn't have explicitly reshape and transpose operations as tf.einsum does.

xlaEinsum' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Float, Word16, Float] t
=> OpParams
-> ByteString	equation
-> Tensor v'1 t	a
-> Tensor v'2 t	b
-> Tensor Build t	product

xlaGather Source #

Arguments

:: 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)
=> ByteString	dimension_numbers: A serialized xla::GatherDimensionNumbers proto.
-> Bool	indices_are_sorted: Boolean indicating if the indices are sorted.
-> Tensor v'1 t	operand: The array we're gathering from.
-> Tensor v'2 tindices	start_indices: Array containing the starting indices of the slices we gather.
-> Tensor v'3 tindices	slice_sizes: slice_sizes[i] is the bounds for the slice on dimension i.
-> Tensor Build t	output

Wraps the XLA Gather operator documented at

https://www.tensorflow.org/xla/operation_semantics#gather

xlaGather' Source #

Arguments

:: 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
-> ByteString	dimension_numbers: A serialized xla::GatherDimensionNumbers proto.
-> Bool	indices_are_sorted: Boolean indicating if the indices are sorted.
-> Tensor v'1 t	operand: The array we're gathering from.
-> Tensor v'2 tindices	start_indices: Array containing the starting indices of the slices we gather.
-> Tensor v'3 tindices	slice_sizes: slice_sizes[i] is the bounds for the slice on dimension i.
-> Tensor Build t	output

xlaKeyValueSort Source #

Arguments

:: forall v'1 v'2 k v. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] k, TensorType v)
=> Tensor v'1 k	keys: A `Tensor` of type K.
-> Tensor v'2 v	values: A `Tensor` of type V.
-> (Tensor Build k, Tensor Build v)	(sorted_keys, sorted_values) sorted_keys: A `Tensor` of type K. sorted_values: A `Tensor` of type V.

Wraps the XLA Sort operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#sort .

Sorts a tensor. Currently only sorts in ascending order are supported.

xlaKeyValueSort' Source #

Arguments

:: forall v'1 v'2 k v. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] k, TensorType v)
=> OpParams
-> Tensor v'1 k	keys: A `Tensor` of type K.
-> Tensor v'2 v	values: A `Tensor` of type V.
-> (Tensor Build k, Tensor Build v)	(sorted_keys, sorted_values) sorted_keys: A `Tensor` of type K. sorted_values: A `Tensor` of type V.

xlaPad Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> Tensor v'1 t	input: A `Tensor` of type T.
-> Tensor v'2 t	padding_value: A scalar `Tensor` of type T.
-> Tensor v'3 tindices	padding_low: the padding to apply at the start of each input dimensions
-> Tensor v'4 tindices	padding_high: the padding to apply at the end of each input dimension.
-> Tensor v'5 tindices	padding_interior: the padding to apply between each input element.
-> Tensor Build t	output: A `Tensor` of type T.

Wraps the XLA Pad operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#pad .

xlaPad' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices)
=> OpParams
-> Tensor v'1 t	input: A `Tensor` of type T.
-> Tensor v'2 t	padding_value: A scalar `Tensor` of type T.
-> Tensor v'3 tindices	padding_low: the padding to apply at the start of each input dimensions
-> Tensor v'4 tindices	padding_high: the padding to apply at the end of each input dimension.
-> Tensor v'5 tindices	padding_interior: the padding to apply between each input element.
-> Tensor Build t	output: A `Tensor` of type T.

xlaRecv Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> Shape	shape: The shape of the tensor.
-> ByteString	tensor_name: A string key that identifies the channel.
-> m' (Tensor Value dtype)	tensor: The tensor to receive.

Receives the named tensor from another XLA computation. Wraps the XLA Recv

operator documented at https://www.tensorflow.org/performance/xla/operation_semantics#recv .

xlaRecv' Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Shape	shape: The shape of the tensor.
-> ByteString	tensor_name: A string key that identifies the channel.
-> m' (Tensor Value dtype)	tensor: The tensor to receive.

xlaReplicaId Source #

Arguments

:: Tensor Build Int32

id

Replica ID.

xlaReplicaId' Source #

Arguments

:: OpParams
-> Tensor Build Int32	id

xlaSelfAdjointEig Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Float	epsilon: the tolerance ratio.
-> Bool	lower: a boolean specifies whether the calculation is done with the lower triangular part or the upper triangular part.
-> Int64	max_iter: maximum number of sweep update, i.e., the whole lower triangular part or upper triangular part based on parameter lower. Heuristically, it has been argued that approximately logN sweeps are needed in practice (Ref: Golub & van Loan "Matrix Computation").
-> Tensor v'1 t	a: the input tensor.
-> (Tensor Build t, Tensor Build t)	(w, v) w: The eigenvalues in ascending order, each repeated according to its multiplicity. v: The column v[..., :, i] is the normalized eigenvector corresponding to the eigenvalue w[..., i].

Computes the eigen decomposition of a batch of self-adjoint matrices

(Note: Only real inputs are supported).

Computes the eigenvalues and eigenvectors of the innermost N-by-N matrices in tensor such that tensor[...,:,:] * v[..., :,i] = e[..., i] * v[...,:,i], for i=0...N-1.

xlaSelfAdjointEig' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Float	epsilon: the tolerance ratio.
-> Bool	lower: a boolean specifies whether the calculation is done with the lower triangular part or the upper triangular part.
-> Int64	max_iter: maximum number of sweep update, i.e., the whole lower triangular part or upper triangular part based on parameter lower. Heuristically, it has been argued that approximately logN sweeps are needed in practice (Ref: Golub & van Loan "Matrix Computation").
-> Tensor v'1 t	a: the input tensor.
-> (Tensor Build t, Tensor Build t)	(w, v) w: The eigenvalues in ascending order, each repeated according to its multiplicity. v: The column v[..., :, i] is the normalized eigenvector corresponding to the eigenvalue w[..., i].

xlaSend Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> ByteString	tensor_name: A string key that identifies the channel.
-> Tensor v'1 t	tensor: The tensor to send.
-> m' ControlNode

Sends the named tensor to another XLA computation. Wraps the XLA Send operator

documented at https://www.tensorflow.org/performance/xla/operation_semantics#send .

xlaSend' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> ByteString	tensor_name: A string key that identifies the channel.
-> Tensor v'1 t	tensor: The tensor to send.
-> m' ControlNode

xlaSharding Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input
-> Tensor Build t	output

An op which shards the input based on the given sharding attribute.

xlaSharding' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input
-> Tensor Build t	output

xlaSort Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	input: A `Tensor` of type T.
-> Tensor Build t	output: A `Tensor` of type T.

Wraps the XLA Sort operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#sort .

Sorts a tensor. Currently only sorts in ascending order are supported.

xlaSort' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	input: A `Tensor` of type T.
-> Tensor Build t	output: A `Tensor` of type T.

xlaSpmdFullToShardShape Source #

Arguments

:: forall v'1 t. TensorType t
=> ByteString	manual_sharding
-> Tensor v'1 t	input
-> Tensor Build t	output

An op used by XLA SPMD partitioner to switch from automatic partitioning to

manual partitioning. It annotates the input (full-shape, to be automatically partitioned) with the same sharding used by manual partitioning, and outputs a shard-shaped tensor to be consumed by later manually-partitioned ops. If the shape is not evenly partitionable, the padding region will be masked with 0s.

xlaSpmdFullToShardShape' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> ByteString	manual_sharding
-> Tensor v'1 t	input
-> Tensor Build t	output

xlaSpmdShardToFullShape Source #

Arguments

:: forall v'1 t. TensorType t
=> Shape	full_shape
-> ByteString	manual_sharding
-> Tensor v'1 t	input
-> Tensor Build t	output

An op used by XLA SPMD partitioner to switch from manual partitioning to

automatic partitioning. It converts the shard-shaped, manually partitioned input into full-shaped tensor to be partitioned automatically with the same sharding used by manual partitioning.

xlaSpmdShardToFullShape' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Shape	full_shape
-> ByteString	manual_sharding
-> Tensor v'1 t	input
-> Tensor Build t	output

xlaSvd Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> Float	epsilon: the tolerance ratio.
-> Int64	max_iter: maximum number of sweep update, i.e., the whole lower triangular part or upper triangular part based on parameter lower. Heuristically, it has been argued that approximately log(min (M, N)) sweeps are needed in practice (Ref: Golub & van Loan "Matrix Computation").
-> ByteString	precision_config: a serialized xla::PrecisionConfig proto.
-> Tensor v'1 t	a: the input tensor.
-> (Tensor Build t, Tensor Build t, Tensor Build t)	(s, u, v) s: Singular values. The values are sorted in reverse order of magnitude, so s[..., 0] is the largest value, s[..., 1] is the second largest, etc. u: Left singular vectors. v: Right singular vectors.

Computes the eigen decomposition of a batch of self-adjoint matrices

(Note: Only real inputs are supported).

Computes the eigenvalues and eigenvectors of the innermost M-by-N matrices in tensor such that tensor[...,:,:] = u[..., :, :] * Diag(s[..., :]) * Transpose(v[...,:,:]).

xlaSvd' Source #

Arguments

:: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t
=> OpParams
-> Float	epsilon: the tolerance ratio.
-> Int64	max_iter: maximum number of sweep update, i.e., the whole lower triangular part or upper triangular part based on parameter lower. Heuristically, it has been argued that approximately log(min (M, N)) sweeps are needed in practice (Ref: Golub & van Loan "Matrix Computation").
-> ByteString	precision_config: a serialized xla::PrecisionConfig proto.
-> Tensor v'1 t	a: the input tensor.
-> (Tensor Build t, Tensor Build t, Tensor Build t)	(s, u, v) s: Singular values. The values are sorted in reverse order of magnitude, so s[..., 0] is the largest value, s[..., 1] is the second largest, etc. u: Left singular vectors. v: Right singular vectors.

xlog1py Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

xlog1py' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

xlogy Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

xlogy' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	y
-> Tensor Build t	z

zerosLike Source #

Arguments

:: forall v'1 t. TensorType t
=> Tensor v'1 t	x
-> Tensor Build t	y

zerosLike' Source #

Arguments

:: forall v'1 t. TensorType t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

zeta Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> Tensor v'1 t	x
-> Tensor v'2 t	q
-> Tensor Build t	z

zeta' Source #

Arguments

:: forall v'1 v'2 t. OneOf '[Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor v'2 t	q
-> Tensor Build t	z

zipDataset Source #

Arguments

:: [DataType]	output_types
-> [Tensor v'1 Variant]	input_datasets
-> Tensor Build Variant	handle

zipDataset' Source #

Arguments

:: OpParams
-> [DataType]	output_types
-> [Tensor v'1 Variant]	input_datasets
-> Tensor Build Variant	handle

_Arg Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)

=> Int64

index: This argument is the index-th argument of the function.

Attributes for shape inference: 1. _output_shapes: this attribute should contain a list of TensorShapeProto describing the shape(s) of the tensor(s) this _Arg node will produce. If set, _Arg node's shape inference function will use it as the node's output shapes. 2. _handle_dtypes and _handle_shapes: these attributes can be set on an _Arg node producing resource output(s). If set, value of _handle_dtypes should contain the dtype(s) of the resource(s) and value of _handle_shapes should contain the shape(s) of the resource(s). If both attributes are set, _Arg node's shape inference function will use their values as the node's output handle's type(s) and shape(s).

-> m' (Tensor Value t)

output: The argument.

A graph node which represents an argument to a function.

_Arg' Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Int64	index: This argument is the index-th argument of the function. Attributes for shape inference: 1. _output_shapes: this attribute should contain a list of TensorShapeProto describing the shape(s) of the tensor(s) this _Arg node will produce. If set, _Arg node's shape inference function will use it as the node's output shapes. 2. _handle_dtypes and _handle_shapes: these attributes can be set on an _Arg node producing resource output(s). If set, value of _handle_dtypes should contain the dtype(s) of the resource(s) and value of _handle_shapes should contain the shape(s) of the resource(s). If both attributes are set, _Arg node's shape inference function will use their values as the node's output handle's type(s) and shape(s).
-> m' (Tensor Value t)	output: The argument.

_ArrayToList Source #

Arguments

:: forall v'1 t out_types. (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.

_ArrayToList' Source #

Arguments

:: forall v'1 t out_types. (TensorType t, TensorTypes out_types)
=> OpParams
-> [Tensor v'1 t]	input
-> TensorList Build out_types	output

_ConfigureDistributedTPU Source #

Arguments

:: forall v'1 m'. 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 #

Arguments

:: forall v'1 m'. 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.

_DeviceArg Source #

Arguments

:: forall t m'. (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.

_DeviceArg' Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Int64	index: This argument is the index-th argument of the function.
-> m' (Tensor Value t)	output: The argument.

_DeviceRetval Source #

Arguments

:: forall v'1 t m'. (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.

_DeviceRetval' Source #

Arguments

:: forall v'1 t m'. (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

_DisconnectHostFromDistributedTPUSystem Source #

Arguments

:: forall m'. 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 #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> m' (Tensor Value Int32)	number_of_tpu_chips: A scalar tensor containing the number of TPU chips on the host.

_FusedBatchNormEx Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t u. (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 v'6 t]	side_input
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u)	(y, batch_mean, batch_variance, reserve_space_1, reserve_space_2, reserve_space_3) y batch_mean batch_variance reserve_space_1 reserve_space_2 reserve_space_3

NOTE*: Do not invoke this operator directly in Python. Grappler is

expected to create these operators.

_FusedBatchNormEx' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 v'5 v'6 t u. (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 v'6 t]	side_input
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u, Tensor Build u)	(y, batch_mean, batch_variance, reserve_space_1, reserve_space_2, reserve_space_3) y batch_mean batch_variance reserve_space_1 reserve_space_2 reserve_space_3

_FusedConv2D Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> [Tensor v'3 t]	args
-> Tensor Build t	output

Performs a convolution followed by a specified series of operations.

The inputs to the convolution are input and filter. The series of operations that follows is specified by the fused_ops attribute, which is a list of TF op names specified as strings (e.g. Relu). They are performed in order, where the (first) input to each op is the output of the preceding op. The first input and the output of each fused_op must be of type T.

Currently supported fused_op combinations are: [X] and [X,A], where X is one of {BiasAdd,FusedBatchNorm} and A is one of {Elu,Relu,Relu6}.

The first input to op X is the Conv2D result, and the additional input(s) to X are specified by args.
If there is an op A specified, the output of op X is the input to op A, and op A produces the _FusedConv2D output. Otherwise, op X produces the _FusedConv2D output.
NOTE*: Do not invoke this operator directly in Python. Grappler is expected to create these operators.

_FusedConv2D' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> [Tensor v'3 t]	args
-> Tensor Build t	output

_FusedDepthwiseConv2dNative Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> [Tensor v'3 t]	args
-> Tensor Build t	output

_FusedDepthwiseConv2dNative' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> ByteString	padding
-> Tensor v'1 t	input
-> Tensor v'2 t	filter
-> [Tensor v'3 t]	args
-> Tensor Build t	output

_FusedMatMul Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Float] t
=> Tensor v'1 t	a
-> Tensor v'2 t	b
-> [Tensor v'3 t]	args
-> Tensor Build t	product

Performs a MatMul followed by a specified series of operations.

The inputs to the MatMul are specified by a and b. The series of operations that follows is specified by the fused_ops attribute, which is a list of TF op names specified as strings (e.g. Relu). They are performed in order, where the (first) input to each op is the output of the preceding op. The first input and the output of each fused_op must be of type T.

Currently supported fused_op combinations are: [BiasAdd] and [BiasAdd,A], where A is one of {Elu,Relu,Relu6}.

The first input to BiasAdd is the Conv2D result, and the additional BiasAdd input is specified by args.
If there is an op A specified, the output of the BiasAdd is the input to op A, and op A produces the _FusedConv2D output. Otherwise, the BiasAdd produces the _FusedConv2D output.
NOTE*: Do not invoke this operator directly in Python. Grappler is expected to create these operators.

_FusedMatMul' Source #

Arguments

:: forall v'1 v'2 v'3 t. OneOf '[Float] t
=> OpParams
-> Tensor v'1 t	a
-> Tensor v'2 t	b
-> [Tensor v'3 t]	args
-> Tensor Build t	product

_HostCast Source #

Arguments

:: forall v'1 srcT dstT. (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.

_HostCast' Source #

Arguments

:: forall v'1 srcT dstT. (TensorType srcT, TensorType dstT)
=> OpParams
-> Tensor v'1 srcT	x
-> Tensor Build dstT	y

_HostRecv Source #

Arguments

:: forall tensor_type m'. (MonadBuild m', TensorType tensor_type)
=> ByteString	recv_device: The name of the device receiving the tensor.
-> ByteString	send_device: The name of the device sending the tensor.
-> Int64	send_device_incarnation: The current incarnation of send_device.
-> ByteString	tensor_name: The name of the tensor to receive.
-> m' (Tensor Value tensor_type)	tensor: The tensor to receive.

Receives the named tensor from send_device on recv_device.

_HostRecv produces its output on host memory whereas _Recv produces its output on device memory.

_HostRecv' Source #

Arguments

:: forall tensor_type m'. (MonadBuild m', TensorType tensor_type)
=> OpParams
-> ByteString	recv_device: The name of the device receiving the tensor.
-> ByteString	send_device: The name of the device sending the tensor.
-> Int64	send_device_incarnation: The current incarnation of send_device.
-> ByteString	tensor_name: The name of the tensor to receive.
-> m' (Tensor Value tensor_type)	tensor: The tensor to receive.

_HostSend Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> ByteString	recv_device: The name of the device receiving the tensor.
-> ByteString	send_device: The name of the device sending the tensor.
-> Int64	send_device_incarnation: The current incarnation of send_device.
-> ByteString	tensor_name: The name of the tensor to send.
-> 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.

_HostSend' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> ByteString	recv_device: The name of the device receiving the tensor.
-> ByteString	send_device: The name of the device sending the tensor.
-> Int64	send_device_incarnation: The current incarnation of send_device.
-> ByteString	tensor_name: The name of the tensor to send.
-> Tensor v'1 t	tensor: The tensor to send.
-> m' ControlNode

_InitializeHostForDistributedTPU Source #

Arguments

:: forall v'1 m'. 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 #

Arguments

:: forall v'1 m'. 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.

_ListToArray Source #

Arguments

:: forall v'1 tin t. (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.

_ListToArray' Source #

Arguments

:: forall v'1 tin t. (TensorTypes tin, TensorType t)
=> OpParams
-> Int64	N
-> TensorList v'1 tin	input
-> [Tensor Build t]	output

_MklMaximum Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. 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) 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

_MklMaximum' Source #

Arguments

:: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Int64, Word16, Double, Float] t
=> OpParams
-> 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) z mkl_z

_MklMul Source #

Arguments

:: 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	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) z mkl_z

Returns x * y element-wise.

NOTE*: Mul supports broadcasting. More about broadcasting here

_MklMul' Source #

Arguments

:: 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	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) z mkl_z

_MklSquaredDifference Source #

Arguments

:: 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	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) z mkl_z

Returns (x - y)(x - y) element-wise.

NOTE*: SquaredDifference supports broadcasting. More about broadcasting here

_MklSquaredDifference' Source #

Arguments

:: 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	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) z mkl_z

_MklSub Source #

Arguments

:: 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	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) z mkl_z

Returns x - y element-wise.

NOTE*: Sub supports broadcasting. More about broadcasting here

_MklSub' Source #

Arguments

:: 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	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) z mkl_z

_NcclBroadcastRecv Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> Int64	num_devices: The number of devices participating in this reduction.
-> ByteString	shared_name: Identifier that is shared between ops of the same broadcast.
-> Tensor v'1 Int32	shape: The shape of the output.
-> m' (Tensor Value t)	output: The broadcast data received from the NcclBroadcastSend op.

Replacement node for NcclBroadcast.

Sends data of shape shape from the _NcclBroadcastSend op registered in the same shared_name. The graph should be constructed so that one device runs _NcclBroadcastSend and `num_devices-1` devices run _NcclBroadcastRecv ops with shared_name value c. Failure to do so will cause the graph execution to fail to complete.

_NcclBroadcastRecv' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> OpParams
-> Int64	num_devices: The number of devices participating in this reduction.
-> ByteString	shared_name: Identifier that is shared between ops of the same broadcast.
-> Tensor v'1 Int32	shape: The shape of the output.
-> m' (Tensor Value t)	output: The broadcast data received from the NcclBroadcastSend op.

_NcclBroadcastSend Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> Int64	num_devices: The number of devices participating in this reduction.
-> ByteString	shared_name: Identifier that is shared between ops of the same broadcast.
-> Tensor v'1 t	input: The input to the broadcast.
-> m' ControlNode

Replacement node for NcclBroadcast.

Sends input to the _NcclBroadcastRecv ops registered in the same shared_name. The graph should be constructed so that one device runs _NcclBroadcastSend and `num_devices-1` devices run _NcclBroadcastRecv ops with shared_name value c. Failure to do so will cause the graph execution to fail to complete.

_NcclBroadcastSend' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> OpParams
-> Int64	num_devices: The number of devices participating in this reduction.
-> ByteString	shared_name: Identifier that is shared between ops of the same broadcast.
-> Tensor v'1 t	input: The input to the broadcast.
-> m' ControlNode

_NcclReduceRecv Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> Int64	num_devices: The number of devices participating in this reduction.
-> ByteString	reduction: the reduction operation to perform.
-> ByteString	shared_name: Identifier that is shared between ops of the same reduce.
-> Tensor v'1 t	input: The input to the reduction.
-> m' (Tensor Value t)	data: The reduced data received from this op and the NcclReduceSend op.

Replacement node for NcclReduce.

Reduces input from this op and the NcclReduceSend ops registered in the same shared_name. The graph should be constructed so that 'num_devices-1' devices run _NcclReduceSend and one device runs _NcclReduceRecv op with shared_name value c. Failure to do so will cause the graph execution to fail to complete.

_NcclReduceRecv' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> OpParams
-> Int64	num_devices: The number of devices participating in this reduction.
-> ByteString	reduction: the reduction operation to perform.
-> ByteString	shared_name: Identifier that is shared between ops of the same reduce.
-> Tensor v'1 t	input: The input to the reduction.
-> m' (Tensor Value t)	data: The reduced data received from this op and the NcclReduceSend op.

_NcclReduceSend Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> Int64	num_devices: The number of devices participating in this reduction.
-> ByteString	reduction: the reduction operation to perform.
-> ByteString	shared_name: Identifier that is shared between ops of the same reduce.
-> Tensor v'1 t	input: The input to the reduction.
-> m' ControlNode

Replacement node for NcclReduce.

Reduces input to the NcclReduceRecv op registered in the same shared_name. The graph should be constructed so that 'num_devices-1' devices run _NcclReduceSend and one device runs _NcclReduceRecv op with shared_name value c. Failure to do so will cause the graph execution to fail to complete.

_NcclReduceSend' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t)
=> OpParams
-> Int64	num_devices: The number of devices participating in this reduction.
-> ByteString	reduction: the reduction operation to perform.
-> ByteString	shared_name: Identifier that is shared between ops of the same reduce.
-> Tensor v'1 t	input: The input to the reduction.
-> m' ControlNode

_ParallelConcatStart Source #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> Shape	shape: 1-D `Tensor` indicating the shape of the output.
-> 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 #

Arguments

:: forall dtype m'. (MonadBuild m', TensorType dtype)
=> OpParams
-> Shape	shape: 1-D `Tensor` indicating the shape of the output.
-> m' (Tensor Value dtype)	output: An empty Tensor of the specified type.

_ParallelConcatUpdate Source #

Arguments

:: forall v'1 v'2 t. 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` object that will be updated in-place.
-> Tensor v'2 t	update: A `Tensor` of rank one less than `value` if `loc` is a scalar, otherwise of rank equal to `value` that contains the new values for `value`.
-> Tensor Build t	output: `value` that has been updated accordingly.

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 #

Arguments

:: forall v'1 v'2 t. 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` object that will be updated in-place.
-> Tensor v'2 t	update: A `Tensor` of rank one less than `value` if `loc` is a scalar, otherwise of rank equal to `value` that contains the new values for `value`.
-> Tensor Build t	output: `value` that has been updated accordingly.

_ReadVariablesOp Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> [Tensor v'1 ResourceHandle]	resources
-> m' (TensorList Value dtypes)	values

_ReadVariablesOp' Source #

Arguments

:: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes)
=> OpParams
-> [Tensor v'1 ResourceHandle]	resources
-> m' (TensorList Value dtypes)	values

_Recv Source #

Arguments

:: forall tensor_type m'. (MonadBuild m', TensorType tensor_type)
=> ByteString	recv_device: The name of the device receiving the tensor.
-> ByteString	send_device: The name of the device sending the tensor.
-> Int64	send_device_incarnation: The current incarnation of send_device.
-> ByteString	tensor_name: The name of the tensor to receive.
-> m' (Tensor Value tensor_type)	tensor: The tensor to receive.

Receives the named tensor from send_device on recv_device.

_Recv' Source #

Arguments

:: forall tensor_type m'. (MonadBuild m', TensorType tensor_type)
=> OpParams
-> ByteString	recv_device: The name of the device receiving the tensor.
-> ByteString	send_device: The name of the device sending the tensor.
-> Int64	send_device_incarnation: The current incarnation of send_device.
-> ByteString	tensor_name: The name of the tensor to receive.
-> m' (Tensor Value tensor_type)	tensor: The tensor to receive.

_Retval Source #

Arguments

:: forall v'1 t m'. (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.

_Retval' Source #

Arguments

:: forall v'1 t m'. (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

_ScopedAllocator Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> Int64	expected_call_count
-> Int64	id
-> ByteString	sa_name
-> 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.

_ScopedAllocator' Source #

Arguments

:: forall t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Int64	expected_call_count
-> Int64	id
-> ByteString	sa_name
-> Shape	shape
-> m' (Tensor Value t)	output

_ScopedAllocatorConcat Source #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', TensorType t)
=> Int64	id
-> ByteString	sa_name
-> Shape	shape
-> Tensor v'1 t	backing
-> [Tensor v'2 t]	inputs
-> m' (Tensor Value t)	output

Acts like a Concat Op that merges multiple 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 #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Int64	id
-> ByteString	sa_name
-> Shape	shape
-> Tensor v'1 t	backing
-> [Tensor v'2 t]	inputs
-> m' (Tensor Value t)	output

_ScopedAllocatorSplit Source #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', TensorType t)
=> Int64	id
-> ByteString	sa_name
-> 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 #

Arguments

:: forall v'1 v'2 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> Int64	id
-> ByteString	sa_name
-> Tensor v'1 t	concat
-> [Tensor v'2 t]	split
-> m' [Tensor Value t]	output

_Send Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> ByteString	recv_device: The name of the device receiving the tensor.
-> ByteString	send_device: The name of the device sending the tensor.
-> Int64	send_device_incarnation: The current incarnation of send_device.
-> ByteString	tensor_name: The name of the tensor to send.
-> Tensor v'1 t	tensor: The tensor to send.
-> m' ControlNode

Sends the named tensor from send_device to recv_device.

_Send' Source #

Arguments

:: forall v'1 t m'. (MonadBuild m', TensorType t)
=> OpParams
-> ByteString	recv_device: The name of the device receiving the tensor.
-> ByteString	send_device: The name of the device sending the tensor.
-> Int64	send_device_incarnation: The current incarnation of send_device.
-> ByteString	tensor_name: The name of the tensor to send.
-> Tensor v'1 t	tensor: The tensor to send.
-> m' ControlNode

_SetGlobalTPUArray Source #

Arguments

:: forall v'1 m'. 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.

_SetGlobalTPUArray' Source #

Arguments

:: forall v'1 m'. 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 #

_SwitchN Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Int64	num_outs
-> Tensor v'1 t	data
-> Tensor v'2 Int32	output_index
-> [Tensor Build t]	outputs

_SwitchN' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Int64	num_outs
-> Tensor v'1 t	data
-> Tensor v'2 Int32	output_index
-> [Tensor Build t]	outputs

_UnaryOpsComposition Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> Tensor v'1 t	x
-> Tensor Build t	y

NOTE*: Do not invoke this operator directly in Python. Graph rewrite pass is

expected to create these operators.

_UnaryOpsComposition' Source #

Arguments

:: forall v'1 t. OneOf '[Word16, Double, Float] t
=> OpParams
-> Tensor v'1 t	x
-> Tensor Build t	y

_VarHandlesOp Source #

Arguments

:: forall m'. MonadBuild m'
=> Int64	N
-> [DataType]	dtypes
-> m' [Tensor Value ResourceHandle]	resources

_VarHandlesOp' Source #

Arguments

:: forall m'. MonadBuild m'
=> OpParams
-> Int64	N
-> [DataType]	dtypes
-> m' [Tensor Value ResourceHandle]	resources

_WaitForDistributedTPU Source #

Arguments

:: forall v'1 m'. 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 #

Arguments

:: forall v'1 m'. 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.

_XlaMerge Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> Tensor v'1 t	partitioned_call
-> Tensor v'2 t	xla_run
-> Tensor Build t	output

XLA Merge Op. For use by the XLA JIT only.

Merges the outputs from the PartitionedCall node and the _XlaRun node. Unlike the TensorFlow Merge op, which requires inputs of some types to be placed on the host, the _XlaMerge op can merge inputs of all types when placed on the device. This prevents the need for copy operations, in particular when an XLA cluster has int32 outputs. The _XlaMerge up does not have a value_index output that identifies the chosen input.

_XlaMerge' Source #

Arguments

:: forall v'1 v'2 t. TensorType t
=> OpParams
-> Tensor v'1 t	partitioned_call
-> Tensor v'2 t	xla_run
-> Tensor Build t	output

_XlaRecvAtHost Source #

Arguments

:: forall v'1 toutputs m'. (MonadBuild m', TensorTypes toutputs)
=> Int64	device_ordinal: The device to use.
-> ByteString	key: A key that is unique in the computation and associates the send with the consumer in the XLA computation.
-> 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 to receive values from a running XLA computation.

_XlaRecvAtHost' Source #

Arguments

:: forall v'1 toutputs m'. (MonadBuild m', TensorTypes toutputs)
=> OpParams
-> Int64	device_ordinal: The device to use.
-> ByteString	key: A key that is unique in the computation and associates the send with the consumer in the XLA computation.
-> 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.

_XlaRun Source #

Arguments

:: forall v'1 v'2 targs tresults m'. (MonadBuild m', TensorTypes targs, TensorTypes tresults)
=> TensorList v'1 targs	args
-> Tensor v'2 ByteString	key
-> m' (TensorList Value tresults)	results

XLA Run Op. For use by the XLA JIT only.

Executes a TensorFlow function previously compiled into a LocalExecutable by an _XlaCompile op.

_XlaRun' Source #

Arguments

:: forall v'1 v'2 targs tresults m'. (MonadBuild m', TensorTypes targs, TensorTypes tresults)
=> OpParams
-> TensorList v'1 targs	args
-> Tensor v'2 ByteString	key
-> m' (TensorList Value tresults)	results

_XlaSendFromHost Source #

Arguments

:: forall v'1 v'2 tinputs m'. (MonadBuild m', TensorTypes tinputs)
=> Int64	device_ordinal: The device to use.
-> ByteString	key: A key that is unique in the computation and associates the send with the consumer in the XLA computation.
-> 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 to send values to a running XLA computation.

_XlaSendFromHost' Source #

Arguments

:: forall v'1 v'2 tinputs m'. (MonadBuild m', TensorTypes tinputs)
=> OpParams
-> Int64	device_ordinal: The device to use.
-> ByteString	key: A key that is unique in the computation and associates the send with the consumer in the XLA computation.
-> 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