Searched refs:ndims_output (Results 1 – 1 of 1) sorted by relevance
167 auto ndims_output = output_shape.dims(); in ExpandInputDimsToOutputShape() local168 auto dim_offset = ndims_output - ndims_input; in ExpandInputDimsToOutputShape()171 reshaped_dims->resize(ndims_output, 1); in ExpandInputDimsToOutputShape()