- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 78 for getOps (0.1 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
return failure(); } // Fetch all VarHandleOp. llvm::StringSet<> variable_names; llvm::SmallVector<TF::VarHandleOp, 4> var_ops; for (auto func_op : module.getOps<func::FuncOp>()) { for (auto var_handle_op : func_op.getOps<TF::VarHandleOp>()) { auto variable_name = GetVariableName(var_handle_op); if (variable_names.count(variable_name)) continue; var_ops.emplace_back(var_handle_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K 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/tf_saved_model_asset_sinking_pass.cc
mlir::cast<FlatSymbolRefAttr>(initializer).getValue()); RewriteFunction(symbol_table, func); } // 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
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.2K 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.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 16 01:49:07 UTC 2022 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops_pass.cc
// Iterates over all functions within the module to (1) create caller-callee // map, and (2) initialize function worklist with functions referenced from // device cluster ops. for (auto func : module.getOps<func::FuncOp>()) { for (auto user : symbol_map.getUsers(func)) { // Populate caller-callee map. if (func::FuncOp caller = user->getParentOfType<func::FuncOp>()) caller_callee_map[caller].insert(func);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 08 20:01:13 UTC 2023 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/lift_quantizable_spots_as_functions.cc
const FunctionNameMatcher matcher(spec.matcher().function_name()); // Iterate over all XlaCallModuleOp in all FuncOps. for (auto func : module_op.getOps<func::FuncOp>()) { for (auto xla_call_module_op : func.getOps<TF::XlaCallModuleOp>()) { if (!matcher.Match(xla_call_module_op)) continue; // Set the text representation of `Method` to matched // `TF::XlaCallModuleOp`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
if (shared_name.empty()) return false; for (func::FuncOp init_func_op : tf_saved_model::GetInitializerFunctions(module_op)) { for (Operation& op : init_func_op.getBody().getOps()) { StringRef other_shared_name = GetSharedName(&op); if (IsHashTableOp(&op) && other_shared_name == shared_name) { return true; } } } return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/insert_call_once_op.cc
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 = func->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function"); if (!dict_attr) continue; OpBuilder builder(func.getContext());
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/tensorflow/transforms/lift_variables.cc
mlir::Builder builder(context); StringAttr resource_name_id = builder.getStringAttr(kResourceNameArgAttr); SmallSet<StringRef, 4> resource_names; for (func::FuncOp func : module.getOps<func::FuncOp>()) { for (int i = 0, e = func.getNumArguments(); i < e; ++i) { auto resource_arg = func.getArgAttrOfType<StringAttr>(i, kResourceNameArgAttr); if (!resource_arg) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0)