Home
last modified time | relevance | path

Searched defs:tensor_split (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset13.py139 def tensor_split( function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp995 std::vector<Tensor> tensor_split(const Tensor& self, IntArrayRef indices, int64_t dim) { in tensor_split() function
1003 std::vector<Tensor> tensor_split(const Tensor& self, const Tensor& tensor_indices_or_sections, int6… in tensor_split() function
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py4106 def tensor_split( function