Searched refs:reshaped_slice_dims (Results 1 – 1 of 1) sorted by relevance
849 std::vector<int64> reshaped_slice_dims; in PartitionBaseCase() local853 reshaped_slice_dims.push_back(dim_size * 2); in PartitionBaseCase()855 reshaped_slice_dims.push_back(dim_size); in PartitionBaseCase()861 reshaped_slice_dims), in PartitionBaseCase()