Home
last modified time | relevance | path

Searched refs:MakeSparseLayout (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util.h41 static Layout MakeSparseLayout(int64 max_sparse_elements);
Dlayout_util_test.cc41 *shape.mutable_layout() = LayoutUtil::MakeSparseLayout(max_sparse_elements); in MakeShapeWithSparseLayout()
294 EXPECT_EQ(LayoutUtil::MaxSparseElements(LayoutUtil::MakeSparseLayout(101)), in TEST_F()
Dlayout_util.cc67 /* static */ Layout LayoutUtil::MakeSparseLayout(int64 max_sparse_elements) { in MakeSparseLayout() function in xla::LayoutUtil
Dshape_util.cc211 *shape.mutable_layout() = LayoutUtil::MakeSparseLayout(max_sparse_elements); in MakeShapeWithSparseLayout()