Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for NodeBuilder (0.29 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top