- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for mutable_attr (0.13 sec)
-
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/compiler/jit/encapsulate_subgraphs_pass_test.cc
// `PerformStaticShapeInferenceBeforeEncapsulation`. for (FunctionDef& fdef : *library->mutable_function()) { for (NodeDef& node_def : *fdef.mutable_node_def()) { node_def.mutable_attr()->erase("_xla_inferred_shapes"); } } return s; } Status Encapsulate(GraphDef* graphdef, FunctionDefLibrary* library) { std::vector<string> encapsulated_functions;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0)