- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 69 for func_20 (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_inline_tpu_island.cc
Block &nested_block = nested_module->getRegion(0).front(); for (func::FuncOp func_op : llvm::make_early_inc_range(nested_block.getOps<func::FuncOp>())) { if (!symbol_table.lookupSymbolIn(getOperation(), func_op.getName())) { nested_block.getOperations().remove(func_op.getOperation()); symbol_table.insert(func_op.getOperation()); } } nested_module->erase(); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/test_cost_analysis_pass.cc
return "Add remarks based on cost analysis for testing purpose."; } void runOnOperation() override { const auto& cost_analysis = getAnalysis<CostAnalysis>(); auto func_op = getOperation(); for (auto& op : func_op.front()) { op.emitRemark() << "Cost: " << cost_analysis.GetCost(&op); } } }; static mlir::PassRegistration<TestCostAnalysis> pass; } // namespace tfrt_compiler
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 23:46:29 UTC 2023 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize_compile_and_replicate_attributes.cc
void runOnOperation() override; }; void CanonicalizeCompileAndReplicateAttributesPass::runOnOperation() { func::FuncOp func_op = getOperation(); ModuleOp module_op = func_op->getParentOfType<ModuleOp>(); mlir::OpBuilder builder(module_op.getContext()); auto walk_result = func_op->walk([&](mlir::Operation* op) { // Convert `_tpu_replicate`. if (op->hasAttr(TF::kTpuReplicateAttr)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/common/subgraph.h
// Take the following example: // // call @func_1_CPU {tac.interface_name = "func_1"} // // "func_1" is the interface name where "func_1_cpu" is the real implementation // we can have multiple FuncOps like "func_1_cpu" and "func_1_gpu" and they // both implement "func_1". // // The attribute on the FuncOp means what it actually implements while the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 28 05:18:47 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_restore_op.cc
kTfSavedModelIndexPathAttr, builder.getArrayAttr({file_prefix_attr}))}); const int insert_idx = func_op.getNumArguments(); func_op.insertArgument(insert_idx, /*argType=*/filename_op_type, arg_attrs, NameLoc::get(file_prefix_attr)); return func_op.getArgument(insert_idx); } // Creates a 1D string array constant for "tensor_names" input of `RestoreV2`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Mar 12 06:02:20 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/func.cc
using ::tensorflow::kDefaultServingSignatureDefKey; using ::tensorflow::kImportModelDefaultGraphFuncName; // Returns true iff the function's symbol is public. bool IsPublicFuncOp(func::FuncOp func_op) { return SymbolTable::getSymbolVisibility(&*func_op) == SymbolTable::Visibility::Public; } } // namespace func::FuncOp FindMainFuncOp(ModuleOp module_op) { if (const auto main_func_op = module_op.lookupSymbol<func::FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 19 06:55:11 UTC 2024 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/strip_noinline_attribute.cc
public: // void runOnOperation() override; void runOnOperation() override { // Strip the "tf._noinline" attribute from top-level functions. for (auto func_op : getOperation().getOps<func::FuncOp>()) func_op->removeAttr("tf._noinline"); } }; } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateStripNoinlineAttributePass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc
OpPassManager &func_pm = pm.nest<func::FuncOp>(); // First operates on the executor dialect: // - remove dead islands. // - fuse islands as much as possible. // - materialize the eventual "pass-through" ops by inlining their content. func_pm.addPass(tf_executor::CreateTFExecutorGraphPruningPass()); func_pm.addPass(tf_executor::CreateTFExecutorIslandCoarseningPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/inference/inference_metrics_pass.cc
public: void runOnOperation() override; }; void InferenceMetricsPass::runOnOperation() { bool has_tpu_partitioned_call = false; ModuleOp module = getOperation(); for (auto func_op : module.getOps<func::FuncOp>()) { func_op->walk( [&](TF::TPUPartitionedCallOp op) { has_tpu_partitioned_call = true; }); if (has_tpu_partitioned_call) break; } std::string has_tpu_partitioned_call_str =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 05 21:24:51 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/tensor_array_side_effect_analysis.h
explicit TensorArraySideEffectAnalysis(mlir::ModuleOp module); // Return if the function contains only non-side-effecting ops or TensorArray // ops. bool HasAtMostTensorArrayEffect(mlir::func::FuncOp func_op) const { return set_.count(func_op) > 0; } private: llvm::DenseSet<mlir::func::FuncOp> set_; }; } // namespace tfrt_compiler } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 30 02:12:49 UTC 2022 - 2K bytes - Viewed (0)