Searched refs:a_batch_stride (Results 1 – 3 of 3) sorted by relevance
38 int64 batch_size, int64 m, int64 n, int64 a_batch_stride, in TriangularSolveThunk() argument53 a_batch_stride_(a_batch_stride), in TriangularSolveThunk()
46 int64 a_batch_stride, int64 b_batch_stride);
1643 int64 a_batch_stride = in EmitTriangularSolveFromMlir() local1655 m, n, a_batch_stride, b_batch_stride)); in EmitTriangularSolveFromMlir()