diff options
Diffstat (limited to 'testing/web-platform/tests/interfaces/webnn.idl')
-rw-r--r-- | testing/web-platform/tests/interfaces/webnn.idl | 43 |
1 files changed, 23 insertions, 20 deletions
diff --git a/testing/web-platform/tests/interfaces/webnn.idl b/testing/web-platform/tests/interfaces/webnn.idl index 0b8ea7cb34..9af2879214 100644 --- a/testing/web-platform/tests/interfaces/webnn.idl +++ b/testing/web-platform/tests/interfaces/webnn.idl @@ -64,11 +64,7 @@ enum MLOperandDataType { }; dictionary MLOperandDescriptor { - // The operand type. required MLOperandDataType dataType; - - // The dimensions field is empty for scalar operands, - // and non-empty for tensor operands. sequence<[EnforceRange] unsigned long> dimensions = []; }; @@ -122,7 +118,7 @@ dictionary MLBatchNormalizationOptions { partial interface MLGraphBuilder { MLOperand batchNormalization(MLOperand input, MLOperand mean, MLOperand variance, - optional MLBatchNormalizationOptions options = {}); + optional MLBatchNormalizationOptions options = {}); }; partial interface MLGraphBuilder { @@ -162,7 +158,9 @@ dictionary MLConv2dOptions { }; partial interface MLGraphBuilder { - MLOperand conv2d(MLOperand input, MLOperand filter, optional MLConv2dOptions options = {}); + MLOperand conv2d(MLOperand input, + MLOperand filter, + optional MLConv2dOptions options = {}); }; enum MLConvTranspose2dFilterOperandLayout { @@ -242,7 +240,14 @@ dictionary MLGatherOptions { }; partial interface MLGraphBuilder { - MLOperand gather(MLOperand input, MLOperand indices, optional MLGatherOptions options = {}); + MLOperand gather(MLOperand input, + MLOperand indices, + optional MLGatherOptions options = {}); +}; + +partial interface MLGraphBuilder { + MLOperand gelu(MLOperand input); + MLActivation gelu(); }; dictionary MLGemmOptions { @@ -329,7 +334,7 @@ dictionary MLInstanceNormalizationOptions { partial interface MLGraphBuilder { MLOperand instanceNormalization(MLOperand input, - optional MLInstanceNormalizationOptions options = {}); + optional MLInstanceNormalizationOptions options = {}); }; dictionary MLLayerNormalizationOptions { @@ -340,7 +345,8 @@ dictionary MLLayerNormalizationOptions { }; partial interface MLGraphBuilder { - MLOperand layerNormalization(MLOperand input, optional MLLayerNormalizationOptions options = {}); + MLOperand layerNormalization(MLOperand input, + optional MLLayerNormalizationOptions options = {}); }; dictionary MLLeakyReluOptions { @@ -509,17 +515,13 @@ partial interface MLGraphBuilder { }; partial interface MLGraphBuilder { - MLOperand softmax(MLOperand input); - MLActivation softmax(); -}; - -dictionary MLSoftplusOptions { - float steepness = 1; + MLOperand softmax(MLOperand input, unsigned long axis); + MLActivation softmax(unsigned long axis); }; partial interface MLGraphBuilder { - MLOperand softplus(MLOperand input, optional MLSoftplusOptions options = {}); - MLActivation softplus(optional MLSoftplusOptions options = {}); + MLOperand softplus(MLOperand input); + MLActivation softplus(); }; partial interface MLGraphBuilder { @@ -532,9 +534,10 @@ dictionary MLSplitOptions { }; partial interface MLGraphBuilder { - sequence<MLOperand> split(MLOperand input, - ([EnforceRange] unsigned long or sequence<[EnforceRange] unsigned long>) splits, - optional MLSplitOptions options = {}); + sequence<MLOperand> split( + MLOperand input, + ([EnforceRange] unsigned long or sequence<[EnforceRange] unsigned long>) splits, + optional MLSplitOptions options = {}); }; partial interface MLGraphBuilder { |