api/org.diffkt.tracing/TracingVisitor

TracingVisitor

interface TracingVisitor<R>

Functions

visit

open fun visit(x: Traceable): R

visitAtan

abstract fun visitAtan(x: TracingTensor.Atan): R

visitAvgPool

abstract fun visitAvgPool(x: TracingTensor.AvgPool): R

visitAvgPoolGrad

abstract fun visitAvgPoolGrad(x: TracingTensor.AvgPoolGrad): R

visitBroadcastTo

abstract fun visitBroadcastTo(x: TracingTensor.BroadcastTo): R

visitCompare

abstract fun visitCompare(x: TracingTensor.Compare): R

visitConcat

abstract fun visitConcat(x: TracingTensor.Concat): R

visitConstant

abstract fun visitConstant(x: TracingTensor.Constant): R

visitConvImpl

abstract fun visitConvImpl(x: TracingTensor.ConvImpl): R

visitCos

abstract fun visitCos(x: TracingTensor.Cos): R

visitDigamma

abstract fun visitDigamma(x: TracingTensor.Digamma): R

visitDiv

abstract fun visitDiv(x: TracingTensor.Div): R

visitExp

abstract fun visitExp(x: TracingTensor.Exp): R

visitExpand

abstract fun visitExpand(x: TracingTensor.Expand): R

visitFlip

abstract fun visitFlip(x: TracingTensor.Flip): R

visitGather

abstract fun visitGather(x: TracingTensor.Gather): R

visitGatherAtIndices

abstract fun visitGatherAtIndices(x: TracingTensor.GatherAtIndices): R

visitIdentityGradient

abstract fun visitIdentityGradient(x: TracingTensor.IdentityGradient): R

visitIfThenElse

abstract fun visitIfThenElse(x: TracingTensor.IfThenElse): R

visitLgamma

abstract fun visitLgamma(x: TracingTensor.Lgamma): R

visitLn

abstract fun visitLn(x: TracingTensor.Ln): R

visitLogSoftmax

abstract fun visitLogSoftmax(x: TracingTensor.LogSoftmax): R

visitLogSoftmaxGrad

abstract fun visitLogSoftmaxGrad(x: TracingTensor.LogSoftmaxGrad): R

visitMatmul

abstract fun visitMatmul(x: TracingTensor.Matmul): R

visitMaxPoolWithIndices

abstract fun visitMaxPoolWithIndices(x: TracingTensor.MaxPoolWithIndices): R

visitMeld

abstract fun visitMeld(x: TracingTensor.Meld): R

visitMinus

abstract fun visitMinus(x: TracingTensor.Minus): R

visitOuterProduct

abstract fun visitOuterProduct(x: TracingTensor.OuterProduct): R

visitPlus

abstract fun visitPlus(x: TracingTensor.Plus): R

visitPolygamma

abstract fun visitPolygamma(x: TracingTensor.Polygamma): R

visitPow

abstract fun visitPow(x: TracingTensor.Pow): R

visitRandomFloats

abstract fun visitRandomFloats(x: TracingTensor.RandomFloats): R

visitRandomSplit

abstract fun visitRandomSplit(x: TracingRandomKey.Split): R

visitRandomSplitPart

abstract fun visitRandomSplitPart(x: TracingRandomKey.SplitPart): R

visitRandomVariable

abstract fun visitRandomVariable(x: TracingRandomKey.Variable): R

visitRelu

abstract fun visitRelu(x: TracingTensor.Relu): R

visitReluGrad

abstract fun visitReluGrad(x: TracingTensor.ReluGrad): R

visitReshape

abstract fun visitReshape(x: TracingTensor.Reshape): R

visitReshapeToScalar

abstract fun visitReshapeToScalar(x: TracingScalar.ReshapeToScalar): R

visitScatter

abstract fun visitScatter(x: TracingTensor.Scatter): R

visitScatterAtIndices

abstract fun visitScatterAtIndices(x: TracingTensor.ScatterAtIndices): R

visitSigmoid

abstract fun visitSigmoid(x: TracingTensor.Sigmoid): R

visitSin

abstract fun visitSin(x: TracingTensor.Sin): R

visitSplit

abstract fun visitSplit(x: TracingTensor.Split): R

visitSplitPart

abstract fun visitSplitPart(x: TracingTensor.SplitPart): R

visitSqrt

abstract fun visitSqrt(x: TracingTensor.Sqrt): R

visitSqueeze

abstract fun visitSqueeze(x: TracingTensor.Squeeze): R

visitSum

abstract fun visitSum(x: TracingTensor.Sum): R

visitTan

abstract fun visitTan(x: TracingTensor.Tan): R

visitTanh

abstract fun visitTanh(x: TracingTensor.Tanh): R

visitTimes

abstract fun visitTimes(x: TracingTensor.Times): R

visitTimesScalar

abstract fun visitTimesScalar(x: TracingTensor.TimesScalar): R

visitTranspose

abstract fun visitTranspose(x: TracingTensor.Transpose): R

visitUnaryMinus

abstract fun visitUnaryMinus(x: TracingTensor.UnaryMinus): R

visitUnsqueeze

abstract fun visitUnsqueeze(x: TracingTensor.Unsqueeze): R

visitVariable

abstract fun visitVariable(x: TracingTensor.Variable): R

visitView1

abstract fun visitView1(x: TracingTensor.View1): R

visitView2

abstract fun visitView2(x: TracingTensor.View2): R

visitView3

abstract fun visitView3(x: TracingTensor.View3): R

visitZero

abstract fun visitZero(x: TracingTensor.Zero): R