// META: title=test WebNN API tanh operation
// META: global=window
// META: variant=?cpu
// META: variant=?gpu
// META: variant=?npu
// META: script=../resources/utils.js
// META: timeout=long

'use strict';

// https://www.w3.org/TR/webnn/#api-mlgraphbuilder-hard-swish
// Computes the nonlinear function y = x * max(0, min(6, (x + 3))) / 6 that is
// introduced by MobileNetV3 on the input tensor element-wise.
//
// MLOperand hardSwish(MLOperand input);

const hardSwishTests = [
  {
    'name': 'hardSwish float32 0D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [0.7341583371162415],
          'descriptor': {shape: [], dataType: 'float32'}
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [0.4569105803966522],
          'descriptor': {shape: [], dataType: 'float32'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float32 1D constant tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.7341583371162415,  9.11885929107666,    3.545238494873047,
            2.621943950653076,   -6.445507526397705,  -1.6835596561431885,
            5.52318000793457,    -5.958856105804443,  -9.169190406799316,
            6.420943737030029,   -3.2930312156677246, 1.041016697883606,
            -7.2463226318359375, -0.9472730755805969, -5.7783522605896,
            3.1852290630340576,  -7.261817932128906,  4.174602508544922,
            3.7802627086639404,  -6.071240425109863,  -9.909919738769531,
            -7.744259357452393,  -8.286120414733887,  8.083491325378418
          ],
          'descriptor': {shape: [24], dataType: 'float32'},
          'constant': true
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.4569105803966522,
            9.11885929107666,
            3.545238494873047,
            2.4567370414733887,
            0,
            -0.3693843185901642,
            5.52318000793457,
            0,
            0,
            6.420943737030029,
            0,
            0.7011276483535767,
            0,
            -0.3240821659564972,
            0,
            3.1852290630340576,
            0,
            4.174602508544922,
            3.7802627086639404,
            0,
            0,
            0,
            0,
            8.083491325378418
          ],
          'descriptor': {shape: [24], dataType: 'float32'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float32 1D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.7341583371162415,  9.11885929107666,    3.545238494873047,
            2.621943950653076,   -6.445507526397705,  -1.6835596561431885,
            5.52318000793457,    -5.958856105804443,  -9.169190406799316,
            6.420943737030029,   -3.2930312156677246, 1.041016697883606,
            -7.2463226318359375, -0.9472730755805969, -5.7783522605896,
            3.1852290630340576,  -7.261817932128906,  4.174602508544922,
            3.7802627086639404,  -6.071240425109863,  -9.909919738769531,
            -7.744259357452393,  -8.286120414733887,  8.083491325378418
          ],
          'descriptor': {shape: [24], dataType: 'float32'}
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.4569105803966522,
            9.11885929107666,
            3.545238494873047,
            2.4567370414733887,
            0,
            -0.3693843185901642,
            5.52318000793457,
            0,
            0,
            6.420943737030029,
            0,
            0.7011276483535767,
            0,
            -0.3240821659564972,
            0,
            3.1852290630340576,
            0,
            4.174602508544922,
            3.7802627086639404,
            0,
            0,
            0,
            0,
            8.083491325378418
          ],
          'descriptor': {shape: [24], dataType: 'float32'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float32 2D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.7341583371162415,  9.11885929107666,    3.545238494873047,
            2.621943950653076,   -6.445507526397705,  -1.6835596561431885,
            5.52318000793457,    -5.958856105804443,  -9.169190406799316,
            6.420943737030029,   -3.2930312156677246, 1.041016697883606,
            -7.2463226318359375, -0.9472730755805969, -5.7783522605896,
            3.1852290630340576,  -7.261817932128906,  4.174602508544922,
            3.7802627086639404,  -6.071240425109863,  -9.909919738769531,
            -7.744259357452393,  -8.286120414733887,  8.083491325378418
          ],
          'descriptor': {shape: [4, 6], dataType: 'float32'}
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.4569105803966522,
            9.11885929107666,
            3.545238494873047,
            2.4567370414733887,
            0,
            -0.3693843185901642,
            5.52318000793457,
            0,
            0,
            6.420943737030029,
            0,
            0.7011276483535767,
            0,
            -0.3240821659564972,
            0,
            3.1852290630340576,
            0,
            4.174602508544922,
            3.7802627086639404,
            0,
            0,
            0,
            0,
            8.083491325378418
          ],
          'descriptor': {shape: [4, 6], dataType: 'float32'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float32 3D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.7341583371162415,  9.11885929107666,    3.545238494873047,
            2.621943950653076,   -6.445507526397705,  -1.6835596561431885,
            5.52318000793457,    -5.958856105804443,  -9.169190406799316,
            6.420943737030029,   -3.2930312156677246, 1.041016697883606,
            -7.2463226318359375, -0.9472730755805969, -5.7783522605896,
            3.1852290630340576,  -7.261817932128906,  4.174602508544922,
            3.7802627086639404,  -6.071240425109863,  -9.909919738769531,
            -7.744259357452393,  -8.286120414733887,  8.083491325378418
          ],
          'descriptor': {shape: [2, 3, 4], dataType: 'float32'}
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.4569105803966522,
            9.11885929107666,
            3.545238494873047,
            2.4567370414733887,
            0,
            -0.3693843185901642,
            5.52318000793457,
            0,
            0,
            6.420943737030029,
            0,
            0.7011276483535767,
            0,
            -0.3240821659564972,
            0,
            3.1852290630340576,
            0,
            4.174602508544922,
            3.7802627086639404,
            0,
            0,
            0,
            0,
            8.083491325378418
          ],
          'descriptor': {shape: [2, 3, 4], dataType: 'float32'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float32 4D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.7341583371162415,  9.11885929107666,    3.545238494873047,
            2.621943950653076,   -6.445507526397705,  -1.6835596561431885,
            5.52318000793457,    -5.958856105804443,  -9.169190406799316,
            6.420943737030029,   -3.2930312156677246, 1.041016697883606,
            -7.2463226318359375, -0.9472730755805969, -5.7783522605896,
            3.1852290630340576,  -7.261817932128906,  4.174602508544922,
            3.7802627086639404,  -6.071240425109863,  -9.909919738769531,
            -7.744259357452393,  -8.286120414733887,  8.083491325378418
          ],
          'descriptor': {shape: [2, 2, 2, 3], dataType: 'float32'}
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.4569105803966522,
            9.11885929107666,
            3.545238494873047,
            2.4567370414733887,
            0,
            -0.3693843185901642,
            5.52318000793457,
            0,
            0,
            6.420943737030029,
            0,
            0.7011276483535767,
            0,
            -0.3240821659564972,
            0,
            3.1852290630340576,
            0,
            4.174602508544922,
            3.7802627086639404,
            0,
            0,
            0,
            0,
            8.083491325378418
          ],
          'descriptor': {shape: [2, 2, 2, 3], dataType: 'float32'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float32 5D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.7341583371162415,  9.11885929107666,    3.545238494873047,
            2.621943950653076,   -6.445507526397705,  -1.6835596561431885,
            5.52318000793457,    -5.958856105804443,  -9.169190406799316,
            6.420943737030029,   -3.2930312156677246, 1.041016697883606,
            -7.2463226318359375, -0.9472730755805969, -5.7783522605896,
            3.1852290630340576,  -7.261817932128906,  4.174602508544922,
            3.7802627086639404,  -6.071240425109863,  -9.909919738769531,
            -7.744259357452393,  -8.286120414733887,  8.083491325378418
          ],
          'descriptor': {shape: [2, 1, 4, 1, 3], dataType: 'float32'}
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.4569105803966522,
            9.11885929107666,
            3.545238494873047,
            2.4567370414733887,
            0,
            -0.3693843185901642,
            5.52318000793457,
            0,
            0,
            6.420943737030029,
            0,
            0.7011276483535767,
            0,
            -0.3240821659564972,
            0,
            3.1852290630340576,
            0,
            4.174602508544922,
            3.7802627086639404,
            0,
            0,
            0,
            0,
            8.083491325378418
          ],
          'descriptor': {shape: [2, 1, 4, 1, 3], dataType: 'float32'}
        }
      }
    }
  },

  // float16 tests
  {
    'name': 'hardSwish float16 0D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput':
            {'data': [0.734375], 'descriptor': {shape: [], dataType: 'float16'}}
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [0.45703125],
          'descriptor': {shape: [], dataType: 'float16'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float16 1D constant tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.734375,    9.1171875,   3.544921875, 2.62109375,   -6.4453125,
            -1.68359375, 5.5234375,   -5.95703125, -9.171875,    6.421875,
            -3.29296875, 1.041015625, -7.24609375, -0.947265625, -5.77734375,
            3.185546875, -7.26171875, 4.17578125,  3.779296875,  -6.0703125,
            -9.90625,    -7.74609375, -8.2890625,  8.0859375
          ],
          'descriptor': {shape: [24], dataType: 'float16'},
          'constant': true
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.45703125,  9.1171875,
            3.544921875, 2.455078125,
            0,           -0.369384765625,
            5.5234375,   0,
            0,           6.421875,
            0,           0.701171875,
            0,           -0.323974609375,
            0,           3.185546875,
            0,           4.17578125,
            3.779296875, 0,
            0,           0,
            0,           8.0859375
          ],
          'descriptor': {shape: [24], dataType: 'float16'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float16 1D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.734375,    9.1171875,   3.544921875, 2.62109375,   -6.4453125,
            -1.68359375, 5.5234375,   -5.95703125, -9.171875,    6.421875,
            -3.29296875, 1.041015625, -7.24609375, -0.947265625, -5.77734375,
            3.185546875, -7.26171875, 4.17578125,  3.779296875,  -6.0703125,
            -9.90625,    -7.74609375, -8.2890625,  8.0859375
          ],
          'descriptor': {shape: [24], dataType: 'float16'}
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.45703125,  9.1171875,
            3.544921875, 2.455078125,
            0,           -0.369384765625,
            5.5234375,   0,
            0,           6.421875,
            0,           0.701171875,
            0,           -0.323974609375,
            0,           3.185546875,
            0,           4.17578125,
            3.779296875, 0,
            0,           0,
            0,           8.0859375
          ],
          'descriptor': {shape: [24], dataType: 'float16'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float16 2D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.734375,    9.1171875,   3.544921875, 2.62109375,   -6.4453125,
            -1.68359375, 5.5234375,   -5.95703125, -9.171875,    6.421875,
            -3.29296875, 1.041015625, -7.24609375, -0.947265625, -5.77734375,
            3.185546875, -7.26171875, 4.17578125,  3.779296875,  -6.0703125,
            -9.90625,    -7.74609375, -8.2890625,  8.0859375
          ],
          'descriptor': {shape: [4, 6], dataType: 'float16'}
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.45703125,  9.1171875,
            3.544921875, 2.455078125,
            0,           -0.369384765625,
            5.5234375,   0,
            0,           6.421875,
            0,           0.701171875,
            0,           -0.323974609375,
            0,           3.185546875,
            0,           4.17578125,
            3.779296875, 0,
            0,           0,
            0,           8.0859375
          ],
          'descriptor': {shape: [4, 6], dataType: 'float16'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float16 3D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.734375,    9.1171875,   3.544921875, 2.62109375,   -6.4453125,
            -1.68359375, 5.5234375,   -5.95703125, -9.171875,    6.421875,
            -3.29296875, 1.041015625, -7.24609375, -0.947265625, -5.77734375,
            3.185546875, -7.26171875, 4.17578125,  3.779296875,  -6.0703125,
            -9.90625,    -7.74609375, -8.2890625,  8.0859375
          ],
          'descriptor': {shape: [2, 3, 4], dataType: 'float16'}
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.45703125,  9.1171875,
            3.544921875, 2.455078125,
            0,           -0.369384765625,
            5.5234375,   0,
            0,           6.421875,
            0,           0.701171875,
            0,           -0.323974609375,
            0,           3.185546875,
            0,           4.17578125,
            3.779296875, 0,
            0,           0,
            0,           8.0859375
          ],
          'descriptor': {shape: [2, 3, 4], dataType: 'float16'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float16 4D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.734375,    9.1171875,   3.544921875, 2.62109375,   -6.4453125,
            -1.68359375, 5.5234375,   -5.95703125, -9.171875,    6.421875,
            -3.29296875, 1.041015625, -7.24609375, -0.947265625, -5.77734375,
            3.185546875, -7.26171875, 4.17578125,  3.779296875,  -6.0703125,
            -9.90625,    -7.74609375, -8.2890625,  8.0859375
          ],
          'descriptor': {shape: [2, 2, 2, 3], dataType: 'float16'}
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.45703125,  9.1171875,
            3.544921875, 2.455078125,
            0,           -0.369384765625,
            5.5234375,   0,
            0,           6.421875,
            0,           0.701171875,
            0,           -0.323974609375,
            0,           3.185546875,
            0,           4.17578125,
            3.779296875, 0,
            0,           0,
            0,           8.0859375
          ],
          'descriptor': {shape: [2, 2, 2, 3], dataType: 'float16'}
        }
      }
    }
  },
  {
    'name': 'hardSwish float16 5D tensor',
    'graph': {
      'inputs': {
        'hardSwishInput': {
          'data': [
            0.734375,    9.1171875,   3.544921875, 2.62109375,   -6.4453125,
            -1.68359375, 5.5234375,   -5.95703125, -9.171875,    6.421875,
            -3.29296875, 1.041015625, -7.24609375, -0.947265625, -5.77734375,
            3.185546875, -7.26171875, 4.17578125,  3.779296875,  -6.0703125,
            -9.90625,    -7.74609375, -8.2890625,  8.0859375
          ],
          'descriptor': {shape: [2, 1, 4, 1, 3], dataType: 'float16'}
        }
      },
      'operators': [{
        'name': 'hardSwish',
        'arguments': [{'input': 'hardSwishInput'}],
        'outputs': 'hardSwishOutput'
      }],
      'expectedOutputs': {
        'hardSwishOutput': {
          'data': [
            0.45703125,  9.1171875,
            3.544921875, 2.455078125,
            0,           -0.369384765625,
            5.5234375,   0,
            0,           6.421875,
            0,           0.701171875,
            0,           -0.323974609375,
            0,           3.185546875,
            0,           4.17578125,
            3.779296875, 0,
            0,           0,
            0,           8.0859375
          ],
          'descriptor': {shape: [2, 1, 4, 1, 3], dataType: 'float16'}
        }
      }
    }
  }
];

webnn_conformance_test(
    hardSwishTests, buildAndExecuteGraph, getPrecisionTolerance);