Lines Matching refs:compressed_output_shape

441   size_t compressed_output_shape[XNN_MAX_TENSOR_DIMS];  in setup_binary_elementwise_nd()  local
445     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()
557   binary_elementwise_op->compute.range[0] = compressed_output_shape[5];  in setup_binary_elementwise_nd()
558   binary_elementwise_op->compute.range[1] = compressed_output_shape[4];  in setup_binary_elementwise_nd()
559   binary_elementwise_op->compute.range[2] = compressed_output_shape[3];  in setup_binary_elementwise_nd()
560   binary_elementwise_op->compute.range[3] = compressed_output_shape[2];  in setup_binary_elementwise_nd()
561   binary_elementwise_op->compute.range[4] = compressed_output_shape[1];  in setup_binary_elementwise_nd()