Searched refs:SubshapeCount (Results 1 – 5 of 5) sorted by relevance
149 ShapeUtil::SubshapeCount( in TEST()153 ShapeUtil::SubshapeCount( in TEST()
315 int64 num_output_buffers = ShapeUtil::SubshapeCount(instr1.shape()) + in FusionWouldBeTooLarge()316 ShapeUtil::SubshapeCount(instr2.shape()); in FusionWouldBeTooLarge()
759 EXPECT_LE(instr->operand_count() + ShapeUtil::SubshapeCount(instr->shape()), in TEST_F()
489 static int64 SubshapeCount(const Shape& shape);
398 /* static */ int64 ShapeUtil::SubshapeCount(const Shape& shape) { in SubshapeCount() function in xla::ShapeUtil