gistfile1.txt

Attributes:
  op_kind: "Aten"
  node_kind "aten::foo(Tensor a, int b, Tensor[] c) -> Tensor"
  
  
class AtenOp {
  
  AtenOp() {
    Graph g;
    schema_ = SchemaParser(node_kind).parseDeclaration();
    for argument in schema.arguments:
      g.addInput(argument.type)
    Node * n = g.addNode(node_kind, g.inputs());
    
    operation = getOperatorForNode(n);
    
  }
  IValue convertToIValue(string, caffe2blob) { ... }
  IValue convertToCaffe2(string, IValue) { ... }
  void run() {
    Stack s;
    for arg in schema_.arguments {
      s.push_back(convertToIValue(arg.type, input(i));
    }
    operation(s);
    for ret in schema_.returns {
      Output(i) = convertToCaffe2(ret.type, stack[i]);
    }
  }
private:
  Operation operation;
  FunctionSchema schema_;
};