Home
last modified time | relevance | path

Searched refs:op_kind_t (Results 1 – 8 of 8) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/
H A Dir_core.cpp132 case op_kind_t::_ge: in is_cmp_op()
133 case op_kind_t::_gt: in is_cmp_op()
142 op_kind_t negate_cmp_op(op_kind_t op_kind) { in negate_cmp_op()
144 case op_kind_t::_ge: return op_kind_t::_le; in negate_cmp_op()
145 case op_kind_t::_gt: return op_kind_t::_lt; in negate_cmp_op()
146 case op_kind_t::_le: return op_kind_t::_ge; in negate_cmp_op()
147 case op_kind_t::_lt: return op_kind_t::_gt; in negate_cmp_op()
148 case op_kind_t::_eq: return op_kind_t::_eq; in negate_cmp_op()
149 case op_kind_t::_ne: return op_kind_t::_ne; in negate_cmp_op()
219 if (utils::one_of(op_kind, op_kind_t::_shl, op_kind_t::_shr)) { in binary_op_type()
[all …]
H A Dir_simplify.cpp451 && utils::one_of(op.op_kind, op_kind_t::_eq, op_kind_t::_ne)) in reduce_lhs_rhs()
1073 binary_op->op_kind, op_kind_t::_div, op_kind_t::_mod)) in _mutate()
1274 if (!utils::one_of(obj.op_kind, op_kind_t::_div, op_kind_t::_mod)) in _mutate()
1460 static op_kind_t flip_cmp_op(op_kind_t op_kind) { in flip_cmp_op()
1462 case op_kind_t::_eq: return op_kind_t::_ne; in flip_cmp_op()
1463 case op_kind_t::_ge: return op_kind_t::_lt; in flip_cmp_op()
1464 case op_kind_t::_gt: return op_kind_t::_le; in flip_cmp_op()
1465 case op_kind_t::_le: return op_kind_t::_gt; in flip_cmp_op()
1466 case op_kind_t::_lt: return op_kind_t::_ge; in flip_cmp_op()
1467 case op_kind_t::_ne: return op_kind_t::_eq; in flip_cmp_op()
[all …]
H A Dpost_op_support.hpp121 float rhs_scale, op_kind_t op_kind) in post_op_t()
138 op_kind_t op_kind() const { return op_kind_; } in op_kind()
189 op_kind_t op_kind_ = op_kind_t::undef;
193 inline op_kind_t alg_kind_to_op_kind(alg_kind_t alg) { in alg_kind_to_op_kind()
201 case alg_kind::binary_ge: return op_kind_t::_ge; in alg_kind_to_op_kind()
202 case alg_kind::binary_gt: return op_kind_t::_gt; in alg_kind_to_op_kind()
203 case alg_kind::binary_le: return op_kind_t::_le; in alg_kind_to_op_kind()
204 case alg_kind::binary_lt: return op_kind_t::_lt; in alg_kind_to_op_kind()
205 case alg_kind::binary_eq: return op_kind_t::_eq; in alg_kind_to_op_kind()
206 case alg_kind::binary_ne: return op_kind_t::_ne; in alg_kind_to_op_kind()
[all …]
H A Dir.cpp54 if (utils::one_of(obj.op_kind, op_kind_t::_min, op_kind_t::_max)) { in _visit()
534 case op_kind_t::_gt: in implies()
538 case op_kind_t::_lt: in implies()
560 case op_kind_t::_lt: in normalize()
564 case op_kind_t::_gt: in normalize()
711 if (!utils::one_of(binary_op->op_kind, op_kind_t::_add, op_kind_t::_sub)) in is_linear_var_transform()
796 case op_kind_t::_eq: in is_single_value()
800 case op_kind_t::_ge: in is_single_value()
801 case op_kind_t::_gt: { in is_single_value()
807 case op_kind_t::_le: in is_single_value()
[all …]
H A Dir_core.hpp789 enum class op_kind_t { enum
816 std::string to_string(op_kind_t kind);
823 bool is_cmp_op(op_kind_t op_kind);
825 op_kind_t negate_cmp_op(op_kind_t op_kind);
865 op_kind_t op_kind;
1136 op_kind_t op_kind;
1171 expr_t &base, expr_t &off, op_kind_t op_kind = op_kind_t::_add);
1319 op_kind_t op_kind;
1362 op_kind_t op_kind;
1479 DECLARE_BINARY_OPERATOR(>, op_kind_t::_gt)
[all …]
H A Dir.hpp413 expr_t make_nary_op(op_kind_t op_kind, const std::vector<expr_t> &args);
589 op_kind_t op_kind() const { return expr_.as<binary_op_t>().op_kind; } in op_kind()
758 case op_kind_t::_ge: in try_prove_compound_relation()
759 case op_kind_t::_le: break; in try_prove_compound_relation()
760 case op_kind_t::_gt: in try_prove_compound_relation()
761 op_kind = op_kind_t::_ge; in try_prove_compound_relation()
765 case op_kind_t::_lt: in try_prove_compound_relation()
766 op_kind = op_kind_t::_le; in try_prove_compound_relation()
774 if (op_kind == op_kind_t::_ge) { in try_prove_compound_relation()
780 if (op_kind == op_kind_t::_le) { in try_prove_compound_relation()
H A Dconv_kernel.hpp1279 case op_kind_t::_and: { in _visit()
1457 case op_kind_t::_add3: in _visit()
1458 case op_kind_t::_mad: { in _visit()
1464 if (obj.op_kind == op_kind_t::_add3) { in _visit()
1477 ir_assert(obj.op_kind == op_kind_t::_minus); in _visit()
1562 case op_kind_t::_ge: in ebinary()
1563 case op_kind_t::_gt: in ebinary()
1564 case op_kind_t::_le: in ebinary()
1565 case op_kind_t::_lt: in ebinary()
1566 case op_kind_t::_eq: in ebinary()
[all …]
H A Dkernel_builder.cpp1239 return nary && (nary->op_kind == op_kind_t::_add); in __anon70af794b0502()
1270 make_nary_op(op_kind_t::_add, invariant_args));
1289 cur_expr = make_nary_op(op_kind_t::_add, other_args);
3273 case op_kind_t::_add3: { in _mutate()
3284 case op_kind_t::_mad: { in _mutate()
3850 rhs = binary_op_t::make(op_kind_t::_mul, rhs, in build_tile_stmt()