TaskRegistry holds all necessary mappings from operators to tasks/arguments/tensors, and tasks to functions. It is responsible for argument compilation as well.
LocalTrainingBacking is responsible for executing a computation graph. It obtains arguments and tensors from the TaskRegistryRelated Issues:
Description of changes:
Implement
LocalTrainingBacking
andTaskRegistry
:TaskRegistry
holds all necessary mappings from operators to tasks/arguments/tensors, and tasks to functions. It is responsible for argument compilation as well.LocalTrainingBacking
is responsible for executing a computation graph. It obtains arguments and tensors from theTaskRegistry
Related Issues:Linked Issues:
Issues closed by this PR:
This change is