stgraph.compiler.val package

Subpackages

• stgraph.compiler.val.mxnet package
  • Module contents
• stgraph.compiler.val.pytorch package
  • Submodules
  • stgraph.compiler.val.pytorch.torch_val module
    • TorchVal
      • TorchVal.backend
      • TorchVal.backend_key
      • TorchVal.device
      • TorchVal.dtype
      • TorchVal.layout
      • TorchVal.requires_grad
      • TorchVal.size
      • TorchVal.sum()
      • TorchVal.val_type
      • TorchVal.view()
  • Module contents
• stgraph.compiler.val.tensorflow package
  • Module contents

Submodules

stgraph.compiler.val.val module

class stgraph.compiler.val.val.Val(tensor, id, fprog)

Bases: ABC

abstract backend()

Return backend system

abstract backend_key()

Return key of backend system

abstract device()

Return which device is on

abstract dtype()

Return the DType

property id

abstract layout()

Return the layout

abstract requires_grad()

Return whether requires gradient

abstract size()

Return the size(i.e. shape)

abstract sum()

property v

abstract val_type()

Return ValueType

abstract view(*args, **kargs)

stgraph.compiler.val.val_factory module

class stgraph.compiler.val.val_factory.ValFactory

Bases: object

create(type: ValType, tensor, backend, id, fprog, reduce_dim)

get_val_backend(backend)

Module contents