- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for FindNodeByName (0.12 sec)
-
tensorflow/compiler/jit/clone_constants_for_better_clustering_test.cc
std::unique_ptr<Graph> result; TF_ASSERT_OK(CloneConstantsForBetterClustering(root, &result)); OutputTensor add1_operand; TF_ASSERT_OK( FindNodeByName(result.get(), "add1")->input_tensor(1, &add1_operand)); OutputTensor add2_operand; TF_ASSERT_OK( FindNodeByName(result.get(), "add2")->input_tensor(1, &add2_operand)); EXPECT_NE(add1_operand.node, add2_operand.node); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass_test.cc
Attr(kXlaCompileTimeConstantInputsAttr, compile_time_constant_inputs), Inputs(m_input, m_begin_s64, m_dynamic_slice_size)); Node* static_shaped_slice = testing::FindNodeByName( result.get(), "slice/static_shaped_slice/static_shaped_slice"); ASSERT_NE(static_shaped_slice, nullptr); EXPECT_THAT(static_shaped_slice, m_dynamic_slice); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
testing::FindNodeByName(graph.get(), "A") ->set_assigned_device_name(xla_cpu_device); testing::FindNodeByName(graph.get(), "tanh0") ->set_assigned_device_name(xla_cpu_device); testing::FindNodeByName(graph.get(), "tanh1") ->set_assigned_device_name(xla_cpu_device); testing::FindNodeByName(graph.get(), "tanh2") ->set_assigned_device_name(xla_cpu_device);
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/build_xla_ops_pass_test.cc
SessionOptions(), "/job:localhost/replica:0/task:0", &devices_) .ok()); } private: std::vector<std::unique_ptr<Device>> devices_; }; using ::tensorflow::testing::FindNodeByName; using ::tensorflow::testing::matchers::Attr; using ::tensorflow::testing::matchers::CtrlDeps; using ::tensorflow::testing::matchers::Inputs; using ::tensorflow::testing::matchers::NodeWith;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/jit/node_matchers.h
} ::testing::Matcher<impl::OutEdge> Const( const ::tensorflow::Input::Initializer& val); } // namespace matchers // If `g` has a node named `name` returns it, otherwise returns null. Node* FindNodeByName(Graph* g, absl::string_view name); } // namespace testing void PrintTo(const Node* n, ::std::ostream* os); void PrintTo(Node* n, ::std::ostream* os); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
{"cluster1:0", "cluster2:0"}, {"cluster1:0", "mul:0"}, {"cluster2:0", "mul:1"}, {"x:0", "cluster1:0"}}; EXPECT_EQ(expected_edges, GraphEdges(*graph)); } const Node* FindNodeByName(const Graph& graph, const string& name) { for (const Node* node : graph.nodes()) { if (node->name() == name) return node; } return nullptr; } bool HasGuaranteeConstAttr(const Node& n) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0)