def activation_ns

stelliferous@0.0.2.0 · kernel_planning.bend:130

import stelliferous@0.0.2.0/kernel_planning.bend as KernelPlanning
import Base
import stelliferous@0.0.2.0/kernel_cost.bend as Cost
import stelliferous@0.0.2.0/math_activation.bend as Activations

130# The activation pass reads and writes each logical output in place; the
131# identity skips the pass.
132def activation_ns(+activation: Activations.Activation,+config: Cost.Configuration) -> F32:
133  Bool.pick(F32,Activations.is_identity(activation),0.0,Cost.element_ns(4.0,Activations.operations(activation),config))

Uses (7)

Used by (1)