- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 29 for node_builder1 (0.17 sec)
-
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
Node* s1 = Sequencer( b2.opts().WithName("F1_sequencer").WithControlInputs({recv1, send1}), "F1"); NodeBuilder node_builder1("F1", "F1", lib_def.get()); node_builder1.Input(a).Input(b); Node* call1 = b2.opts().WithControlInputs({s1, b}).FinalizeBuilder(&node_builder1); Node* key_constant2 = KeyPlaceholder("F2", b2.opts().WithName("F2_key_placeholder"));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dump_graph_test.cc
==============================================================================*/ #include "tensorflow/compiler/mlir/tensorflow/utils/dump_graph.h" #include "tensorflow/core/graph/graph.h" #include "tensorflow/core/graph/node_builder.h" #include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/lib/io/path.h" #include "tensorflow/core/platform/env.h" #include "tensorflow/core/platform/test.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_util.cc
#include "tensorflow/core/framework/op_kernel.h" #include "tensorflow/core/framework/resource_mgr.h" #include "tensorflow/core/framework/types.h" #include "tensorflow/core/graph/algorithm.h" #include "tensorflow/core/graph/node_builder.h" #include "tensorflow/core/tfrt/common/global_state.h" #include "tensorflow/core/util/determinism.h" namespace tensorflow { namespace { constexpr const char* kPjRtDeviceCompilerResourceName = "pjrt_device_compiler";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 4.6K bytes - Viewed (0) -
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) -
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) -
tensorflow/c/c_api_internal.h
}; struct TF_OperationDescription { TF_OperationDescription(TF_Graph* g, const char* op_type, const char* node_name) : node_builder(node_name, op_type, g->graph.op_registry()), graph(g) {} tensorflow::NodeBuilder node_builder; TF_Graph* graph; std::set<tensorflow::string> colocation_constraints; }; struct TF_Operation { tensorflow::Node node; private:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat May 13 00:49:12 UTC 2023 - 7.6K bytes - Viewed (0) -
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) -
tensorflow/c/c_api.cc
Node* ret = nullptr; if (desc->graph->name_map.count(desc->node_builder.node_name())) { status->status = InvalidArgument("Duplicate node name in graph: '", desc->node_builder.node_name(), "'"); } else { if (!desc->colocation_constraints.empty()) { desc->node_builder.Attr( tensorflow::kColocationAttrName,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
tensorflow::StringPiece s(data, length); op_->node_builder.Attr(attr_name, s); return absl::OkStatus(); } Status SetAttrInt(const char* attr_name, int64_t value) override { op_->node_builder.Attr(attr_name, static_cast<int64_t>(value)); return absl::OkStatus(); } Status SetAttrFloat(const char* attr_name, float value) override { op_->node_builder.Attr(attr_name, value); return absl::OkStatus();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 20:00:09 UTC 2024 - 15.4K bytes - Viewed (0) -
pkg/scheduler/framework/plugins/volumebinding/test_utils.go
"k8s.io/utils/ptr" ) type nodeBuilder struct { *v1.Node } func makeNode(name string) nodeBuilder { return nodeBuilder{Node: &v1.Node{ ObjectMeta: metav1.ObjectMeta{ Name: name, Labels: map[string]string{ v1.LabelHostname: name, }, }, }} } func (nb nodeBuilder) withLabel(key, value string) nodeBuilder { if nb.Node.ObjectMeta.Labels == nil {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Sep 13 07:42:19 UTC 2023 - 4.9K bytes - Viewed (0)