• Home
  • Raw
  • Download

Lines Matching refs:binary_elementwise_op

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()
60 binary_elementwise_op->flags = flags; in create_binary_elementwise_nd()
62 binary_elementwise_op->state = xnn_run_state_invalid; in create_binary_elementwise_nd()
64 *binary_elementwise_op_out = binary_elementwise_op; in create_binary_elementwise_nd()
768 xnn_operator_t binary_elementwise_op, in setup_binary_elementwise_nd() argument
786 binary_elementwise_op->state = xnn_run_state_invalid; in setup_binary_elementwise_nd()
790 xnn_operator_type_to_string(binary_elementwise_op->type)); in setup_binary_elementwise_nd()
796 xnn_operator_type_to_string(binary_elementwise_op->type)); in setup_binary_elementwise_nd()
800 if (binary_elementwise_op->type != expected_operator_type) { in setup_binary_elementwise_nd()
803 xnn_operator_type_to_string(binary_elementwise_op->type)); in setup_binary_elementwise_nd()
811 …xnn_operator_type_to_string(binary_elementwise_op->type), num_input1_dims, num_input2_dims, XNN_MA… in setup_binary_elementwise_nd()
868 xnn_operator_type_to_string(binary_elementwise_op->type), in setup_binary_elementwise_nd()
899 binary_elementwise_op->state = xnn_run_state_skip; in setup_binary_elementwise_nd()
903 binary_elementwise_op->context.elementwise_binary = (struct elementwise_binary_context) { in setup_binary_elementwise_nd()
910 memcpy(&binary_elementwise_op->context.elementwise_binary.params, params, params_size); in setup_binary_elementwise_nd()
916binary_elementwise_op->context.elementwise_binary.ukernel = binary_elementwise_op->ukernel.vbinary… in setup_binary_elementwise_nd()
917 binary_elementwise_op->context.elementwise_binary.a = input2; in setup_binary_elementwise_nd()
918 binary_elementwise_op->context.elementwise_binary.b = input1; in setup_binary_elementwise_nd()
922 …memcpy(&binary_elementwise_op->context.elementwise_binary.params, reversed_params, reversed_params… in setup_binary_elementwise_nd()
925binary_elementwise_op->context.elementwise_binary.ukernel = binary_elementwise_op->ukernel.vbinary… in setup_binary_elementwise_nd()
927binary_elementwise_op->context.elementwise_binary.ukernel = binary_elementwise_op->ukernel.vbinary… in setup_binary_elementwise_nd()
932binary_elementwise_op->context.elementwise_binary.a_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = a_stride… in setup_binary_elementwise_nd()
935binary_elementwise_op->context.elementwise_binary.b_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = b_stride… in setup_binary_elementwise_nd()
937binary_elementwise_op->context.elementwise_binary.y_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = y_stride… in setup_binary_elementwise_nd()
943 binary_elementwise_op->compute.type = xnn_parallelization_type_5d; in setup_binary_elementwise_nd()
944binary_elementwise_op->compute.task_5d = (pthreadpool_task_5d_t) xnn_compute_elementwise_binary_5d; in setup_binary_elementwise_nd()
945 binary_elementwise_op->compute.range[0] = compressed_output_shape[5]; in setup_binary_elementwise_nd()
946 binary_elementwise_op->compute.range[1] = compressed_output_shape[4]; in setup_binary_elementwise_nd()
947 binary_elementwise_op->compute.range[2] = compressed_output_shape[3]; in setup_binary_elementwise_nd()
948 binary_elementwise_op->compute.range[3] = compressed_output_shape[2]; in setup_binary_elementwise_nd()
949 binary_elementwise_op->compute.range[4] = compressed_output_shape[1]; in setup_binary_elementwise_nd()
950 binary_elementwise_op->compute.tile[0] = 1; in setup_binary_elementwise_nd()
951 binary_elementwise_op->compute.tile[1] = 1; in setup_binary_elementwise_nd()
952 binary_elementwise_op->state = xnn_run_state_ready; in setup_binary_elementwise_nd()
958 xnn_operator_t binary_elementwise_op, in setup_binary_elementwise_nd_f16() argument
971 binary_elementwise_op, in setup_binary_elementwise_nd_f16()
982 &binary_elementwise_op->params.f16_minmax, sizeof(binary_elementwise_op->params.f16_minmax), in setup_binary_elementwise_nd_f16()
983 &binary_elementwise_op->params.f16_minmax, sizeof(binary_elementwise_op->params.f16_minmax), in setup_binary_elementwise_nd_f16()
989 xnn_operator_t binary_elementwise_op, in setup_binary_elementwise_nd_f32() argument
1002 binary_elementwise_op, expected_operator_type, in setup_binary_elementwise_nd_f32()
1008 &binary_elementwise_op->params.f32_minmax, sizeof(binary_elementwise_op->params.f32_minmax), in setup_binary_elementwise_nd_f32()
1009 &binary_elementwise_op->params.f32_minmax, sizeof(binary_elementwise_op->params.f32_minmax), in setup_binary_elementwise_nd_f32()