- Sort Score
- Result 10 results
- Languages All
Results 21 - 28 of 28 for mutable_attr (0.15 sec)
-
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) -
tensorflow/compiler/jit/compilability_check_util_test.cc
/*out_def=*/{"res:float"}, /*attr_def=*/{}, /*node_def=*/{{{"t0"}, "Identity", {"x"}, {{"T", DT_FLOAT}}}}, /*ret_def*/ {{"res", "t0:output"}}); (*identity_func.mutable_attr())[kXlaMustCompileAttr] = true_attribute; FunctionDef call_identity = FunctionDefHelper::Create( "CallIdentity", /*in_def=*/{"x:float"}, /*out_def=*/{"z:float"}, /*attr_def=*/{},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 22.3K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass.cc
num_resource_inputs), std::back_inserter(result->resource_inputs), IncomingEdgeAsOutput); result->function.set_name(n->type_string()); *result->function.mutable_attr() = n->def().attr(); return absl::OkStatus(); } Status CopyIncomingControlEdges(Graph* g, Node* from, Node* to) { for (const Edge* e : from->in_edges()) { if (e->IsControlEdge()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
node_def.set_name(op->Name()); node_def.set_op(op->Name()); for (int i = 0; i < num_inputs; ++i) { node_def.add_input("dummy_input"); } OperationFromInterface(op)->Attrs().FillAttrValueMap(node_def.mutable_attr()); const tensorflow::OpRegistrationData* op_reg_data; status->status = tensorflow::OpRegistry::Global()->LookUp(node_def.op(), &op_reg_data); if (!status->status.ok()) return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
// an attribute. void AnnotateEagerRuntimeConstructionContext( tensorflow::FunctionDef& function_def) { tensorflow::AttrValue value; SetAttrValue("kEagerRuntime", &value); (*function_def.mutable_attr())["_construction_context"] = value; } } // namespace extern "C" { TFE_ContextOptions* TFE_NewContextOptions() { return new TFE_ContextOptions; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
FunctionDef noinline = compilable; noinline.mutable_signature()->set_name("NoInlineFn"); AddAttr("_noinline", static_cast<bool>(true), noinline.mutable_attr()); FunctionDefLibrary flib; *flib.add_function() = compilable; *flib.add_function() = uncompilable; *flib.add_function() = noinline; FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K 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) -
tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.cc
LOG_FIRST_N(INFO, 1) << "Run Node Expansion Passes"; // Get the FunctionDef and insert that into the context const NodeDef& ndef = orig_op->MutableAttrs()->BuildNodeDef(); auto& ctx = orig_op->EagerContext(); Fprint128 cache_key = orig_op->MutableAttrs()->CacheKey(orig_op->DeviceName()); // Include soft placement policy in cache key since the placement strategy // can change and thus affect which kernel is picked.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Feb 25 16:22:36 UTC 2024 - 3.8K bytes - Viewed (0)