Searched refs:SubshapeCount (Results 1 – 6 of 6) sorted by relevance
286 auto last = first + ShapeUtil::SubshapeCount(src_shape); in CopySubtreeFrom()296 size_t count = ShapeUtil::SubshapeCount(*sub_shape); in SubShapeTree()324 DCHECK_EQ(nodes_.size(), ShapeUtil::SubshapeCount(*shape)); in ShapeTree()
437 static int64_t SubshapeCount(const Shape& shape);
570 /* static */ int64_t ShapeUtil::SubshapeCount(const Shape& shape) { in SubshapeCount() function in xla::ShapeUtil
109 ShapeUtil::SubshapeCount( in TEST()
442 int64_t num_output_buffers = ShapeUtil::SubshapeCount(instr1.shape()) + in FusionFitsInBudget()443 ShapeUtil::SubshapeCount(instr2.shape()); in FusionFitsInBudget()
763 EXPECT_LE(instr->operand_count() + ShapeUtil::SubshapeCount(instr->shape()), in TEST_F()