Searched refs:ExpandView (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 49 ExpandView,
291 args[i] = ExpandView.create(args[i], list(args[indices[0]].get_size()))
436 ex = next(x for x in inputs if isinstance(x, (TensorBox, ExpandView, ir.Constant)))
441 ExpandView.create(
447 ExpandView.create(
803 args[i] = ExpandView.create(args[i], list(args[indices[0]].get_size()))
928 return ExpandView.create(x, tuple(sizes))
945 return TensorBox(ExpandView.create(x.data, tuple(sizes)))
5177 denom = ExpandView.create(denom, list(sum_result.get_size()))
5198 denom = ExpandView.create(denom, list(sum_result.get_size()))
|
| D | ir.py | 150 torch._inductor.ir.ExpandView,
1941 class ExpandView(BaseView): class
1989 return ExpandView(x, new_size)
|