Searched refs:elementwise_binary (Results 1 – 2 of 2) sorted by relevance
515 binary_elementwise_op->context.elementwise_binary = (struct elementwise_binary_context) { in setup_binary_elementwise_nd()522 memcpy(&binary_elementwise_op->context.elementwise_binary.params, params, params_size); in setup_binary_elementwise_nd()528 …binary_elementwise_op->context.elementwise_binary.ukernel = binary_elementwise_op->ukernel.vbinary… in setup_binary_elementwise_nd()529 binary_elementwise_op->context.elementwise_binary.a = input2; in setup_binary_elementwise_nd()530 binary_elementwise_op->context.elementwise_binary.b = input1; in setup_binary_elementwise_nd()534 …memcpy(&binary_elementwise_op->context.elementwise_binary.params, reversed_params, reversed_params… in setup_binary_elementwise_nd()537 …binary_elementwise_op->context.elementwise_binary.ukernel = binary_elementwise_op->ukernel.vbinary… in setup_binary_elementwise_nd()539 …binary_elementwise_op->context.elementwise_binary.ukernel = binary_elementwise_op->ukernel.vbinary… in setup_binary_elementwise_nd()544 …binary_elementwise_op->context.elementwise_binary.a_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = a_stride… in setup_binary_elementwise_nd()547 …binary_elementwise_op->context.elementwise_binary.b_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = b_stride… in setup_binary_elementwise_nd()[all …]
316 struct elementwise_binary_context elementwise_binary; member