Searched refs:elementwise_binary (Results 1 – 2 of 2) sorted by relevance
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()916 …binary_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()925 …binary_elementwise_op->context.elementwise_binary.ukernel = binary_elementwise_op->ukernel.vbinary… in setup_binary_elementwise_nd()927 …binary_elementwise_op->context.elementwise_binary.ukernel = binary_elementwise_op->ukernel.vbinary… in setup_binary_elementwise_nd()932 …binary_elementwise_op->context.elementwise_binary.a_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = a_stride… in setup_binary_elementwise_nd()935 …binary_elementwise_op->context.elementwise_binary.b_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = b_stride… in setup_binary_elementwise_nd()[all …]
375 struct elementwise_binary_context elementwise_binary; member