Searched refs:max_sparse_elements (Results 1 – 10 of 10) sorted by relevance
55 layout.set_max_sparse_elements(proto.max_sparse_elements()); in CreateFromProto()81 return absl::StrCat("sparse{", max_sparse_elements(), "}"); in ToString()101 lhs.max_sparse_elements() != rhs.max_sparse_elements()) { in operator ()()
94 /* static */ Layout LayoutUtil::MakeSparseLayout(int64 max_sparse_elements) { in MakeSparseLayout() argument97 layout.set_max_sparse_elements(max_sparse_elements); in MakeSparseLayout()303 return layout.max_sparse_elements(); in MaxSparseElements()461 hash_value = Hash64Combine(hash_value, layout.max_sparse_elements()); in Hash()
207 int64 max_sparse_elements) { in MakeShapeWithSparseLayout() argument210 *shape.mutable_layout() = LayoutUtil::MakeSparseLayout(max_sparse_elements); in MakeShapeWithSparseLayout()691 int64 max_sparse_elements = LayoutUtil::MaxSparseElements(shape.layout()); in ValidateShapeSize() local692 if (max_sparse_elements < 0) { in ValidateShapeSize()693 return max_sparse_elements; in ValidateShapeSize()696 max_sparse_elements, ByteSizeOfPrimitiveType(shape.element_type())); in ValidateShapeSize()701 MultiplyWithoutOverflow(max_sparse_elements, shape.rank()); in ValidateShapeSize()
53 static Layout MakeSparseLayout(int64 max_sparse_elements);
186 int64 max_sparse_elements() const { return max_sparse_elements_; } in max_sparse_elements() function
39 int64 max_sparse_elements) { in MakeShapeWithSparseLayout() argument41 *shape.mutable_layout() = LayoutUtil::MakeSparseLayout(max_sparse_elements); in MakeShapeWithSparseLayout()
407 int64 max_sparse_elements);
127 const int64 max_sparse_elements = in SetPiece() local130 new char[max_sparse_elements * in SetPiece()133 new SparseIndexArray(max_sparse_elements, shape.rank())); in SetPiece()
154 int64 max_sparse_elements = 5; field
3567 int64 max_sparse_elements; in ParseShape() local3569 !ParseInt64(&max_sparse_elements) || in ParseShape()3574 LayoutUtil::MakeSparseLayout(max_sparse_elements); in ParseShape()