Searched refs:compressed_b_shape (Results 1 – 1 of 1) sorted by relevance
526 const size_t* compressed_b_shape = compressed_input2_shape; in setup_binary_elementwise_nd() local532 compressed_b_shape = compressed_input1_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()546 if (compressed_b_shape[i] != 1) { in setup_binary_elementwise_nd()551 b_stride *= compressed_b_shape[i]; in setup_binary_elementwise_nd()