- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 19 for getPrivate (0.2 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
auto save_func = builder.create<func::FuncOp>( NameLoc::get(builder.getStringAttr(kTfQuantSaveFuncName)), /*sym_name=*/kTfQuantSaveFuncName, func_type); save_func.addEntryBlock(); save_func.setPrivate(); return save_func; } // Creates a save function that contains the `TF::SaveV2Op` for the variables in // `var_handle_ops`. The `var_handle_ops` are cloned into the new function and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
ModuleOp to) { SymbolTable &to_symbol_table = symbol_tables.getSymbolTable(to); for (auto func : llvm::make_early_inc_range(from.getOps<func::FuncOp>())) { func->remove(); func.setPrivate(); to_symbol_table.insert(func); } } void CopyStablehloModuleAttrs(ModuleOp stablehlo_module, XlaCallModuleOp op) { op->setAttr(kStablehloModuleAttrsAttrName,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
globalTensor.getLoc(), name, globalTensor.getType(), globalTensor.getIsMutable(), initial_value, /*visibility=*/globalBuilder.getStringAttr("private")); variableOp.setPrivate(); } SymbolTable syms(module); for (auto func : module.getOps<func::FuncOp>()) { if (!tf_saved_model::IsExported(func)) { continue; } bool success = true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc
func::FuncOp::create(op.getLoc(), func_name, func_type); // This function is not externally visible and marking it private would allow // symbol-dce pass to remove it when it is not referenced anymore. outlined_func.setPrivate(); // Create function body. Block* outlined_func_block = outlined_func.addEntryBlock(); // Replace uses of live-in values within cluster_op region with function // arguments.
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/quantization/tensorflow/passes/insert_quantized_functions.cc
if (symbol_table.lookup(func.getSymName()) != nullptr) continue; // Set the function to private and insert to the module. func::FuncOp new_func = func.clone(); new_func.setPrivate(); symbol_table.insert(new_func); // For consistency, we require all quantized composite function to have // the "tf_quant.quantized_ops" attribute.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
} b.create<func::ReturnOp>(yield_op->getLoc(), args); yield_op->erase(); SymbolTable(region.getParentOfType<ModuleOp>()).insert(outlined_func); outlined_func.setPrivate(); return outlined_func; } // Replace region with call to outline function. void ReplaceRegionWithCall(StringRef name, Region& region, bool passthru_extra_args, int num_loop_carried,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
auto name = rewriter.getStringAttr("gelu_decomp"); func::FuncOp new_func = rewriter.create<func::FuncOp>( insertion_point->front().getLoc(), name, ftype); new_func.setPrivate(); new_func.addEntryBlock(); rewriter.setInsertionPointToStart(&new_func.getBody().front()); auto one_val = DenseElementsAttr::get(type, kOne); auto one_cst =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
func::FuncOp cond = func::FuncOp::create(loc, "new_while_cond", cond_func_type); func::FuncOp body = func::FuncOp::create(loc, "new_while_body", body_func_type); cond.setPrivate(); body.setPrivate(); symbol_table.insert(cond); symbol_table.insert(body); OpBuilder cond_builder = OpBuilder::atBlockBegin(cond.addEntryBlock());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc
auto func_uses = SymbolTable::getSymbolUses(func, &module.getBodyRegion()); if (func_uses.has_value() && llvm::hasSingleElement(func_uses.value())) return func; func::FuncOp cloned = func.clone(); cloned.setPrivate(); cloned.setName( StringAttr::get(func.getContext(), func.getName().str() + "_lifted")); SymbolTable(module).insert(cloned); return cloned; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0)