Skip to content
This repository was archived by the owner on Aug 21, 2025. It is now read-only.

Commit ba34404

Browse files
author
Samantha Andow
authored
_nested_tensor -> nested_tensor (#637)
1 parent 78c2a6b commit ba34404

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

functorch/csrc/VmapGeneratedPlumbing.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25019,13 +25019,13 @@ ::std::vector<at::Tensor> unflatten_dense_tensors_generated_plumbing(const at::T
2501925019
return makeBatchedVector(std::get<0>(results), std::get<1>(results), cur_level);
2502025020
}
2502125021
template <typename batch_rule_t, batch_rule_t batch_rule>
25022-
at::Tensor _nested_tensor_generated_plumbing(at::TensorList list, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
25022+
at::Tensor nested_tensor_generated_plumbing(at::TensorList list, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory) {
2502325023
c10::impl::ExcludeDispatchKeyGuard guard(kBatchedKey);
2502425024
auto maybe_layer = maybeCurrentDynamicLayer();
2502525025
TORCH_INTERNAL_ASSERT(maybe_layer.has_value());
2502625026
int64_t cur_level = maybe_layer->layerId();
2502725027
if (!isBatchedAtLevel(list, cur_level)) {
25028-
return at::_ops::_nested_tensor::call(list, dtype, layout, device, pin_memory);
25028+
return at::_ops::nested_tensor::call(list, dtype, layout, device, pin_memory);
2502925029
}
2503025030

2503125031
auto results = batch_rule(list, dtype, layout, device, pin_memory);

0 commit comments

Comments
 (0)