Searched defs:maybe_batch_dim (Results 1 – 1 of 1) sorted by relevance
12 Tensor moveBatchDimToFront(const Tensor& tensor, std::optional<int64_t> maybe_batch_dim) { in moveBatchDimToFront()22 int64_t rankWithoutBatchDim(const Tensor& tensor, std::optional<int64_t> maybe_batch_dim) { in rankWithoutBatchDim()30 int64_t numelWithoutBatchDim(const Tensor& tensor, std::optional<int64_t> maybe_batch_dim) { in numelWithoutBatchDim()