Searched refs:normalized_output_shape (Results 1 – 1 of 1) sorted by relevance
136 size_t normalized_output_shape[XNN_MAX_TENSOR_DIMS]; in setup_constant_pad_nd() local140 normalized_output_shape[i] = 1; in setup_constant_pad_nd()153 …normalized_output_shape[XNN_MAX_TENSOR_DIMS - 1 - num_squeezed_dims] = pre_padding + input_dim + p… in setup_constant_pad_nd()164 normalized_output_shape[XNN_MAX_TENSOR_DIMS - num_squeezed_dims] *= input_dim; in setup_constant_pad_nd()181 size_t output_stride = normalized_output_shape[XNN_MAX_TENSOR_DIMS - 1]; in setup_constant_pad_nd()188 output_stride *= normalized_output_shape[XNN_MAX_TENSOR_DIMS - 1 - i]; in setup_constant_pad_nd()191 …constant_pad_op->context.pad.output_size[0] = normalized_output_shape[XNN_MAX_TENSOR_DIMS - 1] << … in setup_constant_pad_nd()198 constant_pad_op->compute.range[0] = normalized_output_shape[0]; in setup_constant_pad_nd()199 constant_pad_op->compute.range[1] = normalized_output_shape[1]; in setup_constant_pad_nd()200 constant_pad_op->compute.range[2] = normalized_output_shape[2]; in setup_constant_pad_nd()[all …]