Searched refs:a_batch_stride_ (Results 1 – 4 of 4) sorted by relevance
53 a_batch_stride_(a_batch_stride), in TriangularSolveThunk()78 << " a_batch_stride=" << a_batch_stride_ in ExecuteOnStream()91 se::DeviceMemoryBase(a_base + i * a_batch_stride_, a_batch_stride_); in ExecuteOnStream()
48 a_batch_stride_(n * n * in CholeskyThunk()82 se::DeviceMemoryBase(a_base + i * a_batch_stride_, a_batch_stride_); in ExecuteOnStream()
68 const int64 a_batch_stride_; variable
67 const int64 a_batch_stride_; variable