Searched refs:compressed_input2_shape (Results 1 – 1 of 1) sorted by relevance
440 size_t compressed_input2_shape[XNN_MAX_TENSOR_DIMS]; in setup_binary_elementwise_nd() local444 compressed_input2_shape[i] = 1; in setup_binary_elementwise_nd()465 compressed_input2_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd()482 compressed_input2_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()509 compressed_input2_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd()526 const size_t* compressed_b_shape = compressed_input2_shape; in setup_binary_elementwise_nd()531 compressed_a_shape = compressed_input2_shape; in setup_binary_elementwise_nd()536 } else if (compressed_input2_shape[0] == 1) { in setup_binary_elementwise_nd()538 } else if (compressed_input1_shape[0] == compressed_input2_shape[0]) { in setup_binary_elementwise_nd()