Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_OperationOutputNumConsumers (0.24 sec)

  1. 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 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  2. 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 23 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
  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 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
Back to top