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

Function TensorEquals

cpp/src/arrow/compare.cc:1302–1323  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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

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