Searched refs:data_tile_assignment_dims (Results 1 – 1 of 1) sorted by relevance
679 std::vector<int64> data_tile_assignment_dims; in ScatterDataSharding() local682 data_tile_assignment_dims.push_back(1); in ScatterDataSharding()684 data_tile_assignment_dims.push_back( in ScatterDataSharding()690 data_tile_assignment_dims.push_back( in ScatterDataSharding()695 Product(data_tile_assignment_dims)) { in ScatterDataSharding()698 new_tile_assignment.Reshape(data_tile_assignment_dims); in ScatterDataSharding()