Searched refs:binary_elementwise_op (Results 1 – 2 of 2) sorted by relevance
| /external/XNNPACK/src/operators/ |
| D | binary-elementwise-nd.c | 43 xnn_operator_t binary_elementwise_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in create_binary_elementwise_nd() local
44 if (binary_elementwise_op == NULL) { in create_binary_elementwise_nd()
52 memcpy(&binary_elementwise_op->params, params, params_size); in create_binary_elementwise_nd()
55 binary_elementwise_op->ukernel.vbinary.op_function = vbinary_fused_ukernels->op_ukernel; in create_binary_elementwise_nd()
56 binary_elementwise_op->ukernel.vbinary.opc_function = vbinary_fused_ukernels->opc_ukernel; in create_binary_elementwise_nd()
57 binary_elementwise_op->ukernel.vbinary.ropc_function = vbinary_fused_ukernels->ropc_ukernel; in create_binary_elementwise_nd()
59 binary_elementwise_op->type = operator_type; in create_binary_elementwise_nd()
61 binary_elementwise_op->state = xnn_run_state_invalid; in create_binary_elementwise_nd()
63 *binary_elementwise_op_out = binary_elementwise_op; in create_binary_elementwise_nd()
373 xnn_operator_t binary_elementwise_op, in setup_binary_elementwise_nd() argument
[all …]
|
| /external/XNNPACK/test/ |
| D | binary-elementwise-operator-tester.h | 279 xnn_operator_t binary_elementwise_op = nullptr; in TestQS8() local
288 0, &binary_elementwise_op); in TestQS8()
297 ASSERT_NE(nullptr, binary_elementwise_op); in TestQS8()
300 …rator, decltype(&xnn_delete_operator)> auto_binary_elementwise_op(binary_elementwise_op, xnn_delet… in TestQS8()
306 binary_elementwise_op, in TestQS8()
319 xnn_run_operator(binary_elementwise_op, nullptr /* thread pool */)); in TestQS8()
425 xnn_operator_t binary_elementwise_op = nullptr; in TestF16() local
429 status = xnn_create_add_nd_f16(output_min, output_max, 0, &binary_elementwise_op); in TestF16()
432 status = xnn_create_multiply_nd_f16(output_min, output_max, 0, &binary_elementwise_op); in TestF16()
441 ASSERT_NE(nullptr, binary_elementwise_op); in TestF16()
[all …]
|