Lines Matching refs:set_dim
103 output_shape.set_dim(0, output_rows);236 output_shape.set_dim(0, 0);276 output_shape.set_dim(0, output_rows);676 output_shape.set_dim(0, output_rows);1029 output_shape.set_dim(0, M);