Searched refs:IsNestedTuple (Results 1 – 9 of 9) sorted by relevance
337 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape({}))); in TEST()338 EXPECT_FALSE(ShapeUtil::IsNestedTuple( in TEST()340 EXPECT_TRUE(ShapeUtil::IsNestedTuple( in TEST()342 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()344 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()346 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()348 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()
475 static bool IsNestedTuple(const Shape& shape);
375 /* static */ bool ShapeUtil::IsNestedTuple(const Shape& shape) { in IsNestedTuple() function in xla::ShapeUtil
2120 CHECK(!ShapeUtil::IsNestedTuple(*shape_)); in BorrowingLiteral()
104 if (ShapeUtil::IsNestedTuple(shape)) { in TransferLiteralToInfeed()200 if (ShapeUtil::IsNestedTuple(literal_shape)) { in TransferLiteralFromOutfeed()
406 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(data_shape)); in HandleInfeed()548 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(operand_shape)); in HandleOutfeed()
35 return shape.IsTuple() && !ShapeUtil::IsNestedTuple(shape); in IsNonNestedTuple()
154 if (ShapeUtil::IsNestedTuple(shaped_buffer->on_device_shape())) { in DeconstructTuple()
660 if (!ShapeUtil::IsNestedTuple(while_shape)) { in TryFlattenNestedTuples()