Searched refs:ElementHasBitWidth (Results 1 – 3 of 3) sorted by relevance
148 if (ShapeUtil::ElementHasBitWidth(index_shape, 64) && in TorchGather()258 if (ShapeUtil::ElementHasBitWidth(index_shape, 64) && in TorchIndexSelect()
328 return ElementIsIntegral(shape) && ElementHasBitWidth(shape, bits); in ElementIsIntegralWithBits()331 /* static */ bool ShapeUtil::ElementHasBitWidth(const Shape& shape, int bits) { in ElementHasBitWidth() function in xla::ShapeUtil
460 static bool ElementHasBitWidth(const Shape& shape, int bits);