Searched refs:known_shape (Results 1 – 3 of 3) sorted by relevance
75 const auto& known_shape = known_shape_iter->second; in ReadPartialShapesFromShapeMap() local76 CHECK_GT(known_shape.size(), in->src_output()) << known_shape_iter->first; in ReadPartialShapesFromShapeMap()79 (*input_shapes)[in->dst_input()] = known_shape[in->src_output()]; in ReadPartialShapesFromShapeMap()
699 known_shape = original_shape is not None and None not in original_shape700 if known_shape:719 if known_shape:
1137 Output known_shape = ops::Const(s.WithOpName("known_shape"), 0.0f, {2, 2}); in TEST_F() local1163 muls_deduced_output_shape.push_back(known_shape); in TEST_F()