Searched refs:_check_special (Results 1 – 1 of 1) sorted by relevance
372 _check_special(prim.name, buf.xla_shape(), buf)374 def _check_special(name, xla_shape, buf): function