conf.es6 735 字节
Newer Older
W
wangqun 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48
/* eslint-disable */
/**
 * @file batchnorm的配置文件
 * @author chenhaoze
 */
export default {
    dep: [
		{
			func: 'getValueFromTensorPos',
			conf: {
				TENSOR_NAME: 'origin'
			}
		},
		{
			func: 'transferFromNHWCtoNCHW',
			conf:{

			}
		}
    ],
    conf: [
		'WIDTH_SHAPE_ORIGIN',
		'HEIGHT_SHAPE_ORIGIN',
		'LENGTH_SHAPE_ORIGIN',
		'WIDTH_TEXTURE_ORIGIN',
		'HEIGHT_TEXTURE_ORIGIN',
		'CHANNEL_ORIGIN',

		'WIDTH_SHAPE_OUT',
		'HEIGHT_SHAPE_OUT',
		'WIDTH_TEXTURE_OUT',
		'HEIGHT_TEXTURE_OUT',
		'CHANNEL_OUT',
		'OFFSET_Y_OUT',

		'MULTI_VALUE',
		'BIAS_VALUE',
		'ACTIVE_FUNCTION'
    ],
    input: [
			{
				tensor: 'origin',
				variable: 'texture',
				setter: 'initTexture',
				type: 'texture'
			}
    ]
};