find_hidden_param_op_relations_from_minfo

find_hidden_param_op_relations_from_minfo#

class braintrace.find_hidden_param_op_relations_from_minfo(minfo, hid_path_to_group)[source]#

Find ETP relations from a ModuleInfo.

Builds a mapping from ALL brainstate.ParamState invars so that plain ParamState weights used with ETP primitives are recognised.

Return type:

Sequence[HiddenParamOpRelation]