We arbitrarily set this as the boundary between "large" and "small" instructions.
| 119 | // We arbitrarily set this as the boundary between "large" and "small" |
| 120 | // instructions. |
| 121 | bool IsSmall(const HloInstruction* instr) { |
| 122 | if (ShapeUtil::HasPrimitiveType(instr->shape(), OPAQUE_TYPE) || |
| 123 | ShapeUtil::HasPrimitiveType(instr->shape(), TOKEN)) { |
| 124 | return true; |
| 125 | } |
| 126 | return ShapeUtil::ElementsInRecursive(instr->shape()) < 4096; |
| 127 | } |
| 128 | |
| 129 | // Node color schemes, used by NodeColorAttributes. |
| 130 | enum ColorScheme { |
no test coverage detected