Searched refs:IsNestedTuple (Results 1 – 11 of 11) sorted by relevance
339 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape({}))); in TEST()340 EXPECT_FALSE(ShapeUtil::IsNestedTuple( in TEST()342 EXPECT_TRUE(ShapeUtil::IsNestedTuple( in TEST()344 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()346 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()348 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()350 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()
524 static bool IsNestedTuple(const Shape& shape);
2370 CHECK(!ShapeUtil::IsNestedTuple(*shape_)); in MutableBorrowingLiteral()2429 CHECK(!ShapeUtil::IsNestedTuple(*shape_)); in BorrowingLiteral()
485 /* static */ bool ShapeUtil::IsNestedTuple(const Shape& shape) { in IsNestedTuple() function in xla::ShapeUtil
35 return shape.IsTuple() && !ShapeUtil::IsNestedTuple(shape); in IsNonNestedTuple()
142 if (ShapeUtil::IsNestedTuple(shaped_buffer->on_device_shape())) { in DeconstructTuple()
871 if (!ShapeUtil::IsNestedTuple(while_shape)) { in TryFlattenNestedTuples()
104 if (ShapeUtil::IsNestedTuple(shape)) { in TransferLiteralToInfeed()200 if (ShapeUtil::IsNestedTuple(literal.shape())) { in TransferLiteralFromOutfeed()
411 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(data_shape)); in HandleInfeed()530 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(operand_shape)); in HandleOutfeed()
434 TF_RET_CHECK(!xla::ShapeUtil::IsNestedTuple(scoped_buffers.on_host_shape())); in AllocateOutputTensors()
2963 TF_RET_CHECK(shape.IsTuple() && !ShapeUtil::IsNestedTuple(shape)); in AllToAllTuple()