Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 18 for NodeBuilder (0.16 sec)

  1. testing/internal-integ-testing/src/main/groovy/org/gradle/integtests/fixtures/resolve/ResolveTestFixture.groovy

            }
        }
    
        static class GraphBuilder {
            private final Map<String, NodeBuilder> nodes = [:]
            private NodeBuilder root
    
            final Set<String> virtualConfigurations = []
    
            Collection<NodeBuilder> getNodes() {
                return nodes.values()
            }
    
            Collection<NodeBuilder> getNodesWithoutRoot() {
                def nodes = new HashSet<>()
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Thu Apr 04 07:21:38 UTC 2024
    - 37.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc

      Scope root = Scope::NewRootScope().ExitOnError();
    
      Output a = ops::_Arg(root.WithOpName("A"), DT_RESOURCE, 0);
      std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())});
    
      Node* call;
      NameAttrList f_name_attr;
      f_name_attr.set_name(fd.signature().name());
      TF_ASSERT_OK(
          NodeBuilder("B", "StatefulPartitionedCall", &root.graph()->flib_def())
              .Input(inputs)
              .Attr("Tin", {DT_RESOURCE})
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 17 19:51:50 UTC 2024
    - 10.3K bytes
    - Viewed (0)
  3. tensorflow/cc/ops/while_loop.cc

      // not-yet-created next iteration nodes). Use the underlying NodeBuilder API
      // directly to create the back edge.
      NodeBuilder::NodeOut enter_input(enter_output.node(), enter_output.index());
    
      const int next_output_index = 0;
      DataType dtype = enter_output.node()->output_type(0);
      NodeBuilder::NodeOut next_input(NextIterationName(scope, loop_var_idx),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 01:01:21 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/utils/dump_graph_test.cc

        return errors::Unimplemented("Stream not seekable");
      }
    
     private:
      string& str_;
    };
    
    TEST(Dump, TextualIrToFileSuccess) {
      Graph graph(OpRegistry::Global());
      Node* node;
      TF_CHECK_OK(NodeBuilder("A", "NoOp").Finalize(&graph, &node));
    
      setenv("TF_DUMP_GRAPH_PREFIX", testing::TmpDir().c_str(), 1);
      UseMlirForGraphDump(MlirDumpConfig());
      string ret = DumpGraphToFile("tir", graph);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 03:47:51 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/xla_compile_util.cc

      // dependency edge to the _SOURCE node.
      for (int64_t i = 0, end = args.size(); i < end; ++i) {
        Node* node;
        string arg_name = absl::StrCat("_arg", i);
        Status status =
            NodeBuilder(arg_name, FunctionLibraryDefinition::kArgOp)
                .ControlInput(graph->source_node())
                .Attr("T", args[i].kind == XlaArgument::kResource ? DT_RESOURCE
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

        Node* b = Input(b2.opts().WithName("B"));
        Node* control = Input(b2.opts().WithName("Control"));
    
        NodeBuilder nb("F1", "F1", lib_def.get());
        nb.Input(a).ControlInput(control);
        Node* call1 = b2.opts().FinalizeBuilder(&nb);
    
        NodeBuilder nb2("F2", "F2", lib_def.get());
        nb2.Input(b).Input(call1).ControlInput(control);
        Node* call2 = b2.opts().FinalizeBuilder(&nb2);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
  7. tensorflow/cc/framework/scope.h

      // START_SKIP_DOXYGEN
    
      /// Update the builder with properties accumulated in this scope. Does not set
      /// status().
      // TODO(skyewm): NodeBuilder is not part of public API
      void UpdateBuilder(NodeBuilder* builder) const;
      // END_SKIP_DOXYGEN
    
      CompositeOpScopes GetCompositeOpScopes(const string& composite_op_name) const;
    
      bool ok() const;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 09:08:33 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  8. tensorflow/cc/framework/cc_op_gen_util.h

        const ApiDef& api_def,
        const std::unordered_map<string, string>& inferred_input_attrs);
    
    struct OpInfo {
      // graph_op_def: The OpDef used by the runtime, has the names that
      //   must be used when calling NodeBuilder.
      // interface_op_def: The OpDef used in the interface in the generated
      //   code, with possibly overridden names and defaults.
      OpInfo(const OpDef& graph_op_def, const ApiDef& api_def,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:57:05 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  9. tensorflow/cc/framework/scope.cc

        GraphDef graph_def;
        graph()->ToGraphDef(&graph_def);
        UpdateStatus(ConvertGraphDefToGraph(opts, std::move(graph_def), g));
      }
      return *impl()->status_;
    }
    
    void Scope::UpdateBuilder(NodeBuilder* builder) const {
      std::vector<Node*> control_inputs;
      for (const auto& op : impl()->control_deps_) {
        control_inputs.push_back(op.node());
      }
      builder->ControlInputs(control_inputs);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 20.9K bytes
    - Viewed (0)
  10. platforms/software/dependency-management/src/integTest/groovy/org/gradle/integtests/resolve/api/ConfigurationMutationIntegrationTest.groovy

            failure.assertHasCause("Cannot change dependencies of dependency configuration ':conf' after it has been included in dependency resolution.")
        }
    
        void resolvedGraph(@DelegatesTo(ResolveTestFixture.NodeBuilder) Closure closure) {
            resolve.prepare()
            succeeds ":checkDeps"
    
            resolve.expectGraph {
                root(":", "org.test:root:1.1", closure)
            }
        }
    
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Thu Mar 07 19:31:56 UTC 2024
    - 14.8K bytes
    - Viewed (0)
Back to top