Lines Matching refs:n3
254 std::vector<int64> lhs_dims = {lhs.n1(), lhs.n2(), lhs.n3()};
255 std::vector<int64> rhs_dims = {rhs.n1(), rhs.n2(), rhs.n3()};
256 std::vector<int64> out_dims = {rhs.n1(), rhs.n2(), rhs.n3()};
268 for (int64 i2 = 0; i2 < result->n3(); ++i2) {
270 i0 < lhs.n1() && i1 < lhs.n2() && i2 < lhs.n3()
274 i2 >= lhs.n3() ? i2 - lhs.n3() : i2);
288 std::vector<int64> lhs_dims = {lhs.n1(), lhs.n2(), lhs.n3(), lhs.n4()};
289 std::vector<int64> rhs_dims = {rhs.n1(), rhs.n2(), rhs.n3(), rhs.n4()};
290 std::vector<int64> out_dims = {rhs.n1(), rhs.n2(), rhs.n3(), rhs.n4()};
303 for (int64 i2 = 0; i2 < result->n3(); ++i2) {
306 i0 < lhs.n1() && i1 < lhs.n2() && i2 < lhs.n3() && i3 < lhs.n4()
310 i2 >= lhs.n3() ? i2 - lhs.n3() : i2,
362 CHECK_LE(starts[2], input.n3());
365 CHECK_LE(limits[2], input.n3());
376 for (int64 i2 = 0; i2 < result->n3(); ++i2) {
393 CHECK_LE(starts[2], input.n3());
397 CHECK_LE(limits[2], input.n3());
410 for (int64 i2 = 0; i2 < result->n3(); ++i2) {
542 operand.n3()};
601 operand.n3(), operand.n4()};