Searched refs:split_batching_rule (Results 1 – 2 of 2) sorted by relevance
237 std::vector<Tensor> split_batching_rule(const Tensor& self, int64_t split_size, int64_t dim) { in split_batching_rule() function737 m.impl("split.Tensor", split_batching_rule); in TORCH_LIBRARY_IMPL()
432 std::vector<Tensor> split_batching_rule(const Tensor& self, int64_t split_size, int64_t dim) { in split_batching_rule() function1123 m.impl("split.Tensor", split_batching_rule); in TORCH_LIBRARY_IMPL()