Searched refs:reduce_2 (Results 1 – 4 of 4) sorted by relevance
74 // CHECK-LABEL: func @reduce_275 func @reduce_2(%arg0: tensor<?x?xf32>, %arg1: tensor<?x?xf32>) -> (tensor<?x?xf32>, tensor<?xf32>) {
655 HloInstruction* reduce_2 = in TEST_F() local659 EXPECT_THAT(reduce_2, op::Reduce(broadcast_2, op::Constant())); in TEST_F()736 HloInstruction* reduce_2 = in TEST_F() local742 EXPECT_THAT(reduce_2, in TEST_F()
3664 def reduce_2(obj): function3666 dispatch_table[AAA] = reduce_2
3311 reduce_2(PyObject *obj) in reduce_2() function3512 return reduce_2(self); in _common_reduce()