Searched refs:k_and_expected (Results 1 – 1 of 1) sorted by relevance
47 std::map<int, Array2D<T>> k_and_expected() const { in k_and_expected() function in xla::__anon94e7bccf0111::MatrixTest81 for (const auto& kv : k_and_expected<T>()) { in TestMatrixDiagonal()95 for (const auto& kv : k_and_expected<T>()) { in TestSetMatrixDiagonal()131 std::map<int, Array3D<T>> k_and_expected = { in TestMatrixDiagonal4D() local142 for (const auto& kv : k_and_expected) { in TestMatrixDiagonal4D()