Searched refs:keep_nested_shapes (Results 1 – 2 of 2) sorted by relevance
215 bool keep_nested_shapes,
133 const tensorflow::FunctionDef* function_def, bool keep_nested_shapes, in InferShapesForFunction() argument174 if (keep_nested_shapes && inference_status.ok()) { in InferShapesForFunction()