| 136 | } |
| 137 | |
| 138 | Status DatasetOpsTestBase::ExpectEqual(const Tensor& a, const Tensor& b) { |
| 139 | switch (a.dtype()) { |
| 140 | #define CASE(DT) \ |
| 141 | case DataTypeToEnum<DT>::value: \ |
| 142 | TF_RETURN_IF_ERROR(IsEqual<DT>(a, b)); \ |
| 143 | break; |
| 144 | TF_CALL_NUMBER_TYPES(CASE); |
| 145 | TF_CALL_tstring(CASE); |
| 146 | TF_CALL_uint32(CASE); |
| 147 | TF_CALL_uint64(CASE); |
| 148 | // TODO(feihugis): figure out how to support variant tensors. |
| 149 | #undef CASE |
| 150 | default: |
| 151 | return errors::Internal("Unsupported dtype: ", a.dtype()); |
| 152 | } |
| 153 | return Status::OK(); |
| 154 | } |
| 155 | |
| 156 | template <typename T> |
| 157 | bool compare(const Tensor& t1, const Tensor& t2) { |