Searched refs:ElementsInRecursive (Results 1 – 4 of 4) sorted by relevance
201 static int64 ElementsInRecursive(const Shape& shape);
546 /* static */ int64 ShapeUtil::ElementsInRecursive(const Shape& shape) { in ElementsInRecursive() function in xla::ShapeUtil553 count += ElementsInRecursive(element_shape); in ElementsInRecursive()
805 ShapeUtil::ElementsInRecursive(random->shape()); in HandleRngBitGenerator()
126 return ShapeUtil::ElementsInRecursive(instr->shape()) < 4096; in IsSmall()