Searched refs:assertMatchesShape (Results 1 – 2 of 2) sorted by relevance
124 reader->assertMatchesShape(dimRank, dimShape);194 void assertMatchesShape(uint64_t rank, const uint64_t *shape) const;
65 void SparseTensorReader::assertMatchesShape(uint64_t rank, in assertMatchesShape() function in SparseTensorReader