- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for getSharedName (0.19 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/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/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) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc
} if (TF::VarHandleOp var_handle_op = llvm::dyn_cast<TF::VarHandleOp>(&op)) { auto ANONYMOUS_NAME = ::tensorflow::ResourceHandle::ANONYMOUS_NAME; if (var_handle_op.getSharedName() == ANONYMOUS_NAME) { return std::make_optional(LocalVarOp(var_handle_op)); } } return {}; } // Eliminate local variables that are only assigned to but never read, and thus // are dead.
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/ir/tf_ops_n_z.cc
llvm::SmallDenseMap<ResourceHandle, int64_t> &resource_handle_id_map, int64_t &next_id) { llvm::StringRef device = GetDeviceOrEmpty(getOperation()); return {GetResourceHandleValueAndIdBase(getContainer(), getSharedName(), device, getWriter(), resource_handle_id_map, next_id)}; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)