Searched refs:get_reduce_op (Results 1 – 2 of 2) sorted by relevance
28 def get_reduce_op(reduction_str): function83 return get_reduce_op(self.reduction)(inputs, axis=self.axis)95 return get_reduce_op(self.reduction)(weighted_inputs, axis=self.axis)
7252 ReduceOp get_reduce_op(nir_op op, unsigned bit_size) in get_reduce_op() function7409 ReduceOp reduce_op = get_reduce_op(op, instr->src[0].ssa->bit_size); in emit_uniform_scan()8079 ReduceOp reduce_op = get_reduce_op(op, bit_size); in visit_intrinsic()