Searched refs:CreateReduceWindow (Results 1 – 11 of 11) sorted by relevance
200 auto output = builder.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_F()
300 auto reduce_window = add(HloInstruction::CreateReduceWindow( in HandleBatchGroupCount()
1582 b.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_P()1634 b.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_P()1691 b.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_P()1752 b.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_P()
586 builder.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_F()
632 static std::unique_ptr<HloInstruction> CreateReduceWindow(
1413 builder.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_F()1456 builder.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_F()3620 builder.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_F()3706 builder.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_F()4265 builder.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_P()
172 builder.AddInstruction(HloInstruction::CreateReduceWindow( in TEST_F()
432 instruction = CreateReduceWindow(shape, operands(0), operands(1), in CreateFromProto()1054 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateReduceWindow( in CreateReduceWindow() function in xla::HloInstruction
3300 reduce_window, HloInstruction::CreateReduceWindow( in HandleReduceWindow()
1059 instruction = builder->AddInstruction(HloInstruction::CreateReduceWindow( in ParseInstructionRhs()
685 builder.AddInstruction(HloInstruction::CreateReduceWindow( in XLA_TEST_F()