Searched refs:sourceSlicedSizes (Results 1 – 1 of 1) sorted by relevance
303 auto sourceSlicedSizes = source.sizes().vec(); in TORCH_PRECOMPUTE_META_FUNC() local304 if (!sourceSlicedSizes.empty()) { in TORCH_PRECOMPUTE_META_FUNC()305 sourceSlicedSizes.erase(sourceSlicedSizes.begin() + dim); in TORCH_PRECOMPUTE_META_FUNC()307 if (selfSlicedSizes.size() != sourceSlicedSizes.size() || in TORCH_PRECOMPUTE_META_FUNC()309 sourceSlicedSizes.begin())) { in TORCH_PRECOMPUTE_META_FUNC()313 ss << " and source slice shape: " << sourceSlicedSizes << " at dimension 0."; in TORCH_PRECOMPUTE_META_FUNC()