Home
last modified time | relevance | path

Searched refs:float_only_operator_set (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/
Dtensorexpr_fuser.cpp878 static const OperatorSet float_only_operator_set{ in typesAreSupported() local
942 if (node->isMemberOf(float_only_operator_set) && !isFloatingType(*st)) { in typesAreSupported()
950 } else if (node->isMemberOf(float_only_operator_set)) { in typesAreSupported()