Searched refs:reduce_instruction (Results 1 – 2 of 2) sorted by relevance
80 HloInstruction* reduce_instruction = in XLA_TEST_P() local82 ASSERT_EQ(reduce_instruction->opcode(), HloOpcode::kReduce); in XLA_TEST_P()86 Shape* reduce_output_shape = reduce_instruction->mutable_shape(); in XLA_TEST_P()91 reduce_instruction->mutable_operand(0)->mutable_shape(); in XLA_TEST_P()
1451 HloInstruction* reduce_instruction = b.AddInstruction( in TEST_F() local1457 Literal result = hlo_eval.Evaluate(reduce_instruction).ConsumeValueOrDie(); in TEST_F()1487 HloInstruction* reduce_instruction = b.AddInstruction( in BM_ReducePrecisely() local1494 hlo_eval.Evaluate(reduce_instruction).ConsumeValueOrDie(); in BM_ReducePrecisely()