Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 27 of 27 for NodeBuilder (0.13 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/c/c_api_experimental.cc

    #include "tensorflow/core/platform/strcat.h"
    #include "tensorflow/core/protobuf/config.pb.h"
    #include "tensorflow/core/protobuf/tensorflow_server.pb.h"
    
    using tensorflow::FunctionDef;
    using tensorflow::Node;
    using tensorflow::NodeBuilder;
    using tensorflow::Status;
    using tensorflow::errors::InvalidArgument;
    
    namespace {
    typedef std::unique_ptr<TF_Function, decltype(&TF_DeleteFunction)>
        UniqueFuncPtr;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  4. tensorflow/cc/gradients/array_grad.cc

        inputs.push_back(op.input(i));
      }
    
      auto shapes = ShapeN(scope, inputs);
      const auto unique_name = scope.GetUniqueNameForOp("ConcatOffset");
      auto builder =
          ::tensorflow::NodeBuilder(unique_name, "ConcatOffset")
              .Input(::tensorflow::ops::AsNodeOut(scope, op.input(dim_index)))
              .Input(::tensorflow::ops::AsNodeOutList(scope, shapes.output));
      scope.UpdateBuilder(&builder);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 10 23:33:32 UTC 2023
    - 31.7K bytes
    - Viewed (0)
  5. 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)
  6. tensorflow/c/c_api.cc

    using tensorflow::GraphDef;
    using tensorflow::mutex_lock;
    using tensorflow::NameRangeMap;
    using tensorflow::NameRangesForNode;
    using tensorflow::NewSession;
    using tensorflow::Node;
    using tensorflow::NodeBuilder;
    using tensorflow::OpDef;
    using tensorflow::PartialTensorShape;
    using tensorflow::RunMetadata;
    using tensorflow::RunOptions;
    using tensorflow::Session;
    using tensorflow::Status;
    using tensorflow::string;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

      std::string new_node_name =
          update_inplace ? node->name()
                         : GetUniqueNodeName(node->name(), index, node_name_map);
    
      Node* placeholder_node;
      NodeBuilder builder(new_node_name, "Placeholder");
      builder.Attr("shape", shape);
      builder.Attr("dtype", dtype);
      TF_RETURN_IF_ERROR(builder.Finalize(graph_.get(), &placeholder_node));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top