Searched refs:ElementsInRecursive (Results 1 – 3 of 3) sorted by relevance
436 /* static */ int64 ShapeUtil::ElementsInRecursive(const Shape& shape) { in ElementsInRecursive() function in xla::ShapeUtil443 count += ElementsInRecursive(element_shape); in ElementsInRecursive()
200 static int64 ElementsInRecursive(const Shape& shape);
125 return ShapeUtil::ElementsInRecursive(instr->shape()) < 4096; in IsSmall()