Searched defs:addBatchDim (Results 1 – 2 of 2) sorted by relevance
130 Tensor addBatchDim(const Tensor& tensor, int64_t level, int64_t dim) { in addBatchDim() function
184 Tensor addBatchDim(const Tensor& tensor, int64_t dim, int64_t level) { in addBatchDim() function