Searched refs:IsNestedTuple (Results 1 – 9 of 9) sorted by relevance
315 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape({}))); in TEST()316 EXPECT_FALSE(ShapeUtil::IsNestedTuple( in TEST()318 EXPECT_TRUE(ShapeUtil::IsNestedTuple( in TEST()320 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()322 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()324 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()326 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()
462 static bool IsNestedTuple(const Shape& shape);
328 /* static */ bool ShapeUtil::IsNestedTuple(const Shape& shape) { in IsNestedTuple() function in xla::ShapeUtil
2274 CHECK(!ShapeUtil::IsNestedTuple(*shape_)); in BorrowingLiteral()
104 if (ShapeUtil::IsNestedTuple(shape)) { in TransferLiteralToInfeed()200 if (ShapeUtil::IsNestedTuple(literal_shape)) { in TransferLiteralFromOutfeed()
347 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(data_shape)); in HandleInfeed()489 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(operand_shape)); in HandleOutfeed()
34 return shape.IsTuple() && !ShapeUtil::IsNestedTuple(shape); in IsNonNestedTuple()
153 if (ShapeUtil::IsNestedTuple(shaped_buffer->on_device_shape())) { in DeconstructTuple()
659 if (!ShapeUtil::IsNestedTuple(while_shape)) { in TryFlattenNestedTuples()