Searched refs:b_batch_stride (Results 1 – 3 of 3) sorted by relevance
39 int64 b_batch_stride, const HloInstruction* hlo) in TriangularSolveThunk() argument54 b_batch_stride_(b_batch_stride) { in TriangularSolveThunk()
45 int64 a_batch_stride, int64 b_batch_stride,
1898 int64 b_batch_stride = m * n * elem_size; in BuildTriangularSolveThunk() local1904 b_batch_stride, inst); in BuildTriangularSolveThunk()