Source code
Revision control
Copy as Markdown
Other Tools
{
"tests": [
{
"name": "maxPool2d float32 4D constant tensor default options",
"inputs": {
"input": {
"shape": [1, 2, 5, 5],
"data": [
89.00830216793153,
-45.720390400208274,
-61.3061304134967,
-4.014514560596496,
-94.54893342858352,
46.28090328619564,
99.28312923953135,
-10.057873368956962,
9.742474583994337,
-39.03501766575275,
75.08192382950091,
12.819415189421207,
-33.01505690327188,
38.691340789603316,
66.09259104681504,
97.903480409299,
-8.737769993284147,
-53.4216238072017,
72.10852084777076,
-40.42309116918719,
-35.68864442661396,
-87.64779401381033,
38.87424286093716,
39.38360329656629,
7.429088108317444,
-76.72171237621149,
50.21706410294061,
-52.89547724835329,
-44.64233565670091,
-97.86752535352848,
81.73119248706783,
5.428491238061412,
-29.227728678237995,
72.44899280781078,
-59.34124718116585,
39.19959912145927,
-65.99438957588619,
-4.204323589500888,
-60.5458643662661,
55.89052583821697,
80.30483906634527,
72.88830243502153,
-46.59610987974246,
20.503878887964206,
-31.126462826158445,
-57.29456052682171,
-26.62357805164706,
15.93575469730375,
-78.77953474824318,
72.33577555559427
],
"type": "float32",
"constant": true
}
},
"expected": {
"name": "output",
"shape": [1, 2, 1, 1],
"data": [
99.28312683105469,
81.73119354248047
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor default options",
"inputs": {
"input": {
"shape": [1, 2, 5, 5],
"data": [
89.00830216793153,
-45.720390400208274,
-61.3061304134967,
-4.014514560596496,
-94.54893342858352,
46.28090328619564,
99.28312923953135,
-10.057873368956962,
9.742474583994337,
-39.03501766575275,
75.08192382950091,
12.819415189421207,
-33.01505690327188,
38.691340789603316,
66.09259104681504,
97.903480409299,
-8.737769993284147,
-53.4216238072017,
72.10852084777076,
-40.42309116918719,
-35.68864442661396,
-87.64779401381033,
38.87424286093716,
39.38360329656629,
7.429088108317444,
-76.72171237621149,
50.21706410294061,
-52.89547724835329,
-44.64233565670091,
-97.86752535352848,
81.73119248706783,
5.428491238061412,
-29.227728678237995,
72.44899280781078,
-59.34124718116585,
39.19959912145927,
-65.99438957588619,
-4.204323589500888,
-60.5458643662661,
55.89052583821697,
80.30483906634527,
72.88830243502153,
-46.59610987974246,
20.503878887964206,
-31.126462826158445,
-57.29456052682171,
-26.62357805164706,
15.93575469730375,
-78.77953474824318,
72.33577555559427
],
"type": "float32"
}
},
"expected": {
"name": "output",
"shape": [1, 2, 1, 1],
"data": [
99.28312683105469,
81.73119354248047
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor options.windowDimensions",
"inputs":{
"input": {
"shape": [1, 2, 5, 5],
"data": [
89.00830216793153,
-45.720390400208274,
-61.3061304134967,
-4.014514560596496,
-94.54893342858352,
46.28090328619564,
99.28312923953135,
-10.057873368956962,
9.742474583994337,
-39.03501766575275,
75.08192382950091,
12.819415189421207,
-33.01505690327188,
38.691340789603316,
66.09259104681504,
97.903480409299,
-8.737769993284147,
-53.4216238072017,
72.10852084777076,
-40.42309116918719,
-35.68864442661396,
-87.64779401381033,
38.87424286093716,
39.38360329656629,
7.429088108317444,
-76.72171237621149,
50.21706410294061,
-52.89547724835329,
-44.64233565670091,
-97.86752535352848,
81.73119248706783,
5.428491238061412,
-29.227728678237995,
72.44899280781078,
-59.34124718116585,
39.19959912145927,
-65.99438957588619,
-4.204323589500888,
-60.5458643662661,
55.89052583821697,
80.30483906634527,
72.88830243502153,
-46.59610987974246,
20.503878887964206,
-31.126462826158445,
-57.29456052682171,
-26.62357805164706,
15.93575469730375,
-78.77953474824318,
72.33577555559427
],
"type": "float32"
}
},
"options": {
"windowDimensions": [3, 3]
},
"expected": {
"name": "output",
"shape": [1, 2, 3, 3],
"data": [
99.28312683105469,
99.28312683105469,
66.09259033203125,
99.28312683105469,
99.28312683105469,
72.1085205078125,
97.90348052978516,
72.1085205078125,
72.1085205078125,
81.73119354248047,
72.44898986816406,
72.44898986816406,
81.73119354248047,
72.8883056640625,
72.44898986816406,
80.30484008789062,
72.8883056640625,
72.33577728271484
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor options.padding",
"inputs": {
"input": {
"shape": [1, 2, 5, 5],
"data": [
89.00830216793153,
-45.720390400208274,
-61.3061304134967,
-4.014514560596496,
-94.54893342858352,
46.28090328619564,
99.28312923953135,
-10.057873368956962,
9.742474583994337,
-39.03501766575275,
75.08192382950091,
12.819415189421207,
-33.01505690327188,
38.691340789603316,
66.09259104681504,
97.903480409299,
-8.737769993284147,
-53.4216238072017,
72.10852084777076,
-40.42309116918719,
-35.68864442661396,
-87.64779401381033,
38.87424286093716,
39.38360329656629,
7.429088108317444,
-76.72171237621149,
50.21706410294061,
-52.89547724835329,
-44.64233565670091,
-97.86752535352848,
81.73119248706783,
5.428491238061412,
-29.227728678237995,
72.44899280781078,
-59.34124718116585,
39.19959912145927,
-65.99438957588619,
-4.204323589500888,
-60.5458643662661,
55.89052583821697,
80.30483906634527,
72.88830243502153,
-46.59610987974246,
20.503878887964206,
-31.126462826158445,
-57.29456052682171,
-26.62357805164706,
15.93575469730375,
-78.77953474824318,
72.33577555559427
],
"type": "float32"
}
},
"options": {
"padding": [1, 0, 0, 1]
},
"expected": {
"name": "output",
"shape": [1, 2, 2, 2],
"data": [
99.28312683105469,
99.28312683105469,
99.28312683105469,
99.28312683105469,
81.73119354248047,
72.8883056640625,
81.73119354248047,
72.8883056640625
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor options.strides",
"inputs": {
"input": {
"shape": [1, 2, 5, 5],
"data": [
89.00830216793153,
-45.720390400208274,
-61.3061304134967,
-4.014514560596496,
-94.54893342858352,
46.28090328619564,
99.28312923953135,
-10.057873368956962,
9.742474583994337,
-39.03501766575275,
75.08192382950091,
12.819415189421207,
-33.01505690327188,
38.691340789603316,
66.09259104681504,
97.903480409299,
-8.737769993284147,
-53.4216238072017,
72.10852084777076,
-40.42309116918719,
-35.68864442661396,
-87.64779401381033,
38.87424286093716,
39.38360329656629,
7.429088108317444,
-76.72171237621149,
50.21706410294061,
-52.89547724835329,
-44.64233565670091,
-97.86752535352848,
81.73119248706783,
5.428491238061412,
-29.227728678237995,
72.44899280781078,
-59.34124718116585,
39.19959912145927,
-65.99438957588619,
-4.204323589500888,
-60.5458643662661,
55.89052583821697,
80.30483906634527,
72.88830243502153,
-46.59610987974246,
20.503878887964206,
-31.126462826158445,
-57.29456052682171,
-26.62357805164706,
15.93575469730375,
-78.77953474824318,
72.33577555559427
],
"type": "float32"
}
},
"options": {
"windowDimensions": [3, 3],
"strides": [2, 2]
},
"expected": {
"name": "output",
"shape": [1, 2, 2, 2],
"data": [
99.28312683105469,
66.09259033203125,
97.90348052978516,
72.1085205078125,
81.73119354248047,
72.44898986816406,
80.30484008789062,
72.33577728271484
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor options.dilations",
"inputs": {
"input": {
"shape": [1, 2, 5, 5],
"data": [
89.00830216793153,
-45.720390400208274,
-61.3061304134967,
-4.014514560596496,
-94.54893342858352,
46.28090328619564,
99.28312923953135,
-10.057873368956962,
9.742474583994337,
-39.03501766575275,
75.08192382950091,
12.819415189421207,
-33.01505690327188,
38.691340789603316,
66.09259104681504,
97.903480409299,
-8.737769993284147,
-53.4216238072017,
72.10852084777076,
-40.42309116918719,
-35.68864442661396,
-87.64779401381033,
38.87424286093716,
39.38360329656629,
7.429088108317444,
-76.72171237621149,
50.21706410294061,
-52.89547724835329,
-44.64233565670091,
-97.86752535352848,
81.73119248706783,
5.428491238061412,
-29.227728678237995,
72.44899280781078,
-59.34124718116585,
39.19959912145927,
-65.99438957588619,
-4.204323589500888,
-60.5458643662661,
55.89052583821697,
80.30483906634527,
72.88830243502153,
-46.59610987974246,
20.503878887964206,
-31.126462826158445,
-57.29456052682171,
-26.62357805164706,
15.93575469730375,
-78.77953474824318,
72.33577555559427
],
"type": "float32"
}
},
"options": {
"windowDimensions": [3, 3],
"dilations": [2, 2]
},
"expected": {
"name": "output",
"shape": [1, 2, 1, 1],
"data": [
89.00830078125,
72.33577728271484
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor options.layout=nchw",
"inputs": {
"input": {
"shape": [1, 2, 5, 5],
"data": [
89.00830216793153,
-45.720390400208274,
-61.3061304134967,
-4.014514560596496,
-94.54893342858352,
46.28090328619564,
99.28312923953135,
-10.057873368956962,
9.742474583994337,
-39.03501766575275,
75.08192382950091,
12.819415189421207,
-33.01505690327188,
38.691340789603316,
66.09259104681504,
97.903480409299,
-8.737769993284147,
-53.4216238072017,
72.10852084777076,
-40.42309116918719,
-35.68864442661396,
-87.64779401381033,
38.87424286093716,
39.38360329656629,
7.429088108317444,
-76.72171237621149,
50.21706410294061,
-52.89547724835329,
-44.64233565670091,
-97.86752535352848,
81.73119248706783,
5.428491238061412,
-29.227728678237995,
72.44899280781078,
-59.34124718116585,
39.19959912145927,
-65.99438957588619,
-4.204323589500888,
-60.5458643662661,
55.89052583821697,
80.30483906634527,
72.88830243502153,
-46.59610987974246,
20.503878887964206,
-31.126462826158445,
-57.29456052682171,
-26.62357805164706,
15.93575469730375,
-78.77953474824318,
72.33577555559427
],
"type": "float32"
}
},
"options": {
"layout": "nchw"
},
"expected": {
"name": "output",
"shape": [1, 2, 1, 1],
"data": [
99.28312683105469,
81.73119354248047
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor options.layout=nhwc",
"inputs": {
"input": {
"shape": [1, 5, 5, 2],
"data": [
89.00830216793153,
-76.72171237621149,
-45.720390400208274,
50.21706410294061,
-61.3061304134967,
-52.89547724835329,
-4.014514560596496,
-44.64233565670091,
-94.54893342858352,
-97.86752535352848,
46.28090328619564,
81.73119248706783,
99.28312923953135,
5.428491238061412,
-10.057873368956962,
-29.227728678237995,
9.742474583994337,
72.44899280781078,
-39.03501766575275,
-59.34124718116585,
75.08192382950091,
39.19959912145927,
12.819415189421207,
-65.99438957588619,
-33.01505690327188,
-4.204323589500888,
38.691340789603316,
-60.5458643662661,
66.09259104681504,
55.89052583821697,
97.903480409299,
80.30483906634527,
-8.737769993284147,
72.88830243502153,
-53.4216238072017,
-46.59610987974246,
72.10852084777076,
20.503878887964206,
-40.42309116918719,
-31.126462826158445,
-35.68864442661396,
-57.29456052682171,
-87.64779401381033,
-26.62357805164706,
38.87424286093716,
15.93575469730375,
39.38360329656629,
-78.77953474824318,
7.429088108317444,
72.33577555559427
],
"type": "float32"
}
},
"options": {
"layout": "nhwc"
},
"expected": {
"name": "output",
"shape": [1, 1, 1, 2],
"data": [
99.28312683105469,
81.73119354248047
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor options.roundingType=floor",
"inputs": {
"input": {
"shape": [1, 2, 5, 5],
"data": [
89.00830216793153,
-45.720390400208274,
-61.3061304134967,
-4.014514560596496,
-94.54893342858352,
46.28090328619564,
99.28312923953135,
-10.057873368956962,
9.742474583994337,
-39.03501766575275,
75.08192382950091,
12.819415189421207,
-33.01505690327188,
38.691340789603316,
66.09259104681504,
97.903480409299,
-8.737769993284147,
-53.4216238072017,
72.10852084777076,
-40.42309116918719,
-35.68864442661396,
-87.64779401381033,
38.87424286093716,
39.38360329656629,
7.429088108317444,
-76.72171237621149,
50.21706410294061,
-52.89547724835329,
-44.64233565670091,
-97.86752535352848,
81.73119248706783,
5.428491238061412,
-29.227728678237995,
72.44899280781078,
-59.34124718116585,
39.19959912145927,
-65.99438957588619,
-4.204323589500888,
-60.5458643662661,
55.89052583821697,
80.30483906634527,
72.88830243502153,
-46.59610987974246,
20.503878887964206,
-31.126462826158445,
-57.29456052682171,
-26.62357805164706,
15.93575469730375,
-78.77953474824318,
72.33577555559427
],
"type": "float32"
}
},
"options": {
"windowDimensions": [3, 3],
"padding": [1, 0, 0, 1],
"strides": [2, 2],
"roundingType": "floor"
},
"expected": {
"name": "output",
"shape": [1, 2, 2, 2],
"data": [
99.28312683105469,
9.742474555969238,
99.28312683105469,
72.1085205078125,
81.73119354248047,
72.44898986816406,
81.73119354248047,
72.44898986816406
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor options.roundingType=ceil",
"inputs": {
"input": {
"shape": [1, 2, 5, 5],
"data": [
89.00830216793153,
-45.720390400208274,
-61.3061304134967,
-4.014514560596496,
-94.54893342858352,
46.28090328619564,
99.28312923953135,
-10.057873368956962,
9.742474583994337,
-39.03501766575275,
75.08192382950091,
12.819415189421207,
-33.01505690327188,
38.691340789603316,
66.09259104681504,
97.903480409299,
-8.737769993284147,
-53.4216238072017,
72.10852084777076,
-40.42309116918719,
-35.68864442661396,
-87.64779401381033,
38.87424286093716,
39.38360329656629,
7.429088108317444,
-76.72171237621149,
50.21706410294061,
-52.89547724835329,
-44.64233565670091,
-97.86752535352848,
81.73119248706783,
5.428491238061412,
-29.227728678237995,
72.44899280781078,
-59.34124718116585,
39.19959912145927,
-65.99438957588619,
-4.204323589500888,
-60.5458643662661,
55.89052583821697,
80.30483906634527,
72.88830243502153,
-46.59610987974246,
20.503878887964206,
-31.126462826158445,
-57.29456052682171,
-26.62357805164706,
15.93575469730375,
-78.77953474824318,
72.33577555559427
],
"type": "float32"
}
},
"options": {
"windowDimensions": [3, 3],
"padding": [1, 0, 0, 1],
"strides": [2, 2],
"roundingType": "ceil"
},
"expected": {
"name": "output",
"shape": [1, 2, 3, 3],
"data": [
99.28312683105469,
9.742474555969238,
-39.03501892089844,
99.28312683105469,
72.1085205078125,
66.09259033203125,
97.90348052978516,
72.1085205078125,
7.429088115692139,
81.73119354248047,
72.44898986816406,
-59.34124755859375,
81.73119354248047,
72.44898986816406,
55.890525817871094,
80.30484008789062,
72.33577728271484,
72.33577728271484
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor options.outputSizes ignores options.roundingType=floor",
"inputs": {
"input": {
"shape": [1, 2, 5, 5],
"data": [
89.00830216793153,
-45.720390400208274,
-61.3061304134967,
-4.014514560596496,
-94.54893342858352,
46.28090328619564,
99.28312923953135,
-10.057873368956962,
9.742474583994337,
-39.03501766575275,
75.08192382950091,
12.819415189421207,
-33.01505690327188,
38.691340789603316,
66.09259104681504,
97.903480409299,
-8.737769993284147,
-53.4216238072017,
72.10852084777076,
-40.42309116918719,
-35.68864442661396,
-87.64779401381033,
38.87424286093716,
39.38360329656629,
7.429088108317444,
-76.72171237621149,
50.21706410294061,
-52.89547724835329,
-44.64233565670091,
-97.86752535352848,
81.73119248706783,
5.428491238061412,
-29.227728678237995,
72.44899280781078,
-59.34124718116585,
39.19959912145927,
-65.99438957588619,
-4.204323589500888,
-60.5458643662661,
55.89052583821697,
80.30483906634527,
72.88830243502153,
-46.59610987974246,
20.503878887964206,
-31.126462826158445,
-57.29456052682171,
-26.62357805164706,
15.93575469730375,
-78.77953474824318,
72.33577555559427
],
"type": "float32"
}
},
"options": {
"windowDimensions": [3, 3],
"padding": [1, 0, 0, 1],
"strides": [2, 2],
"roundingType": "floor",
"outputSizes": [3, 3]
},
"expected": {
"name": "output",
"shape": [1, 2, 3, 3],
"data": [
99.28312683105469,
9.742474555969238,
-39.03501892089844,
99.28312683105469,
72.1085205078125,
66.09259033203125,
97.90348052978516,
72.1085205078125,
7.429088115692139,
81.73119354248047,
72.44898986816406,
-59.34124755859375,
81.73119354248047,
72.44898986816406,
55.890525817871094,
80.30484008789062,
72.33577728271484,
72.33577728271484
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor options.outputSizes ignores options.roundingType=ceil",
"inputs": {
"input": {
"shape": [1, 2, 5, 5],
"data": [
89.00830216793153,
-45.720390400208274,
-61.3061304134967,
-4.014514560596496,
-94.54893342858352,
46.28090328619564,
99.28312923953135,
-10.057873368956962,
9.742474583994337,
-39.03501766575275,
75.08192382950091,
12.819415189421207,
-33.01505690327188,
38.691340789603316,
66.09259104681504,
97.903480409299,
-8.737769993284147,
-53.4216238072017,
72.10852084777076,
-40.42309116918719,
-35.68864442661396,
-87.64779401381033,
38.87424286093716,
39.38360329656629,
7.429088108317444,
-76.72171237621149,
50.21706410294061,
-52.89547724835329,
-44.64233565670091,
-97.86752535352848,
81.73119248706783,
5.428491238061412,
-29.227728678237995,
72.44899280781078,
-59.34124718116585,
39.19959912145927,
-65.99438957588619,
-4.204323589500888,
-60.5458643662661,
55.89052583821697,
80.30483906634527,
72.88830243502153,
-46.59610987974246,
20.503878887964206,
-31.126462826158445,
-57.29456052682171,
-26.62357805164706,
15.93575469730375,
-78.77953474824318,
72.33577555559427
],
"type": "float32"
}
},
"options": {
"windowDimensions": [3, 3],
"padding": [1, 0, 0, 1],
"strides": [2, 2],
"roundingType": "ceil",
"outputSizes": [2, 2]
},
"expected": {
"name": "output",
"shape": [1, 2, 2, 2],
"data": [
99.28312683105469,
9.742474555969238,
99.28312683105469,
72.1085205078125,
81.73119354248047,
72.44898986816406,
81.73119354248047,
72.44898986816406
],
"type": "float32"
}
},
{
"name": "maxPool2d float32 4D tensor options.dilations with options.strides",
"inputs": {
"input": {
"shape": [1, 7, 7, 2],
"data": [
34.69258129387845,
-24.706248375920524,
-60.42807114468629,
35.93883484387953,
60.896094267313345,
79.42220031805164,
-77.85906306951374,
54.18820945858141,
-21.57693529351286,
-49.103906741859404,
78.55176346486067,
74.28213308844582,
-87.92769416001408,
79.82047109868057,
11.680922626564922,
-12.1545053403525,
-22.332939699432202,
33.08486279636597,
8.640676946303174,
47.04064404360258,
95.78234680258441,
-88.01998403139729,
-98.53630176965567,
16.15897660557981,
21.396089910647248,
95.13234999362083,
-40.80724330720389,
-88.70921764059032,
-40.772770487731826,
67.89843024034857,
-50.337466344983504,
-96.56610664792846,
12.508098405268967,
-6.63581156175988,
-44.80198313101762,
80.27474506768564,
-65.68411481136147,
-14.884790604208746,
-66.54857566488332,
20.07233794529273,
-27.954270006840204,
-56.62217270552601,
82.74794838911092,
93.30175304845017,
-34.07339352229894,
-22.871643145139146,
73.25525716488457,
41.14021562026471,
-19.755147411323023,
80.07700931110838,
-69.89277244982563,
14.013250274581338,
-61.36640253626604,
51.53046561516868,
43.538863140698226,
-89.5888690731458,
51.45121768091764,
73.92397095355011,
-80.25264684029128,
94.72747432067723,
95.25411370348488,
58.12575321550594,
19.885722826536778,
-70.030129951855,
63.41951918063978,
-54.11785116480967,
-97.2280762988258,
-60.65826309587585,
-31.049980820398645,
-14.64655330832754,
-63.73688919275588,
47.34630581820389,
85.5640530081308,
-53.38925321464192,
-70.84739324778467,
47.35504709117748,
83.38397624574233,
7.36169510433227,
46.85823546522715,
98.13465616102368,
-43.93968566738362,
50.33780282391024,
37.455634062049,
-54.52760852219014,
-6.212307719031031,
34.41734148282714,
11.816772365254153,
72.44517799372989,
86.34609553228327,
4.146569170394486,
88.33040249135547,
98.29994529007328,
-66.72379613555366,
58.06434425465281,
-51.168278021154954,
-17.768582927864657,
9.961172472871425,
-52.738431655100946
],
"type": "float32"
}
},
"options": {
"windowDimensions": [3, 3],
"padding": [1, 0, 0, 1],
"strides": [2, 2],
"dilations": [1, 1],
"layout": "nhwc"
},
"expected": {
"name": "output",
"shape": [1, 3, 3, 2],
"data": [
60.896095275878906,
79.42220306396484,
95.7823486328125,
79.42220306396484,
78.55176544189453,
95.1323471069336,
82.7479476928711,
93.30175018310547,
95.7823486328125,
80.27474212646484,
43.53886413574219,
95.1323471069336,
95.25411224365234,
94.72747802734375,
95.25411224365234,
98.13465881347656,
63.419517517089844,
98.13465881347656
],
"type": "float32"
}
}
]
}