Searched refs:IsMatMul (Results 1 – 3 of 3) sorted by relevance
235 IsMatMul(*contraction)) { in IsCpuCompatibleDataType()239 IsMatMul(*contraction)) { in IsCpuCompatibleDataType()245 } else if (IsMatMul(*contraction)) { in IsCpuCompatibleDataType()292 DCHECK(IsMatMul(*matmul)) << "Expected MatMul op"; in IsCpuCompatibleDataType()317 } else if (IsMatMul(node)) { in IsCpuCompatibleDataType()420 IsMatMul(*contraction_node_def) || in IsCpuCompatibleDataType()472 if (!IsMatMul(*contraction_node_def) && IsTanh(*node_def)) return false; in IsCpuCompatibleDataType()475 if (!(IsConv2D(*contraction_node_def) || IsMatMul(*contraction_node_def)) && in IsCpuCompatibleDataType()1035 DCHECK(IsMatMul(matmul)) << "Input node must be a MatMul"; in IsCpuCompatibleDataType()1085 } else if (IsMatMul(contraction)) { in IsCpuCompatibleDataType()[all …]
113 bool IsMatMul(const NodeDef& node);
365 bool IsMatMul(const NodeDef& node) { return node.op() == "MatMul"; } in IsMatMul() function