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