Searched defs:makeBatched (Results 1 – 3 of 3) sorted by relevance
25 Tensor makeBatched(const Tensor& tensor, std::optional<int64_t> bdim, int64_t level) { in makeBatched() function
174 Tensor makeBatched(const Tensor& tensor, int64_t bdim, int64_t level) { in makeBatched() function
116 Tensor makeBatched(const Tensor& tensor, BatchDims bdims) { in makeBatched() function