Function _tensor_nbytes

utils/quant.py:419–422

(tensor: torch.Tensor | None)

def _tensor_nbytes(tensor: torch.Tensor | None) -> int:
    if tensor is None:
        return 0
    return tensor.numel() * tensor.element_size()


def _materialize_transformer_engine_weights_for_inference(