Searched refs:MakeGatherDimNumbers (Results 1 – 5 of 5) sorted by relevance
1780 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()1795 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()1810 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()1826 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()1843 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()1861 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()1879 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()1897 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()1912 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()1927 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()[all …]
1370 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()1405 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()
1379 static GatherDimensionNumbers MakeGatherDimNumbers(
2340 /* static */ GatherDimensionNumbers HloGatherInstruction::MakeGatherDimNumbers( in MakeGatherDimNumbers() function in xla::HloGatherInstruction
1637 HloGatherInstruction::MakeGatherDimNumbers( in ParseInstructionRhs()