| 1310 | class TestSparseCSFTensorEquality : public TestSparseTensorBase<ValueType> { |
| 1311 | public: |
| 1312 | void SetUp() { |
| 1313 | shape_ = {2, 3, 4, 5}; |
| 1314 | |
| 1315 | values1_[0][0][0][1] = 1; |
| 1316 | values1_[0][0][0][2] = 2; |
| 1317 | values1_[0][1][0][0] = 3; |
| 1318 | values1_[0][1][0][2] = 4; |
| 1319 | values1_[0][1][1][0] = 5; |
| 1320 | values1_[1][1][1][0] = 6; |
| 1321 | values1_[1][1][1][1] = 7; |
| 1322 | values1_[1][1][1][2] = 8; |
| 1323 | |
| 1324 | length_ = sizeof(values1_); |
| 1325 | |
| 1326 | values2_[0][0][0][1] = 1; |
| 1327 | values2_[0][0][0][2] = 2; |
| 1328 | values2_[0][1][0][0] = 3; |
| 1329 | values2_[0][1][0][2] = 9; |
| 1330 | values2_[0][1][1][0] = 5; |
| 1331 | values2_[1][1][1][0] = 6; |
| 1332 | values2_[1][1][1][1] = 7; |
| 1333 | values2_[1][1][1][2] = 8; |
| 1334 | |
| 1335 | auto buffer1 = Buffer::Wrap(values1_, length_); |
| 1336 | auto buffer2 = Buffer::Wrap(values2_, length_); |
| 1337 | |
| 1338 | DCHECK_OK(NumericTensor<ValueType>::Make(buffer1, shape_).Value(&tensor1_)); |
| 1339 | DCHECK_OK(NumericTensor<ValueType>::Make(buffer2, shape_).Value(&tensor2_)); |
| 1340 | } |
| 1341 | |
| 1342 | protected: |
| 1343 | using TestSparseTensorBase<ValueType>::shape_; |