feed.js 310 字节
Newer Older
Y
Yanzhan Yang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
module.exports.create = function (args, scope, gl) {
  const output = scope[args.outputs.Out[0]]
  const input = scope[args.inputs.X[0]]
  return {
    inferShape() {
      output.dim = output.dim.map(d => d === -1 ? 1 : d)
    },
    compute() {
      console.log(input)
      console.log(output)
    }
  }
}