-The following code showcases the computation with dynamic input dimensions.
-
-function sizeOfShape(array) {
- return array.reduce(
- (accumulator, currentValue) => accumulator * currentValue);
-}
-
-const context = navigator.ml.createContext();
-
-// Create a graph with dynamic shaped inputs.
-const builder = new MLGraphBuilder(context);
-const descA = {type: 'float32', dimensions: [-1, 4]};
-const a = builder.input('a', descA);
-const descB = {type: 'float32', dimensions: [4, -1]};
-const b = builder.input('b', descB);
-const c = builder.matmul(a, b);
-const graph = builder.build({'c': c});
-
-function allocateAndCompute(shapeA, shapeB, shapeC) {
- const bufferA = new Float32Array(sizeOfShape(shapeA)).fill(0.5);
- const bufferB = new Float32Array(sizeOfShape(shapeB)).fill(0.5);
- const bufferC = new Float32Array(sizeOfShape(shapeC));
-
- // Specify the shape of inputs when computing.
- const inputs = {
- 'a': {resource: bufferA, dimensions: shapeA},
- 'b': {resource: bufferB, dimensions: shapeB},
- };
- const outputs = {'c': bufferC};
- context.compute(graph, inputs, outputs);
- console.log(`values: ${bufferC}`);
-}
-
-allocateAndCompute([3, 4], [4, 3], [3, 3]);
-allocateAndCompute([4, 4], [4, 4], [4, 4]);
-allocateAndCompute([5, 4], [4, 5], [5, 5]);
-
-
-The following code showcases the computation with optional outputs.
+The following code showcases the synchronous computation with optional outputs in a worker.
const context = navigator.ml.createContext();
@@ -809,8 +746,8 @@ Asynchronously carries out the computational workload of a compiled graph {{MLGr
@@ -818,8 +755,8 @@ partial interface MLContext {
**Arguments:**
- *graph*: an {{MLGraph}}. The compiled graph to be executed.
- - *inputs*: an {{MLNamedArrayInputs}}. The resources and optional dimensions of inputs.
- - *outputs*: an {{MLNamedArrayOutputs}}. The pre-allocated resources of required outputs.
+ - *inputs*: an {{MLNamedArrayBufferViews}}. The resources of inputs.
+ - *outputs*: an {{MLNamedArrayBufferViews}}. The pre-allocated resources of required outputs.
**Returns:** Promise<{{undefined}}>.
@@ -828,54 +765,37 @@ partial interface MLContext {
1. For each |key| -> |value| of |inputs|:
1. |graph|.{{MLGraph/[[inputDescriptors]]}}[|key|] must exist.
1. Let |inputDesc| be |graph|.{{MLGraph/[[inputDescriptors]]}}[|key|].
- 1. Let |inputSize| be 1.
- 1. If |value| is an {{MLArrayInput}}, then:
- 1. The length of |value|.{{MLArrayInput/dimensions}} must be the same as the length of |inputDesc|.{{MLOperandDescriptor/dimensions}}.
- 1. Let |i| be 0.
- 1. While true:
- 1. Let |dimension| be |value|.{{MLArrayInput/dimensions}}[|i|].
- 1. |dimension| must be greater than 0.
- 1. If |inputDesc|.{{MLOperandDescriptor/dimensions}}[|i|] is greater than 0, then |dimension| must be equal to |inputDesc|.{{MLOperandDescriptor/dimensions}}[|i|].
- 1. Set |inputSize| to the product of |inputSize| and |dimension|.
- 1. Increment |i| by 1.
- 1. If |i| if equal to the length of |value|.{{MLArrayInput/dimensions}}, then break.
- 1. Else:
- 1. For each |dimension| of |inputDesc|.{{MLOperandDescriptor/dimensions}}:
- 1. The value of |dimension| must be greater than 0.
- 1. Set |inputSize| to the product of |inputSize| and |dimension|.
- 1. If |value| is an {{MLArrayInput}}, then let |resource| be |value|.{{MLArrayInput/resource}}.
- 1. If |resource| is an {{ArrayBufferView}}, then:
- 1. The kind of |resource| must be compatible with |inputDesc|.{{MLOperandDescriptor/type}} according to [this table](#appendices-mloperandtype-arraybufferview-compatibility).
- 1. The length of |resource| must be the same as |inputSize|.
+ 1. The type of {{ArrayBufferView}} |value| must match |inputDesc|.{{MLOperandDescriptor/type}} according to [this table](#appendices-mloperandtype-arraybufferview-compatibility).
+ 1. |value|.\[[ByteLength]] must equal to [=byte length=] of |inputDesc|.
1. For each |key| -> |value| of |outputs|:
- 1. |graph|.{{MLGraph/[[outputNames]]}}[|key|] must exist.
+ 1. |graph|.{{MLGraph/[[outputDescriptors]]}}[|key|] must exist.
+ 1. Let |outputDesc| be |graph|.{{MLGraph/[[outputDescriptors]]}}[|key|].
+ 1. The type of {{ArrayBufferView}} |value| must match |outputDesc|.{{MLOperandDescriptor/type}} according to [this table](#appendices-mloperandtype-arraybufferview-compatibility).
+ 1. |value|.\[[ByteLength]] must equal to [=byte length=] of |outputDesc|.