Searched refs:unary_post_op_algorithm (Results 1 – 4 of 4) sorted by relevance
320 const c10::string_view& unary_post_op_algorithm) { in create_attr_by_post_op() argument335 unary_post_op_algorithm == "none" || unary_post_op_algorithm == "tanh", in create_attr_by_post_op()336 …nednn qlinear: algorithm for post op gelu must be none or tanh but got ", unary_post_op_algorithm); in create_attr_by_post_op()337 auto post_algorithm = unary_post_op_algorithm == "none" ? in create_attr_by_post_op()
931 c10::string_view& unary_post_op_algorithm) { in linear_int8_with_onednn_weight() argument1043 unary_post_op_algorithm in linear_int8_with_onednn_weight()1272 c10::string_view unary_post_op_algorithm) { in run_pointwise_binary() argument1280 unary_post_op, unary_post_op_args, unary_post_op_algorithm in run_pointwise_binary()1304 c10::string_view unary_post_op_algorithm) { in run_pointwise_binary_tensor() argument1314 unary_post_op, unary_post_op_args, unary_post_op_algorithm in run_pointwise_binary_tensor()
1707 unary_post_op_algorithm, argument1746 unary_post_op_algorithm,
991 unary_post_op_algorithm=unary_algorithmm,