Searched refs:IsNestedTuple (Results 1 – 9 of 9) sorted by relevance
299 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape({}))); in TEST()300 EXPECT_FALSE(ShapeUtil::IsNestedTuple( in TEST()302 EXPECT_TRUE(ShapeUtil::IsNestedTuple( in TEST()304 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()306 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()308 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()310 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()
402 static bool IsNestedTuple(const Shape& shape);
320 /* static */ bool ShapeUtil::IsNestedTuple(const Shape& shape) { in IsNestedTuple() function in xla::ShapeUtil
43 CHECK(!ShapeUtil::IsNestedTuple(hlo_instruction()->shape())); in ExecuteOnStream()
60 if (ShapeUtil::IsNestedTuple(shape)) { in TransferLiteralToInfeed()
104 if (ShapeUtil::IsNestedTuple(shape)) { in TransferLiteralToInfeed()201 if (ShapeUtil::IsNestedTuple(literal_shape)) { in TransferLiteralFromOutfeed()
341 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(shape)); in HandleInfeed()465 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(operand_shape)); in HandleOutfeed()
111 if (ShapeUtil::IsNestedTuple(shaped_buffer->on_device_shape())) { in DeconstructTuple()
88 CHECK(!ShapeUtil::IsNestedTuple(tuple_shape)); in Tuple()