Home
last modified time | relevance | path

Searched refs:SubshapeCount (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion.cc91 int64 num_output_buffers = ShapeUtil::SubshapeCount(a->shape()) + in FusionWouldBeTooLarge()
92 ShapeUtil::SubshapeCount(b->shape()); in FusionWouldBeTooLarge()
Dmulti_output_fusion_test.cc618 EXPECT_LE(instr->operand_count() + ShapeUtil::SubshapeCount(instr->shape()), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h476 static int64 SubshapeCount(const Shape& shape);
Dshape_util.cc351 /* static */ int64 ShapeUtil::SubshapeCount(const Shape& shape) { in SubshapeCount() function in xla::ShapeUtil