- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for max_iterations_ (0.26 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference_pass.cc
} auto failure_or_converged = InferModuleShape( getOperation(), max_iterations_, /*ops_to_skip=*/{}, input_shapes_); if (failed(failure_or_converged)) return signalPassFailure(); if (!failure_or_converged.value()) { getOperation().emitError() << "shape inference pass did not reach convergence after " << max_iterations_; return signalPassFailure(); } } private:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops_pass.cc
// `max_iterations` are reached. Returns failure if resource ops decomposition // does not converge after `max_iterations`. // TODO(prakalps): This can be useful to a lot of other passes in bridge. // Extract out as a separate utility. LogicalResult ApplyPatternsLocallyUntilConverged( Operation* op_with_regions, FrozenRewritePatternSet& patterns, int max_iterations) { bool changed = true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 08 20:01:13 UTC 2023 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_ops.td
The Pmap executes body function in parallel for all ranges up to $max_iterations. The pseudo code: for(int i = 0; i < $max_iterations; i++) { body_fn(MlrtFture($tensor_list_or_flow_in[i]), MlrtPromise($tensor_list_or_flow_in[i+1]), i, i, $invariant_args); } return $tensor_list_or_flow_in[$max_iterations] }]; let arguments = (ins TF_Tensor:$max_iterations,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:35:32 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.h
// reached convergence, false otherwise. // If input shapes are provided, first refines the `main` function using // InferShapeForFunction. FailureOr<bool> InferModuleShape(ModuleOp module, int64_t max_iterations = 10, ArrayRef<TypeID> ops_to_skip = {}, ArrayRef<ArrayRef<int64_t>> input_shapes = {});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 3.5K bytes - Viewed (0)