| 809 | } // namespace |
| 810 | |
| 811 | bool CompareConstantArrays(const Array& lhs_array, const Array& rhs_array) { |
| 812 | bool attrs_equal = lhs_array.shape() == rhs_array.shape() && |
| 813 | lhs_array.data_type == rhs_array.data_type && |
| 814 | lhs_array.final_data_type == rhs_array.final_data_type && |
| 815 | HaveSameMinMax(lhs_array, rhs_array) && |
| 816 | HaveSameQuantizationParams(lhs_array, rhs_array) && |
| 817 | lhs_array.narrow_range == rhs_array.narrow_range; |
| 818 | if (!attrs_equal) { |
| 819 | return false; |
| 820 | } |
| 821 | switch (lhs_array.data_type) { |
| 822 | case ArrayDataType::kBool: |
| 823 | return CompareArrayBuffers<ArrayDataType::kBool>(lhs_array, rhs_array); |
| 824 | case ArrayDataType::kFloat: |
| 825 | return CompareArrayBuffers<ArrayDataType::kFloat>(lhs_array, rhs_array); |
| 826 | case ArrayDataType::kInt8: |
| 827 | return CompareArrayBuffers<ArrayDataType::kInt8>(lhs_array, rhs_array); |
| 828 | case ArrayDataType::kUint8: |
| 829 | return CompareArrayBuffers<ArrayDataType::kUint8>(lhs_array, rhs_array); |
| 830 | case ArrayDataType::kInt16: |
| 831 | return CompareArrayBuffers<ArrayDataType::kInt16>(lhs_array, rhs_array); |
| 832 | case ArrayDataType::kUint16: |
| 833 | return CompareArrayBuffers<ArrayDataType::kUint16>(lhs_array, rhs_array); |
| 834 | case ArrayDataType::kInt32: |
| 835 | return CompareArrayBuffers<ArrayDataType::kInt32>(lhs_array, rhs_array); |
| 836 | case ArrayDataType::kUint32: |
| 837 | return CompareArrayBuffers<ArrayDataType::kUint32>(lhs_array, rhs_array); |
| 838 | case ArrayDataType::kInt64: |
| 839 | return CompareArrayBuffers<ArrayDataType::kInt64>(lhs_array, rhs_array); |
| 840 | case ArrayDataType::kUint64: |
| 841 | return CompareArrayBuffers<ArrayDataType::kUint64>(lhs_array, rhs_array); |
| 842 | case ArrayDataType::kString: |
| 843 | return CompareArrayBuffers<ArrayDataType::kString>(lhs_array, rhs_array); |
| 844 | case ArrayDataType::kComplex64: |
| 845 | return CompareArrayBuffers<ArrayDataType::kComplex64>(lhs_array, |
| 846 | rhs_array); |
| 847 | default: |
| 848 | LOG(FATAL) << "Unsupported data type: " |
| 849 | << ArrayDataTypeName(lhs_array.data_type); |
| 850 | return false; |
| 851 | } |
| 852 | } |
| 853 | |
| 854 | namespace { |
| 855 | // Take an array name, which may be something like "name:3_5" and make it |
no test coverage detected