Searched defs:ignore_index (Results 1 – 1 of 1) sorted by relevance
467 std::vector<int64> ignore_index; in ReshapeToTileDimension() local473 buckets(ignore_index)[index[dim]].push_back(device); in ReshapeToTileDimension() local