Searched refs:compressed_output_shape (Results 1 – 1 of 1) sorted by relevance
441 size_t compressed_output_shape[XNN_MAX_TENSOR_DIMS]; in setup_binary_elementwise_nd() local445 compressed_output_shape[i] = 1; in setup_binary_elementwise_nd()466 compressed_output_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd()474 compressed_output_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()483 compressed_output_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()501 compressed_output_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()510 compressed_output_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd()519 .elements = compressed_output_shape[0] << log2_element_size, in setup_binary_elementwise_nd()541 …e = compressed_a_shape[0], b_stride = compressed_b_shape[0], y_stride = compressed_output_shape[0]; in setup_binary_elementwise_nd()552 y_stride *= compressed_output_shape[i]; in setup_binary_elementwise_nd()[all …]