- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for mutable_attr (0.35 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
} (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr; (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr; } } TF_RETURN_IF_ERROR( SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr())); DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
} (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr; (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr; } } TF_RETURN_IF_ERROR( SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr())); DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/cc/tools/freeze_saved_model.cc
NodeDef* const_node) { const_node->set_name(variable_node.name()); const_node->set_op("Const"); (*const_node->mutable_attr())["dtype"] = variable_node.attr().at("dtype"); variable_value.AsProtoTensorContent( (*const_node->mutable_attr())["value"].mutable_tensor()); } // Converts a ReadVariableOp NodeDef to an Identity NodeDef. void ConvertReadVariableOpToIdentity(const NodeDef& node,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 11 08:05:36 UTC 2023 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; std::vector<string> shape_inference_graphs; bool has_outside_compilation; NameAttrList name_attrs; name_attrs.set_name("cluster"); *name_attrs.mutable_attr() = attrs; TF_CHECK_OK(ExtractOutsideCompilationTest( "_xla", "_oc", "cluster", name_attrs, "cluster_rewritten", "host_graph", host_compute_core, &fld, &shape_inference_graphs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
// executed on TPU or GPU devices, which do not have `Cast` registered as a // runtime OpKernel. void RemoveIdentityCast(NodeDef* node_def) { auto attr = node_def->mutable_attr(); if (node_def->op() == "Cast" && attr->contains("SrcT") && attr->contains("DstT") && attr->at("SrcT").type() == attr->at("DstT").type() && attr->contains("Truncate") && !attr->at("Truncate").b()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cluster_util.cc
std::find(node.output_types().begin(), node.output_types().end(), DT_RESOURCE) != node.output_types().end(); } void RemoveFromXlaCluster(NodeDef* node_def) { node_def->mutable_attr()->erase(kXlaClusterAttr); } void RemoveFromXlaCluster(Node* node) { node->ClearAttr(kXlaClusterAttr); } namespace { typedef xla_config_registry::XlaGlobalJitLevel XlaGlobalJitLevel;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler.h
NameAttrList name; name.set_name(def.op()); *name.mutable_attr() = def.attr(); // Remove the "_class" attribute from the attribute set used to create the // compilation cache key. This attribute is information for the colocator // and causes false uniqueness between nodes. name.mutable_attr()->erase("_class"); return CompileImpl(compile_options, options, name, args, CompileScope::kOp,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 22.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
TF_RETURN_IF_ERROR(ConvertAttributes(attr.getAttrs().getValue(), /*attrs_to_ignore=*/{}, remove_ref_type, value->mutable_func()->mutable_attr())); return absl::OkStatus(); } Status ConvertAttribute(const mlir::StringAttr& attr, AttrValue* value) { absl::string_view attr_value(attr.getValue().data(), attr.getValue().size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
NodeDef* node_def) { const NameAttrList* name_attr; TF_RETURN_IF_ERROR(GetNodeAttr(node.attrs(), attr_name, &name_attr)); node_def->set_op(name_attr->name()); *(node_def->mutable_attr()) = name_attr->attr(); return absl::OkStatus(); } absl::StatusOr<std::vector<NodeDef>> MakeCallNodesFromAttribute( const Node& node, absl::string_view attr_name, absl::string_view call_name) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
return; } auto fdef_or = func->record->mutable_fdef(); if (!fdef_or.ok()) { status->status = fdef_or.status(); return; } (*(fdef_or.value()->mutable_attr()))[string(attr_name)] = attr_value; status->status = absl::OkStatus(); } void TF_FunctionGetAttrValueProto(TF_Function* func, const char* attr_name,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 13.6K bytes - Viewed (0)