Searched refs:MakeGatherDimNumbers (Results 1 – 6 of 6) sorted by relevance
2480 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()2495 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()2510 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()2526 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()2544 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()2561 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()2578 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()2595 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()2613 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()2631 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()[all …]
74 HloGatherInstruction::MakeGatherDimNumbers( in ExpandInstruction()
1598 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()1634 HloGatherInstruction::MakeGatherDimNumbers( in TEST_F()
2121 static GatherDimensionNumbers MakeGatherDimNumbers(
3079 /* static */ GatherDimensionNumbers HloGatherInstruction::MakeGatherDimNumbers( in MakeGatherDimNumbers() function in xla::HloGatherInstruction
2830 HloGatherInstruction::MakeGatherDimNumbers( in CreateInstruction()