- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 153 for eraseOp (0.55 sec)
-
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
fused_ln_lstm_func_ = createLstmCompositeFunc(builder_.get(), true, false); } void TearDown() override { fused_lstm_func_.erase(); fused_lstm_func_cifg_.erase(); fused_ln_lstm_func_.erase(); builder_.reset(); } func::FuncOp fused_lstm_func_; func::FuncOp fused_lstm_func_cifg_; func::FuncOp fused_ln_lstm_func_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
rewriter.create<OutfeedOp>(op.getLoc(), token_type, op.getInputs(), token, /*outfeed_config=*/rewriter.getStringAttr("")); rewriter.eraseOp(op); return success(); } }; // Converts tf.TopKV2 to chlo.top_k. class ConvertTopKV2Op : public OpRewritePattern<TF::TopKV2Op> { public: using OpRewritePattern::OpRewritePattern;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc
OpBuilder* builder) { builder->create<func::ReturnOp>(cluster_return_op.getLoc(), cluster_return_op.getOperands()); cluster_return_op.erase(); } // Builds a function that outlines region attached to cluster_op or launch_op, // and inserts built function into given module. template <typename ClusterOrLaunchOp>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
builder->create<tf_device::ReturnOp>(old_terminator->getLoc(), merged_execute_launch.getResults()); old_terminator->erase(); // Remove the original TPUExecute op. execute_launch.erase(); // Move all regions from old parallel_execute to new parallel_execute. for (auto region : llvm::zip(new_parallel_execute_op->getRegions(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc
!cond.getArgument(i).use_empty()) { explicitly_consumed_ids.push_back(i); } } // Empty consumed_element_ids implies none of results is used. if (explicitly_consumed_ids.empty()) { while_op.erase(); return true; } // If every element is consumed, one can't reduce any operand. if (explicitly_consumed_ids.size() == n) { return false; } // Build the dependency graph.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_device_copy_conversion.cc
StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); if (should_fold_op_func(op.getOperand(), op_device)) { op.replaceAllUsesWith(op.getOperand()); op.erase(); } return WalkResult::advance(); }); func_op.walk([&should_fold_op_func](TF::IdentityNOp op) { StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); bool should_fold = llvm::all_of(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.h
StringPiece func_name); // Runs the decompose passes on the user_module. Status DecomposeGraph(mlir::ModuleOp user_module); // Erases the tfr_module created. void Destroy(); private: mlir::ModuleOp tfr_module_; mlir::PassManager pm_; GraphExportConfig export_confs_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 11:12:54 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc
return std::get<TF::MlirLocalVarOp>(op).getResource(); } } void LocalVarOp_erase(LocalVarOp &op) { if (auto var_handle_op = std::get_if<TF::VarHandleOp>(&op)) { var_handle_op->erase(); } else { std::get<TF::MlirLocalVarOp>(op).erase(); } } std::optional<LocalVarOp> IsLocalVarOp(Operation &op) { if (TF::MlirLocalVarOp mlir_local_var_op = llvm::dyn_cast<TF::MlirLocalVarOp>(&op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc
block.eraseArgument(argument_index_to_remove); argument_types.erase(argument_types.begin() + argument_index_to_remove); } // Rewrite return if there are variable writes. const int return_operands_size = return_operands.size(); if (return_operands_size > num_results_before) { builder.create<func::ReturnOp>(return_op.getLoc(), return_operands); return_op.erase(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
} // Clean up unused asset ops. for (auto asset : llvm::make_early_inc_range(module.getOps<AssetOp>())) { if (symbol_table.symbolKnownUseEmpty(asset, module)) { asset.erase(); } } } private: // Replaces bounded-input arguments of the function with constant ops in the // body and removes the arguments.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.2K bytes - Viewed (0)