Searched defs:expected_shapes (Results 1 – 7 of 7) sorted by relevance
25 std::map<string, std::vector<PartialTensorShape>> expected_shapes) { in ShapeAnnotationsMatch()
80 std::vector<Shape> expected_shapes(reduce->input_count()); in HandleReduce() local
459 std::vector<Shape> expected_shapes; in MakeReduceHlo() local
186 def testNestedStructure(self, components, expected_shapes): argument
169 expected_shapes=None, argument
1383 def _norm_backwards_test_helper(self, op, args, output_mask, expected_shapes): argument
234 std::vector<std::vector<int64_t>> expected_shapes, in checkProfiledEvents()