Searched refs:a_batch_stride_ (Results 1 – 4 of 4) sorted by relevance
53 a_batch_stride_(a_batch_stride), in TriangularSolveThunk()79 << " a_batch_stride=" << a_batch_stride_ in ExecuteOnStream()92 se::DeviceMemoryBase(a_base + i * a_batch_stride_, a_batch_stride_); in ExecuteOnStream()
48 a_batch_stride_(n * n * ShapeUtil::ByteSizeOfPrimitiveType(type)), in CholeskyThunk()78 se::DeviceMemoryBase(a_base + i * a_batch_stride_, a_batch_stride_); in ExecuteOnStream()
64 const int64 a_batch_stride_; variable
66 const int64 a_batch_stride_; variable