1 Star 0 Fork 16

Paddles / Paddle.js

forked from PaddlePaddle / Paddle.js 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
ops.ts 2.45 KB
一键复制 编辑 原始数据 按行查看 历史
export default {
relu: {
behaviors: ['transToPrelu']
},
relu6: {
behaviors: ['transToRelu6']
},
leaky_relu: {
behaviors: ['transToLeakyrelu']
},
transToLeakyrelu: {
behaviors: ['transToLeakyrelu']
},
scale: {
behaviors: ['transToScale']
},
sigmoid: {
behaviors: ['transToSigmoid']
},
hard_sigmoid: {
behaviors: ['transToHardSigmoid']
},
pow: {
behaviors: ['transToPow']
},
sqrt: {
behaviors: ['transToSqrt']
},
tanh: {
behaviors: ['transToTanh']
},
concat_mul: {
behaviors: ['normalizeDim']
},
concat: {
behaviors: ['normalizeDim']
},
conv2d_elementwise_add: {
behaviors: [
'mergeAttrs',
'checkIsMerge',
'setActiveFunc'
]
},
conv2d_transpose: {
behaviors: [
'adaptPaddings',
'isApplySeparableConv',
'batchComputeConv2d',
'processBias'
]
},
conv2d: {
behaviors: [
'adaptPaddings',
'isApplySeparableConv',
'batchComputeConv2d',
'processBias'
]
},
elementwise_add: {
behaviors: [
'processElementwiseAxis',
'genElementwiseCounterPos'
]
},
elementwise_div: {
behaviors: [
'processElementwiseAxis',
'genElementwiseCounterPos'
]
},
elementwise_mul: {
behaviors: [
'processElementwiseAxis',
'genElementwiseCounterPos'
]
},
elementwise_sub: {
behaviors: [
'processElementwiseAxis',
'genElementwiseCounterPos'
]
},
pool2d_max: {
behaviors: [
'isMax',
'setPacked',
'setAdaptive',
'isGlobalPooling'
]
},
poo2d_winograd: {
behaviors: [
'isMax',
'setPacked',
'isGlobalPooling'
]
},
pool2d: {
behaviors: [
'isMax',
'setPacked',
'setAdaptive'
]
},
reduce_mean: {
behaviors: ['normalizeDim']
},
reduce_sum: {
behaviors: ['normalizeDim']
},
split: {
behaviors: ['normalizeDim']
},
transpose2: {
behaviors: ['normalizeDim']
},
dropout: {
behaviors: ['normalizeDim']
}
};
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
TypeScript
1
https://gitee.com/paddles/Paddle.js.git
git@gitee.com:paddles/Paddle.js.git
paddles
Paddle.js
Paddle.js
release/v2.2.5

搜索帮助

344bd9b3 5694891 D2dac590 5694891