Home
last modified time | relevance | path

Searched refs:IsSparseArray (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_shape_verifier.cc24 if (LayoutUtil::IsSparseArray(shape)) { in Preprocess()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_shape_verifier.cc24 if (LayoutUtil::IsSparseArray(shape)) { in Preprocess()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dlayout_util_test.cc94 EXPECT_TRUE(LayoutUtil::IsSparseArray(src)); in TEST_F()
95 EXPECT_FALSE(LayoutUtil::IsSparseArray(dst)); in TEST_F()
H A Dlayout_util.h103 static bool IsSparseArray(const Shape& shape);
H A Dlayout_util.cc300 /* static */ bool LayoutUtil::IsSparseArray(const Shape& shape) { in IsSparseArray() function in xla::LayoutUtil