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