Searched defs:maindiag (Results 1 – 4 of 4) sorted by relevance
39 const Scalar* __restrict__ maindiag, in TridiagonalMatMulKernel()64 const Tensor& maindiag = context->input(1); in Compute() local
90 const auto& maindiag = inputs[1].row(0); in ComputeMatrix() local
43 maindiag, argument94 def _makeTridiagonalMatrix(self, superdiag, maindiag, subdiag): argument
286 ShapeHandle maindiag; in TridiagonalMatMulShapeFn() local