Searched refs:MakeScatterDimNumbers (Results 1 – 5 of 5) sorted by relevance
2214 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2228 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2242 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2256 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2269 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2286 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2304 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2323 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2343 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2358 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()[all …]
1416 static ScatterDimensionNumbers MakeScatterDimNumbers(
1455 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()
2429 HloScatterInstruction::MakeScatterDimNumbers( in MakeScatterDimNumbers() function in xla::HloScatterInstruction
1673 HloScatterInstruction::MakeScatterDimNumbers( in ParseInstructionRhs()