MCPcopy Create free account
hub / github.com/apache/arrow / TensorEquals

Function TensorEquals

cpp/src/arrow/compare.cc:1304–1325  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1302} // namespace
1303
1304bool TensorEquals(const Tensor& left, const Tensor& right, const EqualOptions& opts) {
1305 if (left.type_id() != right.type_id()) {
1306 return false;
1307 } else if (left.size() == 0 && right.size() == 0) {
1308 return true;
1309 } else if (left.shape() != right.shape()) {
1310 return false;
1311 }
1312
1313 switch (left.type_id()) {
1314 // TODO: Support half-float tensors
1315 // case Type::HALF_FLOAT:
1316 case Type::FLOAT:
1317 return FloatTensorEquals<FloatType>(left, right, opts);
1318
1319 case Type::DOUBLE:
1320 return FloatTensorEquals<DoubleType>(left, right, opts);
1321
1322 default:
1323 return IntegerTensorEquals(left, right);
1324 }
1325}
1326
1327namespace {
1328

Callers 1

EqualsMethod · 0.85

Calls 4

IntegerTensorEqualsFunction · 0.85
type_idMethod · 0.45
sizeMethod · 0.45
shapeMethod · 0.45

Tested by

no test coverage detected