- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for getSharedName (0.24 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
StringRef shared_name = GetSharedName(hash_table); 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) {
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/tensorflow/utils/session_utils.cc
// then fallback to shared_name attribute. if (auto loc = mlir::dyn_cast<NameLoc>(var_handle_op->getLoc())) return loc.getName().str(); return var_handle_op.getSharedName().str(); } std::vector<std::string> GetVariableNames( llvm::ArrayRef<TF::VarHandleOp> var_handle_ops) { std::vector<std::string> names; names.reserve(var_handle_ops.size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/save_variables.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:36:55 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_duplicate_resource_ops.cc
TF::MutableHashTableV2Op>(resource_op)) { return resource_op; } return nullptr; } // Returns the `shared_name` attribute value if exists. If not, returns an // empty string. StringRef GetSharedName(Operation* op) { if (!op->hasAttrOfType<StringAttr>(kSharedNameAttr)) return ""; return op->getAttrOfType<StringAttr>(kSharedNameAttr).getValue(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 26 04:26:16 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/localize_var_handles.cc
} else if (auto handle = llvm::dyn_cast<TF::VarHandleOp>(source)) { container = handle.getContainer(); shared_name = handle.getSharedName(); } else if (auto it = llvm::dyn_cast<TF::IteratorOp>(source)) { container = it.getContainer(); shared_name = it.getSharedName(); } else { // Can't happen, as long as this file and resource_dataflow.cc are in sync. return; } OpBuilder builder(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize_variables.cc
{mlir::cast<TensorType>(ref_qtype)}, builder.getContext())); auto new_var_handle_op = builder.create<VarHandleOp>( var_handle_op.getLoc(), output_type, var_handle_op.getContainer(), var_handle_op.getSharedName()); var_handle_op->replaceAllUsesWith(new_var_handle_op); var_handle_op.erase(); } } void QuantizeVariablesPass::runOnOperation() { ResourceIdMap resource_id_map;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_analyzer_test_pass.cc
if (auto attr = var_handle_op->getAttrOfType<StringAttr>("device")) { resource_key.device = attr.getValue(); } resource_key.container = var_handle_op.getContainer(); resource_key.shared_name = var_handle_op.getSharedName(); return resource_key; } // Prints the analysis result for each resource ops found in `module_op` for // testing purposes. void PrintAnalysisResults(const TF::ResourceAnalyzer& analyzer,
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/quantization/tensorflow/passes/insert_restore_op.cc
SmallVector<Type> tensor_types{}; SmallVector<std::string> tensor_names{}; for (auto var_handle_op : target_var_handle_ops) { tensor_names.emplace_back(var_handle_op.getSharedName().str()); // Location must be set to the same name as the shared name. The Location is // later tranlated to the op's name when exported to `GraphDef`. This is
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/tensorflow/ir/host_runtime/tfrt_ops.cc
resource_vec.push_back(GetResourceHandleValueAndIdBase( mlir::cast<mlir::StringAttr>(getContainer()[index]).getValue(), mlir::cast<mlir::StringAttr>(getSharedName()[index]).getValue(), device, getResults()[index], resource_handle_id_map, next_id)); } } return resource_vec; } LogicalResult _TfrtGetResourceOp::verify() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
// native resource design is based on integer keys to identify the // corresponding resource objects. auto table_id = static_cast<int32_t>(::llvm::hash_value(hashtable_op.getSharedName())); auto key_dtype = hashtable_op.getKeyDtype(); auto value_dtype = hashtable_op.getValueDtype(); rewriter.replaceOpWithNewOp<TFL::HashtableOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0)