Searched defs:maindiag (Results 1 – 5 of 5) 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
46 maindiag, argument102 def _makeTridiagonalMatrix(self, superdiag, maindiag, subdiag): argument
42 maindiag, argument93 def _makeTridiagonalMatrix(self, superdiag, maindiag, subdiag): argument
286 ShapeHandle maindiag; in TridiagonalMatMulShapeFn() local