Searched refs:b_batch_stride (Results 1 – 3 of 3) sorted by relevance
39 int64 b_batch_stride) in TriangularSolveThunk() argument54 b_batch_stride_(b_batch_stride) { in TriangularSolveThunk()
46 int64 a_batch_stride, int64 b_batch_stride);
1645 int64 b_batch_stride = m * n * elem_size; in EmitTriangularSolveFromMlir() local1655 m, n, a_batch_stride, b_batch_stride)); in EmitTriangularSolveFromMlir()