Searched refs:ElementHasBitWidth (Results 1 – 3 of 3) sorted by relevance
163 if (ShapeUtil::ElementHasBitWidth(index_shape, 64) && in TorchGather()273 if (ShapeUtil::ElementHasBitWidth(index_shape, 64) && in TorchIndexSelect()
509 static bool ElementHasBitWidth(const Shape& shape, int bits);
438 return ElementIsIntegral(shape) && ElementHasBitWidth(shape, bits); in ElementIsIntegralWithBits()441 /* static */ bool ShapeUtil::ElementHasBitWidth(const Shape& shape, int bits) { in ElementHasBitWidth() function in xla::ShapeUtil