stgraph.compiler.op package

Subpackages

• stgraph.compiler.op.agg package
  • Submodules
  • stgraph.compiler.op.agg.agg_op module
  • Module contents
• stgraph.compiler.op.mxnet package
  • Module contents
• stgraph.compiler.op.pytorch package
  • Submodules
  • stgraph.compiler.op.pytorch.torch_op module
    • TorchOp
      • TorchOp.to_schema()
  • Module contents
• stgraph.compiler.op.tensorflow package
  • Module contents

Submodules

stgraph.compiler.op.op module

class stgraph.compiler.op.op.Op(op, fprog)

Bases: ABC

abstract to_schema()

translate backend-specific ops to uniform schema

stgraph.compiler.op.op_factory module

class stgraph.compiler.op.op_factory.OpFactory

Bases: object

create(op, backend, fprog)

Module contents