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