Searched refs:is_stacked (Results 1 – 1 of 1) sorted by relevance
811 if not inp.is_stacked:833 if not x.is_stacked:843 if inp.is_stacked:863 t, is_stacked, _ = self.input(index)864 if not is_stacked:877 t, is_stacked, _ = self.input(index)878 if is_stacked:1055 def wrap(tensor, is_stacked=True, is_sparse_stacked=False): argument1057 assert isinstance(is_stacked, bool)1060 assert not is_sparse_stacked or is_stacked, ("If the wrapped tensor is "[all …]