habitat.profiling.backward#

Module Contents#

Classes#

BackwardHelper

Functions#

backward_available(operation_output)

flatten_operation_output(operation_output)

get_grad_fn(retval)

get_accumulate_grad_inputs(initial_grad_fn, ...)

class habitat.profiling.backward.BackwardHelper(backward_runnable, ag_dict)[source]#
classmethod new_from(operation_outputs)[source]#
run_accumulate_grad()[source]#
habitat.profiling.backward.backward_available(operation_output)[source]#
habitat.profiling.backward.flatten_operation_output(operation_output)[source]#
habitat.profiling.backward.get_grad_fn(retval)[source]#
habitat.profiling.backward.get_accumulate_grad_inputs(initial_grad_fn, backward_runnable)[source]#