- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 278 for getOperation (0.24 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/optimize_graph.cc
#include "tensorflow/compiler/mlir/quantization/stablehlo/passes/optimize_graph.inc" void OptimizeGraphPass::runOnOperation() { RewritePatternSet patterns(&getContext()); populateWithGenerated(patterns); auto func = getOperation(); if (failed(applyPatternsAndFoldGreedily(func, std::move(patterns)))) { signalPassFailure(); } } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Dec 21 20:11:52 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc
replicate_op.GetReplicaOperandForBlockArgument(block_arg, /*replica=*/0)); shape_op.replaceAllUsesWith(new_shape_op.getOperation()); shape_op.erase(); } } // Checks if op and inner op operands are all replicate invariant. bool IsOpReplicateInvariant(Region* replicate_region, Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/check_accepted_ops_pass.cc
private: std::vector<std::string> accepted_dialects_; std::vector<std::string> optional_accepted_dialects_; }; } // namespace void CheckAcceptedOpsPass::runOnOperation() { getOperation()->walk([&](Operation *op) { auto dialect_name = op->getDialect()->getNamespace(); auto op_name = op->getName().stripDialect(); if (IsAcceptedOp(dialect_name, op_name, accepted_dialects_)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 24 21:06:11 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/optimize.cc
}; #include "tensorflow/compiler/mlir/quantization/tensorflow/passes/optimize.inc" void OptimizePass::runOnOperation() { RewritePatternSet patterns(&getContext()); populateWithGenerated(patterns); auto func = getOperation(); if (failed(applyPatternsAndFoldGreedily(func, std::move(patterns)))) { signalPassFailure(); } } } // namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateOptimizePass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/testing/test_pre_calibration_component.cc
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestPreCalibrationComponentPass) private: void runOnOperation() override; }; void TestPreCalibrationComponentPass::runOnOperation() { ModuleOp module_op = getOperation(); MLIRContext& ctx = getContext(); // Simply runs the PreCalibrationComponent with a default configuration. PreCalibrationComponent component(&ctx); QuantizationConfig quantization_config{};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 21:41:08 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/xla_call_module_to_call.cc
rewriter.replaceOpWithNewOp<func::CallOp>(op, entry_func_op, op.getArgs()); return success(); } }; void XlaCallModuleToCallPass::runOnOperation() { ModuleOp module_op = getOperation(); MLIRContext* ctx = module_op.getContext(); RewritePatternSet patterns(&getContext()); patterns.add<XlaCallModuleOpToCallOp>(ctx); if (failed(applyPatternsAndFoldGreedily(module_op, std::move(patterns)))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 20:02:00 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/metrics/error_collector_inst_test.cc
} public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(MockSuccessPass) explicit MockSuccessPass() = default; private: void runOnOperation() override { getOperation().walk([](Operation* nestedOp) { nestedOp->emitError() << "Error at " << nestedOp->getName().getStringRef().str() << " op"; }); }; }; // MockFailurePass reports errors and fails.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 01:48:36 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/verify_no_outside_compilation_markers_pass.cc
bool HasXlaOutsideCompilationMarker(Operation& op) { return op.getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr; } void VerifyNoOutsideCompilationMarkersPass::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { if (IsDeviceClusterOp(*op) && HasChildLaunchDeviceOp(*op)) { std::string launch_error =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 22 19:52:08 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_variables.cc
: public impl::LegalizeVariablesPassBase<LegalizeVariablesPass> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LegalizeVariablesPass) void runOnOperation() override { auto module = getOperation(); // If TFLite variable legalization is not allowed, then we skip this pass. if (auto legalize_tfl_variables_attr = module->getAttr(kLegalizeTflVariables)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_input_dialect_to_executor_pass.cc
std::string kClusterFuncOpName = "tf_device.cluster_func"; return op->getName().getStringRef().str() == kClusterFuncOpName; } void VerifyInputDialectToExecutorPass::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { if (!tensorflow::tf2xla::internal::IsInBridgeAcceptableDialects(op)) { std::string error = "op is in dialect " +
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 08 16:32:56 UTC 2023 - 2.7K bytes - Viewed (0)