hub / github.com/tensorflow/lingvo / _ApplyProjectionKernel

Method _ApplyProjectionKernel

lingvo/core/layers.py:1294–1370 · view source on GitHub ↗

Applies projection. Args: w: Weight matrix. b: Bias vector (or None). inputs: FProp inputs. mix_kernel: (optional) mix_kernel for block diagonal matmul. Returns: Output tensor with projection applied.

(self,
                             w,
                             b,
                             inputs,
                             mix_kernel=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

FPropMethod · 0.95

_GetWeightsMethod · 0.95

Calls 4

ToAqtInputsMethod · 0.45

QEinsumMethod · 0.45

FromAqtMatmulMethod · 0.45

MeshSplitMethod · 0.45

Tested by

no test coverage detected