Searched refs:num_compressed_dims (Results 1 – 1 of 1) sorted by relevance
438 size_t num_compressed_dims = 0; in setup_binary_elementwise_nd() local463 num_compressed_dims++; in setup_binary_elementwise_nd()465 compressed_input2_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd()466 compressed_output_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd()471 num_compressed_dims++; in setup_binary_elementwise_nd()473 compressed_input1_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()474 compressed_output_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()479 num_compressed_dims++; in setup_binary_elementwise_nd()481 compressed_input1_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()482 compressed_input2_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()[all …]