- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TF_OperationOutputNumConsumers (0.25 sec)
-
tensorflow/c/c_api_test.cc
// Placeholder oper now has a consumer. ASSERT_EQ(1, TF_OperationOutputNumConsumers(TF_Output{feed, 0})); TF_Input feed_port; EXPECT_EQ(1, TF_OperationOutputConsumers(TF_Output{feed, 0}, &feed_port, 1)); EXPECT_EQ(add, feed_port.oper); EXPECT_EQ(0, feed_port.index); // The scalar const oper also has a consumer. ASSERT_EQ(1, TF_OperationOutputNumConsumers(TF_Output{three, 0})); TF_Input three_port; EXPECT_EQ(1,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/c_api.cc
inputs[edge->dst_input()] = {ToOperation(edge->src()), edge->src_output()}; } } } int TF_OperationOutputNumConsumers(TF_Output oper_out) { int count = 0; for (const auto* edge : oper_out.oper->node.out_edges()) { if (edge->src_output() == oper_out.index) { ++count; } } return count;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0)