Searched refs:compressed_output_shape (Results 1 – 1 of 1) sorted by relevance
186 size_t compressed_output_shape[XNN_MAX_TENSOR_DIMS]; in setup_binary_elementwise_nd_f32() local190 compressed_output_shape[i] = 1; in setup_binary_elementwise_nd_f32()211 compressed_output_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd_f32()219 compressed_output_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd_f32()228 compressed_output_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd_f32()244 compressed_output_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd_f32()253 compressed_output_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd_f32()262 .elements = compressed_output_shape[0] * sizeof(float), in setup_binary_elementwise_nd_f32()278 …e = compressed_a_shape[0], b_stride = compressed_b_shape[0], y_stride = compressed_output_shape[0]; in setup_binary_elementwise_nd_f32()289 y_stride *= compressed_output_shape[i]; in setup_binary_elementwise_nd_f32()[all …]