Searched defs:b_batch_stride (Results 1 – 2 of 2) sorted by relevance
39 int64 b_batch_stride, const HloInstruction* hlo) in TriangularSolveThunk()
1898 int64 b_batch_stride = m * n * elem_size; in BuildTriangularSolveThunk() local