revive.common package

Submodules

revive.common.next_ts_policy_function module

revive.common.next_ts_policy_function.next_ts_placeholder_policy_function(data: Dict[str, Tensor]) → Tensor

revive.common.next_ts_transition_function module

revive.common.next_ts_transition_function.next_ts_placeholder_transition_function(data: Dict[str, Tensor]) → Tensor

revive.common.step_node_function module

revive.common.step_node_function.get_next_step_node(data: Dict[str, Tensor]) → Tensor

Module contents