Home
last modified time | relevance | path

Searched refs:ElementsInRecursive (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h205 static int64 ElementsInRecursive(const Shape& shape);
Dshape_util.cc574 /* static */ int64 ShapeUtil::ElementsInRecursive(const Shape& shape) { in ElementsInRecursive() function in xla::ShapeUtil
581 count += ElementsInRecursive(element_shape); in ElementsInRecursive()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc834 ShapeUtil::ElementsInRecursive(random->shape()); in HandleRngBitGenerator()
Dhlo_graph_dumper.cc126 return ShapeUtil::ElementsInRecursive(instr->shape()) < 4096; in IsSmall()
Dmemory_space_assignment.cc125 use_size += ShapeUtil::ElementsInRecursive(use.instruction->shape()); in FindCrossProgramPrefetchCandidate()