- Sort Score
- Result 10 results
- Languages All
Results 111 - 120 of 382 for get_operations (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/resource_analyzer_test_pass.cc
analyzer.IsPotentiallyWritten(var_handle_op.getResource()))); } }); } void ResourceAnalyzerTestPass::runOnOperation() { ModuleOp module_op = getOperation(); TF::ResourceAnalyzer resource_analyzer(module_op); PrintAnalysisResults(resource_analyzer, module_op); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateResourceAnalyzerTestPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 00:21:29 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/drop_savedmodel_semantics.cc
// third_party/tensorflow/compiler/mlir/tfrt/transforms/lower_saved_model.cc // with the original code trimmed and adapted as needed. void runOnOperation() override { auto module = getOperation(); if (!tf_saved_model::HasTfSavedModelSemantics(module)) return; // Clean up functions from tf_saved_model attributes. OpBuilder builder(module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 13 23:39:10 UTC 2022 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/insert_call_once_op.cc
InsertCallOnceOpFromSessionInitializerPass) private: void runOnOperation() override; }; void InsertCallOnceOpFromSessionInitializerPass::runOnOperation() { ModuleOp module = getOperation(); for (func::FuncOp init_func_op : tf_saved_model::GetInitializerFunctions(module)) { for (auto func : module.getOps<func::FuncOp>()) { auto dict_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 2.5K 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/tensorflow/transforms/test_cluster_ops_by_policy.cc
operands.Insert(op->getOperand(0), ValueConstraint::kShape); } return success(); } }; void TestClusteringPolicyPass::runOnOperation() { func::FuncOp func = getOperation(); ValuesConstraintSet constraints; ClusteringPolicySet policies; policies.Add<TestOpsClusteringPolicy>(); // Initialize constraints based on the return type attributes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 09:19:38 UTC 2022 - 3.2K 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/deduplicate_bound_input_bindings.cc
class DedupBoundInputBindingPass : public impl::DedupBoundInputBindingPassBase<DedupBoundInputBindingPass> { void runOnOperation() final; }; void DedupBoundInputBindingPass::runOnOperation() { func::FuncOp func = getOperation(); if (!mlir::tf_saved_model::IsExported(func)) return; llvm::SmallDenseMap<Attribute, unsigned, 8> unique_bound_inputs; llvm::BitVector arg_indices_to_erase(func.getNumArguments());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 09:25:35 UTC 2022 - 2.4K 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/extract_head_tail_outside_compilation.cc
mlir::tf_device::LaunchOp launch = CreateLaunchForBlock( builder, cluster, /*before=*/false, launch_block, host_device); auto operand_not_in_launch = [&](mlir::OpOperand& operand) { return !launch.getOperation()->isProperAncestor(operand.getOwner()); }; for (auto result : llvm::zip(launch.GetBody().getTerminator()->getOperands(), launch.getResults()))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0)