Searched defs:_add_batch_dim (Results 1 – 2 of 2) sorted by relevance
14 Tensor _add_batch_dim(const Tensor& self, int64_t batch_dim, int64_t level) { in _add_batch_dim() function
38 Tensor _add_batch_dim(const Tensor& self, int64_t batch_dim, int64_t level) { in _add_batch_dim() function