| 417 | class TestSparseCOOTensorEquality : public TestSparseTensorBase<ValueType> { |
| 418 | public: |
| 419 | void SetUp() { |
| 420 | shape_ = {2, 3, 4}; |
| 421 | values1_ = {1, 0, 2, 0, 0, 3, 0, 4, 5, 0, 6, 0, |
| 422 | 0, 11, 0, 12, 13, 0, 14, 0, 0, 15, 0, 16}; |
| 423 | values2_ = {1, 0, 2, 0, 0, 3, 0, 4, 5, 0, 6, 0, |
| 424 | 0, 0, 0, 12, 13, 0, 14, 0, 0, 15, 0, 16}; |
| 425 | auto buffer1 = Buffer::Wrap(values1_); |
| 426 | auto buffer2 = Buffer::Wrap(values2_); |
| 427 | DCHECK_OK(NumericTensor<ValueType>::Make(buffer1, this->shape_).Value(&tensor1_)); |
| 428 | DCHECK_OK(NumericTensor<ValueType>::Make(buffer2, this->shape_).Value(&tensor2_)); |
| 429 | } |
| 430 | |
| 431 | protected: |
| 432 | using TestSparseTensorBase<ValueType>::shape_; |