Index

B

balance (torchgpipe.GPipe attribute)

balance_by_size() (in module torchgpipe.balance) balance_by_time() (in module torchgpipe.balance)

C

checkpoint (torchgpipe.GPipe attribute)

chunks (torchgpipe.GPipe attribute)

D

devices (torchgpipe.GPipe attribute)

F

forward() (torchgpipe.GPipe method)

G

GPipe (class in torchgpipe)

I

is_checkpointing() (in module torchgpipe)

is_recomputing() (in module torchgpipe) isolate() (torchgpipe.skip.skippable.Skippable method)

M

module torchgpipe torchgpipe.balance torchgpipe.skip

N

Namespace (class in torchgpipe.skip)

P

pop() (in module torchgpipe.skip)

S

skippable() (in module torchgpipe.skip)

stash() (in module torchgpipe.skip)

T

torchgpipe module torchgpipe.balance module

torchgpipe.skip module

V

verify_skippables() (in module torchgpipe.skip)