Searched refs:normalized_input_shape (Results 1 – 1 of 1) sorted by relevance
135 size_t normalized_input_shape[XNN_MAX_TENSOR_DIMS]; in setup_constant_pad_nd() local139 normalized_input_shape[i] = 1; in setup_constant_pad_nd()152 normalized_input_shape[XNN_MAX_TENSOR_DIMS - 1 - num_squeezed_dims] = input_dim; in setup_constant_pad_nd()163 normalized_input_shape[XNN_MAX_TENSOR_DIMS - num_squeezed_dims] *= input_dim; in setup_constant_pad_nd()178 … constant_pad_op->context.pad.input_size[i] = normalized_input_shape[XNN_MAX_TENSOR_DIMS - 1 - i]; in setup_constant_pad_nd()180 size_t input_stride = normalized_input_shape[XNN_MAX_TENSOR_DIMS - 1]; in setup_constant_pad_nd()187 input_stride *= normalized_input_shape[XNN_MAX_TENSOR_DIMS - 1 - i]; in setup_constant_pad_nd()