Searched refs:kOutputShapes (Results 1 – 2 of 2) sorted by relevance
67 ABSL_CONST_INIT const char kOutputShapes[] = "_output_shape_vector"; variable
1331 attr.count(kOutputShapes) == 0) in UpdateOutputShapesUsingAnnotatedInformation()1335 int output_size = attr.at(kOutputShapes).list().shape_size(); in UpdateOutputShapesUsingAnnotatedInformation()1351 attr.at(kOutputShapes).list().shape(shape_index); in UpdateOutputShapesUsingAnnotatedInformation()