Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dshape_inference.h36 FailureOr<bool> InferModuleShape(ModuleOp module, int64_t max_iterations = 10);
H A Dshape_inference_pass.cc37 InferModuleShape(getOperation(), max_iterations_); in runOnOperation()
H A Dshape_inference.cc2738 FailureOr<bool> InferModuleShape(ModuleOp module, int64_t max_iterations) { in InferModuleShape() function