**Operators:** `torch.ops.aten.amax.default` `torch.ops.aten.scalar_tensor.default` `torch.ops.aten.baddbmm.default` `torch.ops.aten.exp.default` `torch.ops.aten.empty.memory_format` `torch.ops.aten.all.dim`