Searched refs:IsNDOperateWithMD (Results 1 – 1 of 1) sorted by relevance
1298 (IsNDOperateWithMD(4, 0) || IsNDOperateWithMD(4, 1) || in ShouldProcess()1299 IsNDOperateWithMD(4, 4) || IsNDOperateWithMD(0, 4) || in ShouldProcess()1300 IsNDOperateWithMD(1, 4)) && in ShouldProcess()1321 bool IsNDOperateWithMD(int n, int m) const { in IsNDOperateWithMD() function in tensorflow::grappler::__anonbd4a17fb0111::BinaryOpProcessor1389 if (IsNDOperateWithMD(4, 1)) { in CustomizedProcessing()1391 } else if (IsNDOperateWithMD(1, 4)) { in CustomizedProcessing()