Searched refs:handled_functions (Results 1 – 2 of 2) sorted by relevance
110 handled_functions = HANDLED_FUNCTIONS variable in ExpandedWeight142 if func in cls.handled_functions:143 return cls.handled_functions[func].apply(
129 handled_functions = HANDLED_FUNCTIONS_DIAGONAL variable in DiagonalTensor148 if func not in cls.handled_functions:150 return cls.handled_functions[func](*args, **kwargs)269 handled_functions = HANDLED_FUNCTIONS_SUB_DIAGONAL variable in SubDiagonalTensor