Searched defs:EinsumDimensionNumbers (Results 1 – 1 of 1) sorted by relevance
140 struct EinsumDimensionNumbers { struct145 std::vector<int64_t> lhs;146 std::vector<int64_t> rhs;147 std::vector<std::tuple<int64_t, int64_t>> lhs_rhs;148 std::vector<std::tuple<int64_t, int64_t>> lhs_out;149 std::vector<std::tuple<int64_t, int64_t>> rhs_out;150 std::vector<std::tuple<int64_t, int64_t, int64_t>> lhs_rhs_out;