Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_OperationOutputNumConsumers (0.2 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 30 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.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)
Back to top