Searched defs:diag_rank (Results 1 – 2 of 2) sorted by relevance
118 const TensorShape& input_shape, const int64 diag_rank, in SetMatrixDiag()273 const int64 diag_rank = diag_shape.dims(); in Compile() local469 const int diag_rank = diag_shape.dims(); in Compile() local
202 const int diag_rank = diagonal_shape.dims(); in Compute() local