Searched refs:IsNestedTuple (Results 1 – 12 of 12) sorted by relevance
336 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape({}))); in TEST()337 EXPECT_FALSE(ShapeUtil::IsNestedTuple( in TEST()339 EXPECT_TRUE(ShapeUtil::IsNestedTuple( in TEST()341 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()343 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()345 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()347 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()354 EXPECT_FALSE(ShapeUtil::IsNestedTuple(nil)); in TEST()356 ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShapeWithPtrs({&s32}))); in TEST()358 ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShapeWithPtrs({&nil}))); in TEST()[all …]
423 static bool IsNestedTuple(const Shape& shape);
2656 CHECK(!ShapeUtil::IsNestedTuple(*shape_)); in MutableBorrowingLiteral()2715 CHECK(!ShapeUtil::IsNestedTuple(*shape_)); in BorrowingLiteral()
547 /* static */ bool ShapeUtil::IsNestedTuple(const Shape& shape) { in IsNestedTuple() function in xla::ShapeUtil
187 if (ShapeUtil::IsNestedTuple(shape)) { in TransferLiteralToInfeedOnCpu()246 if (ShapeUtil::IsNestedTuple(literal.shape())) { in TransferLiteralFromOutfeedOnCpu()
425 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(data_shape)); in HandleInfeed()544 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(operand_shape)); in HandleOutfeed()
35 return shape.IsTuple() && !ShapeUtil::IsNestedTuple(shape); in IsNonNestedTuple()
143 if (ShapeUtil::IsNestedTuple(shaped_buffer->on_device_shape())) { in DeconstructTuple()
1014 if (!ShapeUtil::IsNestedTuple(while_shape)) { in TryFlattenNestedTuples()
266 TF_RET_CHECK(!xla::ShapeUtil::IsNestedTuple(result_buffers.on_host_shape())); in UpdateOutputVariables()
433 TF_RET_CHECK(!xla::ShapeUtil::IsNestedTuple(scoped_buffers.on_host_shape())); in AllocateOutputTensors()
3159 TF_RET_CHECK(shape.IsTuple() && !ShapeUtil::IsNestedTuple(shape)); in AllToAllTuple()