Home
last modified time | relevance | path

Searched defs:result_tensors (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorFactories.cpp180 std::vector<at::Tensor> result_tensors(ntensors); in NestedTensor_unbind() local
/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Ddim.cpp2836 …auto result_tensors = self_info.tensor->split_with_sizes(at::IntArrayRef(indices.begin(), indices.… in py_split() local