Searched refs:IsDenseArray (Results 1 – 11 of 11) sorted by relevance
198 TF_RET_CHECK(LayoutUtil::IsDenseArray(param_subshape)); in Verify()199 TF_RET_CHECK(LayoutUtil::IsDenseArray(output_subshape)); in Verify()
74 EXPECT_TRUE(Match(&scalar_shape, match::Shape().IsDenseArray())); in TEST()88 EXPECT_TRUE(Match(&array_shape, match::Shape().IsDenseArray())); in TEST()111 EXPECT_FALSE(Match(&array_shape, match::Shape().IsDenseArray())); in TEST()
1439 LayoutUtil::IsDenseArray(instruction->shape()) && in Postprocess()1445 if (LayoutUtil::IsDenseArray(operand_shape) && in Postprocess()
1057 constexpr auto IsDenseArray() const
2165 CHECK(LayoutUtil::IsDenseArray(shape)) << shape.ToString(true); in ParseNonTupleLiteral()
94 static bool IsDenseArray(const Shape& shape);
273 /* static */ bool LayoutUtil::IsDenseArray(const Shape& shape) { in IsDenseArray() function in xla::LayoutUtil333 CHECK(IsDenseArray(shape)); in MinorToMajor()
149 if (LayoutUtil::IsDenseArray(lhs)) { in operator ()()
896 CHECK(LayoutUtil::IsDenseArray(subshape())); in Get()904 CHECK(LayoutUtil::IsDenseArray(subshape())); in Set()1085 TF_RET_CHECK(LayoutUtil::IsDenseArray(this_shape)); in PopulateInternal()
700 CHECK(LayoutUtil::IsDenseArray(permuted_shape)); in Transpose()791 CHECK(LayoutUtil::IsDenseArray(subshape)); in GetAsString()895 CHECK(LayoutUtil::IsDenseArray(shape())); in GetIntegralAsS64()938 CHECK(LayoutUtil::IsDenseArray(shape())); in SetIntegralAsS64()1183 CHECK(LayoutUtil::IsDenseArray(subshape)); in ToStringHelper()1480 LayoutUtil::IsDenseArray(subshape())) { in EqualElements()
269 CHECK(LayoutUtil::IsDenseArray(*shape)); in AppendMajorDimension()607 CHECK(LayoutUtil::IsDenseArray(shape)) << shape.ShortDebugString(); in ByteSizeOfElements()970 CHECK(LayoutUtil::IsDenseArray(shape)); in PermuteDimensions()