Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dshape_inference.h44 FailureOr<bool> InferShapeForFunction(FuncOp func,
Dshape_inference.cc1808 static FailureOr<bool> InferShapeForFunction(ShapeInference& context, in InferShapeForFunction() function
1821 FailureOr<bool> InferShapeForFunction(FuncOp func, in InferShapeForFunction() function
1828 return InferShapeForFunction(context, func, max_iterations); in InferShapeForFunction()
1899 InferShapeForFunction(context, func, max_iterations); in InferModuleShape()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc270 mlir::LogicalResult result = mlir::TF::InferShapeForFunction( in RefineShapes()