Searched refs:TransposeLhsForGtsvBatched (Results 1 – 1 of 1) sorted by relevance
295 TransposeLhsForGtsvBatched(context, lhs, lhs_transposed); in ComputeWithGtsvBatched()328 void TransposeLhsForGtsvBatched(OpKernelContext* context, const Tensor& lhs, in TransposeLhsForGtsvBatched() function in tensorflow::TridiagonalSolveOpGpu