def get_primitive_impl(self, p: core.Primitive) -> Tuple[(Callable, bool)]:
    try:
        return (tf_impl[p], False)
    except KeyError:
        try:
            return (tf_impl_with_avals[p], True)
        except KeyError as err:
            msg = "TensorFlow interpretation rule for '{}' not implemented"
            raise NotImplementedError(msg.format(p)) from err
def get_primitive_impl(self, p: core.Primitive) -> Tuple[(Callable, bool)]:
    try:
        return (tf_impl[p], False)
    except KeyError:
        try:
            return (tf_impl_with_avals[p], True)
        except KeyError as err:
            msg = "TensorFlow interpretation rule for '{}' not implemented"
            raise NotImplementedError(msg.format(p)) from err