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¶