stgraph.compiler.backend package

Subpackages

• stgraph.compiler.backend.mxnet package
  • Module contents
• stgraph.compiler.backend.pytorch package
  • Submodules
  • stgraph.compiler.backend.pytorch.torch_callback module
    • STGraphBackendTorch
      • STGraphBackendTorch.new_zeros_call_back()
      • STGraphBackendTorch.tensor_raw_ptr()
  • stgraph.compiler.backend.pytorch.torch_kernel_wrapper module
    • KernelWrapperTorch
  • Module contents
• stgraph.compiler.backend.tensorflow package
  • Module contents

Submodules

stgraph.compiler.backend.callback module

class stgraph.compiler.backend.callback.STGraphBackend

Bases: ABC

backend_cb(executor)

abstract new_zeros_call_back(size, dtype, device, requires_grad=True)

abstract tensor_raw_ptr(tensor)

stgraph.compiler.backend.kernel_wrapper module

class stgraph.compiler.backend.kernel_wrapper.KernelWrapper

Bases: object

static backward(ctx, *gradout)

static forward(executor, kid, kernel_args, rets, *args)

static setup_context(ctx, inputs, output)

Module contents