class_docstr: Stores inputs to the `memory_efficient_attention` operators
code: def device(self) -> torch.device:
      return self.query.device
  
imports: import math
  from dataclasses import dataclass
  from functools import partial
  from typing import Any, Callable, List, Mapping, Optional, Set, Tuple, Type, Union
  import torch
  from ..._cpp_lib import _built_with_cuda
  from ..common import BaseOperator
  from .attn_bias import AttentionBias, BlockDiagonalMask, LowerTriangularMask, LowerTriangularMaskWithTensorBias
label_desc: Use image node_id 1 for calling the Inputs obj's underlying member method code with example usage: obj.device() and returns: self