Skip to content

Function: layerCategory()

ts
function layerCategory(type): LayerCategory;

Defined in: ml/model.ts:106

Classify a layer/op type string ("Conv2d", "aten::relu") into a family.

Parameters

ParameterType
typestring

Returns

LayerCategory

MIT licensed. WebGL2 required.