Convert some index case to embedding

[swimdi]

Ticket

#535
#544
#639

Problem description

Some input variation of index like below can be replaced as embedding

pytorch2.0_ttnn/tests/autogen_op/ALL/test_ALL_aten_index_Tensor.py

Lines 45 to 59 in 2174709

	["Tensor<[169, 8]> self = ?", "List[Optional[Tensor]] indices = [<[2401]>]"],
	["Tensor<[169, 16]> self = ?", "List[Optional[Tensor]] indices = [<[2401]>]"],
	["Tensor<[169, 32]> self = ?", "List[Optional[Tensor]] indices = [<[2401]>]"],
	["Tensor<[169, 3]> self = ?", "List[Optional[Tensor]] indices = [<[2401]>]"],
	["Tensor<[169, 6]> self = ?", "List[Optional[Tensor]] indices = [<[2401]>]"],
	["Tensor<[169, 12]> self = ?", "List[Optional[Tensor]] indices = [<[2401]>]"],
	["Tensor<[169, 24]> self = ?", "List[Optional[Tensor]] indices = [<[2401]>]"],
	["Tensor<[225, 4]> self = ?", "List[Optional[Tensor]] indices = [<[4096]>]"],
	["Tensor<[225, 8]> self = ?", "List[Optional[Tensor]] indices = [<[4096]>]"],
	["Tensor<[225, 16]> self = ?", "List[Optional[Tensor]] indices = [<[4096]>]"],
	["Tensor<[225, 32]> self = ?", "List[Optional[Tensor]] indices = [<[4096]>]"],
	["Tensor<[225, 3]> self = ?", "List[Optional[Tensor]] indices = [<[4096]>]"],
	["Tensor<[225, 6]> self = ?", "List[Optional[Tensor]] indices = [<[4096]>]"],
	["Tensor<[225, 12]> self = ?", "List[Optional[Tensor]] indices = [<[4096]>]"],
	["Tensor<[225, 24]> self = ?", "List[Optional[Tensor]] indices = [<[4096]>]"],

So add the conversion of it

What's changed

• Convert some case from index to embedding
• Convert aten.index_select.Tensor to aten.index.Tensor
• Add aten.index_select.Tensor and aten.index.Tensor to const fold pass

credit: @jerrysky3

[jerrysky3]

Maybe file a bug?

[swimdi]

ok, I've open an issue

[jerrysky3]

I'm thinking if we should raise an exception when running out of tries. Because that means the graph still can't not converge and probably in a bad state

[swimdi]

ok I added

[jerrysky3]

Do we have unit tests for index_select and unsafe_index?

[swimdi]

ok, I've add