Class MLGraphBuilder

Hierarchy

MLGraphBuilder

Index

Constructors

constructor

Methods

Constructors

constructor

new MLGraphBuilder(context: MLContext): MLGraphBuilder

- Defined in nn/graph_builder.ts:273
Parameters
- context: MLContext
Returns MLGraphBuilder

Methods

add

add(a: MLOperand, b: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:364
spec

Parameters
- a: MLOperand
- b: MLOperand
Returns MLOperand

averagePool2d

averagePool2d(input: MLOperand, options?: MLPooling2dOptions): MLOperand

- Defined in nn/graph_builder.ts:533
spec

Parameters
- input: MLOperand
- options: MLPooling2dOptions = {}
Returns MLOperand

batchNormalization

batchNormalization(input: MLOperand, mean: MLOperand, variance: MLOperand, options?: MLBatchNormalizationOptions): MLOperand

- Defined in nn/graph_builder.ts:326
spec

Parameters
- input: MLOperand
- mean: MLOperand
- variance: MLOperand
- options: MLBatchNormalizationOptions = {}
Returns MLOperand

build

build(outputs: Record<string, MLOperand>): MLGraph

- Defined in nn/graph_builder.ts:289
spec

Parameters
- outputs: Record<string, MLOperand>
Returns MLGraph

clamp

clamp(x: MLOperand, options?: MLClampOptions): MLOperand

- Defined in nn/graph_builder.ts:337
spec

Parameters
- x: MLOperand
- options: MLClampOptions = {}
Returns MLOperand

concat

concat(inputs: MLOperand[], axis: number): MLOperand

- Defined in nn/graph_builder.ts:345
spec

Parameters
- inputs: MLOperand[]
- axis: number
Returns MLOperand

constant

constant(desc: MLOperandDescriptor, bufferView: MLBufferView): MLOperand
constant(value: number, type?: MLOperandType): MLOperand

- Defined in nn/graph_builder.ts:303
spec

Parameters
- desc: MLOperandDescriptor
- bufferView: MLBufferView
Returns MLOperand
- Defined in nn/graph_builder.ts:307
spec

Parameters
- value: number
- Optional type: MLOperandType
Returns MLOperand

conv2d

conv2d(input: MLOperand, filter: MLOperand, options?: MLConv2dOptions): MLOperand

- Defined in nn/graph_builder.ts:353
spec

Parameters
- input: MLOperand
- filter: MLOperand
- options: MLConv2dOptions = {}
Returns MLOperand

div

div(a: MLOperand, b: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:388
spec

Parameters
- a: MLOperand
- b: MLOperand
Returns MLOperand

exp

exp(x: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:424
spec

Parameters
- x: MLOperand
Returns MLOperand

gemm

gemm(a: MLOperand, b: MLOperand, options?: MLGemmOptions): MLOperand

- Defined in nn/graph_builder.ts:457
spec

Parameters
- a: MLOperand
- b: MLOperand
- options: MLGemmOptions = {}
Returns MLOperand

gru

gru(input: MLOperand, weight: MLOperand, recurrentWeight: MLOperand, steps: number, hiddenSize: number, options?: MLGruOptions): MLOperand[]

- Defined in nn/graph_builder.ts:465
spec

Parameters
- input: MLOperand
- weight: MLOperand
- recurrentWeight: MLOperand
- steps: number
- hiddenSize: number
- options: MLGruOptions = {}
Returns MLOperand[]

gruCell

gruCell(input: MLOperand, weight: MLOperand, recurrentWeight: MLOperand, hiddenState: MLOperand, hiddenSize: number, options?: MLGruCellOptions): MLOperand

- Defined in nn/graph_builder.ts:479
spec

Parameters
- input: MLOperand
- weight: MLOperand
- recurrentWeight: MLOperand
- hiddenState: MLOperand
- hiddenSize: number
- options: MLGruCellOptions = {}
Returns MLOperand

input

input(name: string, desc: MLOperandDescriptor): MLOperand

- Defined in nn/graph_builder.ts:296
spec

Parameters
- name: string
- desc: MLOperandDescriptor
Returns MLOperand

instanceNormalization

instanceNormalization(input: MLOperand, options?: MLInstanceNormalizationOptions): MLOperand

- Defined in nn/graph_builder.ts:496
spec

Parameters
- input: MLOperand
- options: MLInstanceNormalizationOptions = {}
Returns MLOperand

leakyRelu

leakyRelu(x: MLOperand, options?: MLLeakyReluOptions): MLOperand

- Defined in nn/graph_builder.ts:506
spec

Parameters
- x: MLOperand
- options: MLLeakyReluOptions = {}
Returns MLOperand

matmul

matmul(a: MLOperand, b: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:514
spec

Parameters
- a: MLOperand
- b: MLOperand
Returns MLOperand

max

max(a: MLOperand, b: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:396
spec

Parameters
- a: MLOperand
- b: MLOperand
Returns MLOperand

maxPool2d

maxPool2d(input: MLOperand, options?: MLPooling2dOptions): MLOperand

- Defined in nn/graph_builder.ts:541
spec

Parameters
- input: MLOperand
- options: MLPooling2dOptions = {}
Returns MLOperand

min

min(a: MLOperand, b: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:404
spec

Parameters
- a: MLOperand
- b: MLOperand
Returns MLOperand

mul

mul(a: MLOperand, b: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:380
spec

Parameters
- a: MLOperand
- b: MLOperand
Returns MLOperand

pad

pad(input: MLOperand, padding: MLOperand, options?: MLPadOptions): MLOperand

- Defined in nn/graph_builder.ts:522
spec

Parameters
- input: MLOperand
- padding: MLOperand
- options: MLPadOptions = {}
Returns MLOperand

pow

pow(a: MLOperand, b: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:413
spec

Parameters
- a: MLOperand
- b: MLOperand
Returns MLOperand

reduceLogSumExp

reduceLogSumExp(input: MLOperand, options?: MLReduceOptions): MLOperand

- Defined in nn/graph_builder.ts:552
spec

Parameters
- input: MLOperand
- options: MLReduceOptions = {}
Returns MLOperand

reduceMax

reduceMax(input: MLOperand, options?: MLReduceOptions): MLOperand

- Defined in nn/graph_builder.ts:560
spec

Parameters
- input: MLOperand
- options: MLReduceOptions = {}
Returns MLOperand

reduceMean

reduceMean(input: MLOperand, options?: MLReduceOptions): MLOperand

- Defined in nn/graph_builder.ts:568
spec

Parameters
- input: MLOperand
- options: MLReduceOptions = {}
Returns MLOperand

reduceMin

reduceMin(input: MLOperand, options?: MLReduceOptions): MLOperand

- Defined in nn/graph_builder.ts:576
spec

Parameters
- input: MLOperand
- options: MLReduceOptions = {}
Returns MLOperand

reduceProduct

reduceProduct(input: MLOperand, options?: MLReduceOptions): MLOperand

- Defined in nn/graph_builder.ts:584
spec

Parameters
- input: MLOperand
- options: MLReduceOptions = {}
Returns MLOperand

reduceSum

reduceSum(input: MLOperand, options?: MLReduceOptions): MLOperand

- Defined in nn/graph_builder.ts:592
spec

Parameters
- input: MLOperand
- options: MLReduceOptions = {}
Returns MLOperand

relu

relu(input: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:432
spec

Parameters
- input: MLOperand
Returns MLOperand

resample

resample(input: MLOperand, options?: MLResampleOptions): MLOperand

- Defined in nn/graph_builder.ts:601
spec

Parameters
- input: MLOperand
- options: MLResampleOptions = {}
Returns MLOperand

reshape

reshape(input: MLOperand, newShape: number[]): MLOperand

- Defined in nn/graph_builder.ts:609
spec

Parameters
- input: MLOperand
- newShape: number[]
Returns MLOperand

sigmoid

sigmoid(x: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:440
spec

Parameters
- x: MLOperand
Returns MLOperand

slice

slice(input: MLOperand, starts: number[], sizes: number[], options?: MLSliceOptions): MLOperand

- Defined in nn/graph_builder.ts:617
spec

Parameters
- input: MLOperand
- starts: number[]
- sizes: number[]
- options: MLSliceOptions = {}
Returns MLOperand

softmax

softmax(x: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:627
spec

Parameters
- x: MLOperand
Returns MLOperand

split

split(input: MLOperand, splits: number | number[], options?: MLSplitOptions): MLOperand[]

- Defined in nn/graph_builder.ts:635
spec

Parameters
- input: MLOperand
- splits: number | number[]
- options: MLSplitOptions = {}
Returns MLOperand[]

squeeze

squeeze(input: MLOperand, options?: MLSqueezeOptions): MLOperand

- Defined in nn/graph_builder.ts:645
spec

Parameters
- input: MLOperand
- options: MLSqueezeOptions = {}
Returns MLOperand

sub

sub(a: MLOperand, b: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:372
spec

Parameters
- a: MLOperand
- b: MLOperand
Returns MLOperand

tanh

tanh(x: MLOperand): MLOperand

- Defined in nn/graph_builder.ts:448
spec

Parameters
- x: MLOperand
Returns MLOperand

transpose

transpose(input: MLOperand, options?: MLTransposeOptions): MLOperand

- Defined in nn/graph_builder.ts:653
spec

Parameters
- input: MLOperand
- options: MLTransposeOptions = {}
Returns MLOperand

Generated using TypeDoc