- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 27 for mutable_attr (0.13 sec)
-
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/device_compilation_cluster_signature_test.cc
} void BM_BuildSignature(::testing::benchmark::State& state) { const int n_args = state.range(0); NameAttrList fn; fn.set_name("afunction"); for (int i = 0; i < n_args; i++) { (*fn.mutable_attr())[absl::StrCat("T", i)].set_type(DT_FLOAT); } std::vector<XlaCompiler::Argument> args(n_args); for (int i = 0; i < n_args; i++) { args[i].kind = (((i % 3) == 0) ? XlaCompiler::Argument::kConstant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
NameAttrList host_then_branch, host_else_branch; host_then_branch.set_name(then_branch_host_func_name); (*host_then_branch.mutable_attr())["_device_ordinal"] = device_ordinal_value; host_else_branch.set_name(else_branch_host_func_name); (*host_else_branch.mutable_attr())["_device_ordinal"] = device_ordinal_value; if_builder.Attr("then_branch", host_then_branch); if_builder.Attr("else_branch", host_else_branch);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/c/kernels/summary_op_test.cc
Status status; NodeDef def; def.set_op("ScalarSummary"); def.set_device(DEVICE_CPU); AttrValue valuesTypeAttr; SetAttrValue(values->dtype(), &valuesTypeAttr); (*def.mutable_attr())["T"] = valuesTypeAttr; def.add_input(strings::StrCat("input1: ", DataTypeString(tags->dtype()))); def.add_input(strings::StrCat("input2: ", DataTypeString(values->dtype()))); std::unique_ptr<OpKernel> kernel =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 6.7K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
def.add_input("input3"); def.add_input("input3"); AttrValue v0; v0.set_type(DataType::DT_INT32); v0.set_i(3); (*def.mutable_attr())["NumInput3"] = v0; AttrValue v1; v1.set_type(DataType::DT_FLOAT); (*def.mutable_attr())["SomeDataTypeAttr"] = v1; return CreateOpKernel(DeviceType(device_name), nullptr, nullptr, def, 1, status); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K 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)