Searched refs:tensor_split_indices_batching_rule (Results 1 – 1 of 1) sorted by relevance
227 std::vector<Tensor> tensor_split_indices_batching_rule(const Tensor& self, IntArrayRef indices, int… in tensor_split_indices_batching_rule() function1104 m.impl("tensor_split.indices", tensor_split_indices_batching_rule); in TORCH_LIBRARY_IMPL()