| { |
| "op": "ai.onnx.BatchNormalization", |
| "cases": [ |
| { |
| "name": "nchw_4x64x112x112_vec4", |
| "preset": "smoke", |
| "vars": { "dtype": "float32", "batch": 4, "channels": 64, "height": 112, "width": 112 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [4, 64, 112, 112], "dist": "normal", "seed": 770, "scale": 0.5 }, |
| "scale": { "dtype": "float32", "shape": [64], "dist": "uniform", "seed": 771, "scale": 0.25, "offset": 1 }, |
| "b": { "dtype": "float32", "shape": [64], "dist": "normal", "seed": 772, "scale": 0.1 }, |
| "inputMean": { "dtype": "float32", "shape": [64], "dist": "normal", "seed": 773, "scale": 0.1 }, |
| "inputVar": { "dtype": "float32", "shape": [64], "data": { "kind": "constant", "value": 1.0 } } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [4, 64, 112, 112] } }, |
| "bench": { |
| "primary": true, |
| "metrics": [ |
| { |
| "type": "bandwidth", |
| "value": "args.batch * args.channels * args.height * args.width * dtypeBytes(args.dtype) * 2 + args.channels * dtypeBytes(args.dtype) * 4" |
| } |
| ] |
| } |
| }, |
| { |
| "name": "nchw_1x64x112x112", |
| "vars": { "dtype": "float32", "batch": 1, "channels": 64, "height": 112, "width": 112 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [1, 64, 112, 112] }, |
| "scale": { "dtype": "float32", "shape": [64] }, |
| "b": { "dtype": "float32", "shape": [64] }, |
| "inputMean": { "dtype": "float32", "shape": [64] }, |
| "inputVar": { "dtype": "float32", "shape": [64], "data": { "kind": "constant", "value": 1.0 } } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [1, 64, 112, 112] } }, |
| "bench": { |
| "primary": true, |
| "metrics": [ |
| { |
| "type": "bandwidth", |
| "value": "args.batch * args.channels * args.height * args.width * dtypeBytes(args.dtype) * 2 + args.channels * dtypeBytes(args.dtype) * 4" |
| } |
| ] |
| } |
| }, |
| { |
| "name": "nc_4096x768", |
| "vars": { "dtype": "float32", "batch": 4096, "channels": 768, "height": 1, "width": 1 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [4096, 768] }, |
| "scale": { "dtype": "float32", "shape": [768] }, |
| "b": { "dtype": "float32", "shape": [768] }, |
| "inputMean": { "dtype": "float32", "shape": [768] }, |
| "inputVar": { "dtype": "float32", "shape": [768], "dist": "constant", "value": 1 } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [4096, 768] } }, |
| "bench": { |
| "metrics": [ |
| { |
| "type": "bandwidth", |
| "value": "args.batch * args.channels * dtypeBytes(args.dtype) * 2 + args.channels * dtypeBytes(args.dtype) * 4" |
| } |
| ] |
| } |
| }, |
| { |
| "name": "nchw_inference_scalar_no_vec4_1x64x63x63", |
| "preset": "edge", |
| "vars": { "dtype": "float32", "batch": 1, "channels": 64, "height": 63, "width": 63 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [1, 64, 63, 63], "dist": "normal", "seed": 6301, "scale": 1 }, |
| "scale": { "dtype": "float32", "shape": [64], "dist": "uniform", "seed": 6302, "scale": 0.25, "offset": 1 }, |
| "b": { "dtype": "float32", "shape": [64], "dist": "normal", "seed": 6303, "scale": 0.1 }, |
| "inputMean": { "dtype": "float32", "shape": [64], "dist": "normal", "seed": 6304, "scale": 0.1 }, |
| "inputVar": { "dtype": "float32", "shape": [64], "dist": "uniform", "seed": 6305, "scale": 0.5, "offset": 1 } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [1, 64, 63, 63] } }, |
| "bench": { |
| "metrics": [ |
| { |
| "type": "bandwidth", |
| "value": "args.batch * args.channels * args.height * args.width * dtypeBytes(args.dtype) * 2 + args.channels * dtypeBytes(args.dtype) * 4" |
| } |
| ] |
| } |
| } |
| ] |
| } |
|
|
