Searched refs:compressed_a_shape (Results 1 – 1 of 1) sorted by relevance
525 const size_t* compressed_a_shape = compressed_input1_shape; in setup_binary_elementwise_nd() local531 compressed_a_shape = compressed_input2_shape; in setup_binary_elementwise_nd()541 …size_t a_stride = compressed_a_shape[0], b_stride = compressed_b_shape[0], y_stride = compressed_o… in setup_binary_elementwise_nd()543 if (compressed_a_shape[i] != 1) { in setup_binary_elementwise_nd()550 a_stride *= compressed_a_shape[i]; in setup_binary_elementwise_nd()