Searched refs:lhs_batch_dims (Results 1 – 2 of 2) sorted by relevance
1018 optional<std::vector<int64>> lhs_batch_dims; in ParseInstruction() local1020 &lhs_batch_dims}; in ParseInstruction()1039 if (lhs_batch_dims) { in ParseInstruction()1040 *dnum.mutable_lhs_batch_dimensions() = {lhs_batch_dims->begin(), in ParseInstruction()1041 lhs_batch_dims->end()}; in ParseInstruction()
860 …ROOT dot = f32[2,3]{1,0} dot(a, b), lhs_batch_dims={0}, lhs_contracting_dims={1}, rhs_contracting_…