Home
last modified time | relevance | path

Searched refs:InferShapeForFunction (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dshape_inference.h45 FailureOr<bool> InferShapeForFunction(func::FuncOp func,
H A Dshape_inference.cc2670 static FailureOr<bool> InferShapeForFunction(ShapeInference& context, in InferShapeForFunction() function
2683 FailureOr<bool> InferShapeForFunction(func::FuncOp func, in InferShapeForFunction() function
2690 return InferShapeForFunction(context, func, max_iterations); in InferShapeForFunction()
2761 InferShapeForFunction(context, func, max_iterations); in InferModuleShape()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dcompile_mlir_util.cc366 mlir::LogicalResult result = mlir::TF::InferShapeForFunction( in RefineShapes()