Searched refs:kTransposeB (Results 1 – 6 of 6) sorted by relevance
37 void Affine::set_transpose_b(bool transpose_b) { (void)AddAttr(kTransposeB, MakeValue(transpose_b))… in set_transpose_b()49 auto value_ptr = GetAttr(kTransposeB); in get_transpose_b()
110 void MatMul::set_transpose_b(bool transpose_b) { (void)AddAttr(kTransposeB, MakeValue(transpose_b))… in set_transpose_b()118 auto value_ptr = GetAttr(kTransposeB); in get_transpose_b()
136 void BatchMatmul::set_transpose_b(bool transpose_b) { (void)AddAttr(kTransposeB, MakeValue(transpos… in set_transpose_b()144 auto value_ptr = GetAttr(kTransposeB); in get_transpose_b()
198 constexpr auto kTransposeB = "transpose_b"; variable
100 if (matmul_prim->GetAttr(ops::kTransposeB) != nullptr) { in Process()
696 …index != 1 || (matmul_prim->GetAttr(ops::kTransposeB) != nullptr && matmul_prim->get_transpose_b()… in CalQuantAssitInfo()