stgraph.compiler.backend.pytorch package

Submodules

stgraph.compiler.backend.pytorch.torch_callback module

class stgraph.compiler.backend.pytorch.torch_callback.STGraphBackendTorch[source]

Bases: STGraphBackend

STGraph backend using PyTorch framework

new_zeros_call_back(size, dtype, device, requires_grad=True)[source]
tensor_raw_ptr(tensor)[source]

stgraph.compiler.backend.pytorch.torch_kernel_wrapper module

class stgraph.compiler.backend.pytorch.torch_kernel_wrapper.KernelWrapperTorch[source]

Bases: KernelWrapper, Function

Module contents