Searched refs:ElementsInRecursive (Results 1 – 5 of 5) sorted by relevance
205 static int64 ElementsInRecursive(const Shape& shape);
574 /* static */ int64 ShapeUtil::ElementsInRecursive(const Shape& shape) { in ElementsInRecursive() function in xla::ShapeUtil581 count += ElementsInRecursive(element_shape); in ElementsInRecursive()
834 ShapeUtil::ElementsInRecursive(random->shape()); in HandleRngBitGenerator()
126 return ShapeUtil::ElementsInRecursive(instr->shape()) < 4096; in IsSmall()
125 use_size += ShapeUtil::ElementsInRecursive(use.instruction->shape()); in FindCrossProgramPrefetchCandidate()