Searched refs:kTransposeA (Results 1 – 5 of 5) sorted by relevance
35 void Affine::set_transpose_a(bool transpose_a) { (void)AddAttr(kTransposeA, MakeValue(transpose_a))… in set_transpose_a()44 auto value_ptr = GetAttr(kTransposeA); in get_transpose_a()
108 void MatMul::set_transpose_a(bool transpose_a) { (void)AddAttr(kTransposeA, MakeValue(transpose_a))… in set_transpose_a()113 auto value_ptr = GetAttr(kTransposeA); in get_transpose_a()
134 void BatchMatmul::set_transpose_a(bool transpose_a) { (void)AddAttr(kTransposeA, MakeValue(transpos… in set_transpose_a()139 auto value_ptr = GetAttr(kTransposeA); in get_transpose_a()
197 constexpr auto kTransposeA = "transpose_a"; variable
97 if (matmul_prim->GetAttr(ops::kTransposeA) != nullptr) { in Process()