Home
last modified time | relevance | path

Searched refs:tensor_split_sections_symint (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesDecompositions.cpp295 m.impl("tensor_split.sections", native::tensor_split_sections_symint); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dnative_ops.cpp877 pnode->Output(0) = at::native::tensor_split_sections_symint(a, b, c); in __anon75e5f0515402()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp959 std::vector<Tensor> tensor_split_sections_symint(const Tensor& self, c10::SymInt sym_sections, int6… in tensor_split_sections_symint() function