- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for node_builder1 (0.11 sec)
-
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
// 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); Node* node = options.FinalizeBuilder(&node_builder); // Add any Enter nodes required to bring the constant to the correct control // flow frame.
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/mark_for_compilation_pass_test.cc
.WithAttr("dtypes", std::move(dtypes)); if (opts.HaveError()) { return nullptr; } NodeBuilder node_builder(name, "Stage", opts.op_registry()); node_builder.Input(values); return opts.FinalizeBuilder(&node_builder); } } // namespace TEST(XlaCompilationTest, StagePipelinePreservedByClusterScopingPass) {
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/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)