Searched refs:binop_reduce (Results 1 – 2 of 2) sorted by relevance
| /external/mesa3d/src/compiler/nir/ |
| D | nir_opcodes.py | 576 def binop_reduce(name, output_size, output_type, src_type, prereduce_expr, function
604 binop_reduce(name, output_size, tbool1, src_type,
606 binop_reduce("b8" + name[1:], output_size, tbool8, src_type,
608 binop_reduce("b16" + name[1:], output_size, tbool16, src_type,
610 binop_reduce("b32" + name[1:], output_size, tbool32, src_type,
846 binop_reduce("fall_equal", 1, tfloat32, tfloat32, "{src0} == {src1}",
848 binop_reduce("fany_nequal", 1, tfloat32, tfloat32, "{src0} != {src1}",
907 binop_reduce("fdot", 1, tfloat, tfloat, "{src0} * {src1}", "{src0} + {src1}",
910 binop_reduce("fdot", 0, tfloat, tfloat,
|
| /external/pytorch/torch/testing/_internal/distributed/ |
| D | multi_threaded_pg.py | 48 def binop_reduce(tensors, op): function
59 ReduceOp.SUM: partial(binop_reduce, op=torch.sum),
60 ReduceOp.AVG: partial(binop_reduce, op=torch.mean),
61 ReduceOp.PRODUCT: partial(binop_reduce, op=torch.prod),
62 ReduceOp.MIN: partial(binop_reduce, op=torch.min),
63 ReduceOp.MAX: partial(binop_reduce, op=torch.max),
|