Searched refs:IsAnyBatchMatMul (Results 1 – 3 of 3) sorted by relevance
31 bool IsAnyBatchMatMul(const NodeDef& node);
54 bool IsAnyBatchMatMul(const NodeDef& node) { in IsAnyBatchMatMul() function60 IsAnyBatchMatMul(node) || IsQuantizedMatMul(node); in IsAnyMatMul()
2306 : (IsAnyBatchMatMul(*node) ? std::set<string>{"ConjugateTranspose"} in TrySimplify()2318 const string attr_a = IsAnyBatchMatMul(*node) ? "adj_x" : "transpose_a"; in TrySimplify()2327 const string attr_b = IsAnyBatchMatMul(*node) ? "adj_y" : "transpose_b"; in TrySimplify()