Searched refs:normalized_element_size (Results 1 – 3 of 3) sorted by relevance
129 size_t normalized_element_size = element_size; in xnn_normalize_transpose_permutation() local131 normalized_element_size = element_size * normalized_shape[output_dims - 1]; in xnn_normalize_transpose_permutation()147 normalized_input_stride[output_dims - 1] = normalized_element_size; in xnn_normalize_transpose_permutation()159 normalized_output_stride[output_dims - 1] = normalized_element_size; in xnn_normalize_transpose_permutation()169 *normalized_element_size_out = normalized_element_size; in xnn_normalize_transpose_permutation()
22 size_t* normalized_element_size,
191 size_t normalized_element_size; in setup_transpose_nd() local193 …&normalized_element_size, normalized_perm, normalized_shape, context->input_stride, context->outpu… in setup_transpose_nd()221 switch (normalized_element_size) { in setup_transpose_nd()241 context->element_size = normalized_element_size; in setup_transpose_nd()253 transpose_op->compute.range[0] = normalized_element_size; in setup_transpose_nd()