- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for InferShapeForFunction (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.h
// InferShapeForFunction. FailureOr<bool> InferModuleShape(ModuleOp module, int64_t max_iterations = 10, ArrayRef<TypeID> ops_to_skip = {}, ArrayRef<ArrayRef<int64_t>> input_shapes = {}); // Given a tensorflow NodeShape string, returns a vector of argument shapes // that can be used with InferShapeForFunction.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
region->getParentOp()->emitWarning() << "shape inference did not reach stable state after " << max_iterations << " iterations"; } return !changed; } static FailureOr<bool> InferShapeForFunction(ShapeInference& context, func::FuncOp func, int64_t max_iterations) { FailureOr<bool> failure_or_converged =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
} } } auto main_func = module.lookupSymbol<mlir::func::FuncOp>("main"); mlir::StatusScopedDiagnosticHandler error_handler(module.getContext()); mlir::LogicalResult result = mlir::TF::InferShapeForFunction( main_func, arg_shapes_copy, producer_version); if (failed(result)) { return error_handler.Combine( errors::Internal("MLIR Shape refinement failed")); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0)