Searched defs:axesToMul (Results 1 – 2 of 2) sorted by relevance
501 auto axesToMul = BatchMatMulDescriptor::GetAxesToMul(dataLayout, tensorShape); in GetAxesNotMul() local519 auto axesToMul = BatchMatMulDescriptor::GetAxesToMul(dataLayout, tensorShape); in GetPermuteVec() local
4134 auto axesToMul = BatchMatMulDescriptor::GetAxesToMul(m_Parameters.m_DataLayoutX, in Validate() local4159 auto axesToMul = BatchMatMulDescriptor::GetAxesToMul(m_Parameters.m_DataLayoutY, in Validate() local