Searched refs:ndims_input (Results 1 – 1 of 1) sorted by relevance
166 auto ndims_input = input_shape.dims(); in ExpandInputDimsToOutputShape() local168 auto dim_offset = ndims_output - ndims_input; in ExpandInputDimsToOutputShape()173 for (int dim_idx = 0; dim_idx < ndims_input; ++dim_idx) in ExpandInputDimsToOutputShape()