- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TF_OperationOutputNumConsumers (0.24 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.h
// are added to the graph. TF_CAPI_EXPORT extern int TF_OperationOutputNumConsumers(TF_Output oper_out); // Get list of all current consumers of a specific output of an // operation. `consumers` must point to an array of length at least // `max_consumers` (ideally set to // TF_OperationOutputNumConsumers(oper_out)). Beware that a concurrent
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K 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)