Searched refs:reduce_instruction (Results 1 – 2 of 2) sorted by relevance
78 HloInstruction* reduce_instruction = in XLA_TEST_P() local80 ASSERT_EQ(reduce_instruction->opcode(), HloOpcode::kReduce); in XLA_TEST_P()84 Shape* reduce_output_shape = reduce_instruction->mutable_shape(); in XLA_TEST_P()89 reduce_instruction->mutable_operand(0)->mutable_shape(); in XLA_TEST_P()
2537 HloInstruction* reduce_instruction = b.AddInstruction( in TEST_F() local2543 Literal result = hlo_eval.Evaluate(reduce_instruction).ConsumeValueOrDie(); in TEST_F()2572 HloInstruction* reduce_instruction = b.AddInstruction( in BM_ReducePrecisely() local2580 hlo_eval.Evaluate(reduce_instruction).ConsumeValueOrDie(); in BM_ReducePrecisely()