Searched refs:IsDenseArray (Results 1 – 13 of 13) sorted by relevance
204 TF_RET_CHECK(LayoutUtil::IsDenseArray(param_subshape)); in Verify()205 TF_RET_CHECK(LayoutUtil::IsDenseArray(output_subshape)); in Verify()
77 EXPECT_TRUE(Match(&scalar_shape, match::Shape().IsDenseArray())); in TEST_F()91 EXPECT_TRUE(Match(&array_shape, match::Shape().IsDenseArray())); in TEST_F()
1707 LayoutUtil::IsDenseArray(instruction->shape()) && in Postprocess()1713 if (LayoutUtil::IsDenseArray(operand_shape) && in Postprocess()
1052 constexpr auto IsDenseArray() const
2290 CHECK(LayoutUtil::IsDenseArray(shape)) << shape.ToString(true); in ParseNonTupleLiteral()
91 static bool IsDenseArray(const Shape& shape);
269 /* static */ bool LayoutUtil::IsDenseArray(const Shape& shape) { in IsDenseArray() function in xla::LayoutUtil316 CHECK(IsDenseArray(shape)); in MinorToMajor()
694 CHECK(LayoutUtil::IsDenseArray(permuted_shape)); in Transpose()785 CHECK(LayoutUtil::IsDenseArray(subshape)); in GetAsString()830 CHECK(LayoutUtil::IsDenseArray(shape())); in GetIntegralAsS64()851 CHECK(LayoutUtil::IsDenseArray(shape())); in GetAsDouble()911 CHECK(LayoutUtil::IsDenseArray(shape())); in SetIntegralAsS64()940 CHECK(LayoutUtil::IsDenseArray(shape())); in SetFromDouble()1075 CHECK(LayoutUtil::IsDenseArray(subshape)); in ToStringHelper()1367 LayoutUtil::IsDenseArray(subshape())) { in EqualElements()
154 if (LayoutUtil::IsDenseArray(lhs)) { in operator ()()
858 CHECK(LayoutUtil::IsDenseArray(subshape())); in Get()866 CHECK(LayoutUtil::IsDenseArray(subshape())); in Set()994 TF_RET_CHECK(LayoutUtil::IsDenseArray(this_shape)); in PopulateInternal()
316 CHECK(LayoutUtil::IsDenseArray(*shape)); in AppendMajorDimension()654 CHECK(LayoutUtil::IsDenseArray(shape)) << shape.ShortDebugString(); in ByteSizeOfElements()970 CHECK(LayoutUtil::IsDenseArray(shape)); in PermuteDimensions()
460 LayoutUtil::IsDenseArray(crs->operand(0)->shape()); in CanImplement()
425 CHECK(xla::LayoutUtil::IsDenseArray(shape)) in __xla_cpu_runtime_AllReduce()