Home
last modified time | relevance | path

Searched refs:actual_shapes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.cc162 bool AreShapesCompatible(const std::vector<TensorShape>& actual_shapes, in AreShapesCompatible() argument
177 for (int i = 0; i < actual_shapes.size(); ++i) { in AreShapesCompatible()
178 if (!match_shape(actual_shapes[i], cached_shapes[i])) { in AreShapesCompatible()
Dutils.h175 bool AreShapesCompatible(const std::vector<TensorShape>& actual_shapes,
/external/tensorflow/tensorflow/python/kernel_tests/
Dlookup_ops_test.py494 actual_shapes = [t.shape for t in table.export()]
503 self.assertLen(actual_shapes, 2)
505 self.assertTrue(inferred_shapes[0].is_compatible_with(actual_shapes[0]))
506 self.assertTrue(inferred_shapes[1].is_compatible_with(actual_shapes[1]))
2100 actual_shapes = [t.shape for t in table.export()]
2109 self.assertLen(actual_shapes, 2)
2111 self.assertTrue(inferred_shapes[0].is_compatible_with(actual_shapes[0]))
2112 self.assertTrue(inferred_shapes[1].is_compatible_with(actual_shapes[1]))
3665 actual_shapes = [t.shape for t in table.export()]
3674 self.assertLen(actual_shapes, 2)
[all …]