Searched refs:PropagateShapeToFunctions (Results 1 – 1 of 1) sorted by relevance
633 FailureOr<bool> PropagateShapeToFunctions(ModuleOp module,1110 FailureOr<bool> res = PropagateShapeToFunctions( in InferShapeForDatasetOpCommon()1229 PropagateShapeToFunctions(module, input_types, {f}, max_iterations); in InferShapeForReduceDataset()2180 FailureOr<bool> ShapeInference::PropagateShapeToFunctions( in PropagateShapeToFunctions() function in mlir::TF::ShapeInference2377 return PropagateShapeToFunctions( in PropagateShapeIntoAttachedFunctions()2385 return PropagateShapeToFunctions(module, case_op.input().getTypes(), in PropagateShapeIntoAttachedFunctions()2398 return PropagateShapeToFunctions( in PropagateShapeIntoAttachedFunctions()2404 return PropagateShapeToFunctions( in PropagateShapeIntoAttachedFunctions()2413 FailureOr<bool> failure_or_converged = PropagateShapeToFunctions( in PropagateShapeIntoAttachedFunctions()2430 return PropagateShapeToFunctions(module, types, {func}, max_iterations); in PropagateShapeIntoAttachedFunctions()