- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for NodeBuilder (0.29 sec)
-
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
{ GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); Node* a = ops::SourceOp("UncompilableNullary", builder.opts().WithName("A")); // Builds a Symbolic gradient for Supported NodeBuilder b_builder("B", "SymbolicGradient", builder.opts().op_registry()); NameAttrList b_name_attr; b_name_attr.set_name("Supported"); b_builder.Attr("f", b_name_attr);
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.cc
// Don't set any value field in the proto, since it is only going to be used // for shape inference. GraphDefBuilder::Options options(graph_out, /*status=*/nullptr); NodeBuilder node_builder(options.GetNameForOp("KnownShape"), "Const", options.op_registry()); node_builder.Attr("dtype", data_type).Attr("value", dummy_proto);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/compiler/jit/deadness_analysis_test.cc
if (!scope.ok()) return; auto _output_index = ops::AsNodeOut(scope, output_index); if (!scope.ok()) return; Node* ret; const auto unique_name = scope.GetUniqueNameForOp("_SwitchN"); auto builder = NodeBuilder(unique_name, "_SwitchN") .Input(_data) .Input(_output_index) .Attr("num_outs", num_outs); scope.UpdateBuilder(&builder);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 51.6K bytes - Viewed (0)