Searched defs:batched_matrices (Results 1 – 2 of 2) sorted by relevance
112 inline int64_t batchCount(const Tensor& batched_matrices) { in batchCount()121 inline int64_t matrixStride(const Tensor& batched_matrices) { in matrixStride()
10 static inline int64_t batchCountTrilTriu(const Tensor& batched_matrices) { in batchCountTrilTriu()